Paul Bird
commited on
Upload RunBlazeFace.cs
Browse files- RunBlazeFace.cs +4 -6
RunBlazeFace.cs
CHANGED
|
@@ -283,13 +283,11 @@ public class RunBlazeFace : MonoBehaviour
|
|
| 283 |
|
| 284 |
var NM1 = _worker.PeekOutput("NMS") as TensorInt;
|
| 285 |
|
| 286 |
-
using var boxCoords2 = boxCoords.ShallowReshape(
|
| 287 |
-
|
| 288 |
-
var output = ops.GatherND(boxCoords2, NM1, 0);
|
| 289 |
|
| 290 |
-
using var regressors2 = regressors.ShallowReshape(
|
| 291 |
-
|
| 292 |
-
var markersOutput = ops.GatherND(regressors2, NM1, 0);
|
| 293 |
|
| 294 |
output.MakeReadable();
|
| 295 |
markersOutput.MakeReadable();
|
|
|
|
| 283 |
|
| 284 |
var NM1 = _worker.PeekOutput("NMS") as TensorInt;
|
| 285 |
|
| 286 |
+
using var boxCoords2 = boxCoords.ShallowReshape(boxCoords.shape.Unsqueeze(0)) as TensorFloat;
|
| 287 |
+
using var output = ops.GatherND(boxCoords2, NM1, 0);
|
|
|
|
| 288 |
|
| 289 |
+
using var regressors2 = regressors.ShallowReshape(regressors.shape.Unsqueeze(0)) as TensorFloat;
|
| 290 |
+
using var markersOutput = ops.GatherND(regressors2, NM1, 0);
|
|
|
|
| 291 |
|
| 292 |
output.MakeReadable();
|
| 293 |
markersOutput.MakeReadable();
|