PB Unity
commited on
Update RunBlazeFace.cs
Browse files- RunBlazeFace.cs +2 -2
RunBlazeFace.cs
CHANGED
@@ -277,10 +277,10 @@ public class RunBlazeFace : MonoBehaviour
|
|
277 |
|
278 |
var NM1 = worker.PeekOutput("NMS") as TensorInt;
|
279 |
|
280 |
-
using var boxCoords2 =
|
281 |
using var output = ops.GatherND(boxCoords2, NM1, 0);
|
282 |
|
283 |
-
using var regressors2 =
|
284 |
using var markersOutput = ops.GatherND(regressors2, NM1, 0);
|
285 |
|
286 |
output.MakeReadable();
|
|
|
277 |
|
278 |
var NM1 = worker.PeekOutput("NMS") as TensorInt;
|
279 |
|
280 |
+
using var boxCoords2 = ops.Reshape(boxCoords, boxCoords.shape.Unsqueeze(0));
|
281 |
using var output = ops.GatherND(boxCoords2, NM1, 0);
|
282 |
|
283 |
+
using var regressors2 = ops.Reshape(regressors,regressors.shape.Unsqueeze(0));
|
284 |
using var markersOutput = ops.GatherND(regressors2, NM1, 0);
|
285 |
|
286 |
output.MakeReadable();
|