Upload RunBlazeFace.cs

RunBlazeFace.cs

@@ -55,8 +55,8 @@ public class RunBlazeFace : MonoBehaviour
 
     IWorker worker;
 
-    //Holds image size
-    int size;
+    //Holds input image size
+    int size = 128;
 
     Ops ops;
     ITensorAllocator allocator;
@@ -181,14 +181,14 @@ public class RunBlazeFace : MonoBehaviour
         int n = 0;
         for (int j = 0; j < 2 * 16 * 16; j++)
         {
-            offsets[n++] = (j / 2) % 16 - 7.5f;
-            offsets[n++] = (j / 2 / 16) - 7.5f;
+            offsets[n++] = 8 * ((j / 2) % 16 - 7.5f);
+            offsets[n++] = 8 * ((j / 2 / 16) - 7.5f);
             n += 2;
         }
         for (int j = 0; j < 6 * 8 * 8; j++)
         {
-            offsets[n++] = 2 * ((j / 6) % 8) - 7f;
-            offsets[n++] = 2 * ((j / 6 / 8)) - 7f;
+            offsets[n++] = 16 * ((j / 6) % 8 - 7f);
+            offsets[n++] = 16 * ((j / 6 / 8) - 7f);
             n += 2;
         }
         return offsets;
@@ -212,12 +212,10 @@ public class RunBlazeFace : MonoBehaviour
 
         model.AddLayer(new Lays.Transpose("scores", "classificators", new int[] { 0, 2, 1 }));
 
-        model.AddConstant(new Lays.Constant("eighth", new float[] { 1 / 8f }));
         model.AddConstant(new Lays.Constant("offsets",
             new TensorFloat(new TensorShape(1, 896, 4), offsets)
          ));
-        model.AddLayer(new Lays.Mul("boxes1scaled", "boxes1", "eighth"));
-        model.AddLayer(new Lays.Add("boxCoords", "boxes1scaled", "offsets"));
+        model.AddLayer(new Lays.Add("boxCoords", "boxes1", "offsets"));
         model.AddOutput("boxCoords");
 
         model.AddConstant(new Lays.Constant("maxOutputBoxes", new int[] { maxOutputBoxes }));
@@ -256,10 +254,10 @@ public class RunBlazeFace : MonoBehaviour
             {
                 var marker = new BoundingBox
                 {
-                    centerX = box.centerX + (regressors[0, n, 4 + j * 2] - regressors[0, n, 0]) * scale.x / 8,
-                    centerY = box.centerY + (regressors[0, n, 4 + j * 2 + 1] - regressors[0, n, 1]) * scale.y / 8,
-                    width = 1.0f * scale.x,
-                    height = 1.0f * scale.y,
+                    centerX = box.centerX + (regressors[0, n, 4 + j * 2] - regressors[0, n, 0]) * scale.x,
+                    centerY = box.centerY + (regressors[0, n, 4 + j * 2 + 1] - regressors[0, n, 1]) * scale.y,
+                    width = 8.0f * scale.x,
+                    height = 8.0f * scale.y,
                 };
                 DrawBox(marker, j < markerTextures.Length ? markerTextures[j] : faceTexture);
             }
@@ -294,7 +292,7 @@ public class RunBlazeFace : MonoBehaviour
 
         ClearAnnotations();
 
-        Vector2 markerScale = previewUI.rectTransform.rect.size/ 16;
+        Vector2 markerScale = previewUI.rectTransform.rect.size / size;
 
         DrawFaces(output, markersOutput, output.shape[0], markerScale);