PB Unity commited on
Commit
58dd36a
·
verified ·
1 Parent(s): bdab942

Upload RunBlazeFace.cs

Browse files
Files changed (1) hide show
  1. RunBlazeFace.cs +7 -10
RunBlazeFace.cs CHANGED
@@ -175,6 +175,7 @@ public class RunBlazeFace : MonoBehaviour
175
  }
176
 
177
  //Calculate the centers of the grid squares for two 16x16 grids and six 8x8 grids
 
178
  float[] GetGridBoxCoords()
179
  {
180
  var offsets = new float[896 * 4];
@@ -204,18 +205,14 @@ public class RunBlazeFace : MonoBehaviour
204
 
205
  size = model.inputs[0].shape.ToTensorShape()[1]; // Input tensor width
206
 
207
- model.AddConstant(new Lays.Constant("zero", new int[] { 0 }));
208
- model.AddConstant(new Lays.Constant("two", new int[] { 2 }));
209
- model.AddConstant(new Lays.Constant("four", new int[] { 4 }));
210
-
211
- model.AddLayer(new Lays.Slice("boxes1", "regressors", "zero", "four", "two"));
212
 
 
213
  model.AddLayer(new Lays.Transpose("scores", "classificators", new int[] { 0, 2, 1 }));
214
-
215
- model.AddConstant(new Lays.Constant("offsets",
216
- new TensorFloat(new TensorShape(1, 896, 4), offsets)
217
- ));
218
- model.AddLayer(new Lays.Add("boxCoords", "boxes1", "offsets"));
219
  model.AddOutput("boxCoords");
220
 
221
  model.AddConstant(new Lays.Constant("maxOutputBoxes", new int[] { maxOutputBoxes }));
 
175
  }
176
 
177
  //Calculate the centers of the grid squares for two 16x16 grids and six 8x8 grids
178
+ //The positions of the faces are given relative to these "anchor points"
179
  float[] GetGridBoxCoords()
180
  {
181
  var offsets = new float[896 * 4];
 
205
 
206
  size = model.inputs[0].shape.ToTensorShape()[1]; // Input tensor width
207
 
208
+ model.AddConstant(new Lays.Constant("0", new int[] { 0 }));
209
+ model.AddConstant(new Lays.Constant("2", new int[] { 2 }));
210
+ model.AddConstant(new Lays.Constant("4", new int[] { 4 }));
211
+ model.AddConstant(new Lays.Constant("offsets", new TensorFloat(new TensorShape(1, 896, 4), offsets)));
 
212
 
213
+ model.AddLayer(new Lays.Slice("boxes", "regressors", "0", "4", "2"));
214
  model.AddLayer(new Lays.Transpose("scores", "classificators", new int[] { 0, 2, 1 }));
215
+ model.AddLayer(new Lays.Add("boxCoords", "boxes", "offsets"));
 
 
 
 
216
  model.AddOutput("boxCoords");
217
 
218
  model.AddConstant(new Lays.Constant("maxOutputBoxes", new int[] { maxOutputBoxes }));