Paul Bird commited on
Commit
fbcf6e3
·
verified ·
1 Parent(s): 50d85cd

Upload RunBlazeFace.cs

Browse files
Files changed (1) hide show
  1. 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
- new TensorShape(1, boxCoords.shape[0], boxCoords.shape[1], boxCoords.shape[2])) as TensorFloat;
288
- var output = ops.GatherND(boxCoords2, NM1, 0);
289
 
290
- using var regressors2 = regressors.ShallowReshape(
291
- new TensorShape(1, regressors.shape[0], regressors.shape[1], regressors.shape[2])) as TensorFloat;
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();