Upload RunYOLO.cs

RunYOLO.cs

@@ -22,7 +22,7 @@ public class RunYOLO : MonoBehaviour
     const string videoName = "giraffes.mp4";
     // Link the classes.txt here:
     public TextAsset labelsAsset;
-    // Create a Raw Image in the scene of size 640x640 and link it here:
+    // Create a Raw Image in the scene and link it here:
     public RawImage displayImage;
     // Link to a bounding box texture here:
     public Sprite boxTexture;
@@ -35,6 +35,8 @@ public class RunYOLO : MonoBehaviour
     private string[] labels;
     private RenderTexture targetRT;
     const BackendType backend = BackendType.GPUCompute;
+
+    //Image size for the model
     private const int imageWidth = 640;
     private const int imageHeight = 640;
 
@@ -112,15 +114,21 @@ public class RunYOLO : MonoBehaviour
         var output = engine.PeekOutput() as TensorFloat;
         output.MakeReadable();
 
+        float displayWidth = displayImage.rectTransform.rect.width;
+        float displayHeight = displayImage.rectTransform.rect.height;
+
+        float scaleX = displayWidth / imageWidth;
+        float scaleY = displayHeight / imageHeight;
+
         //Draw the bounding boxes
         for (int n = 0; n < output.shape[0]; n++)
         {
             var box = new BoundingBox
             {
-                centerX = (output[n, 1] + output[n, 3] - imageWidth) / 2,
-                centerY = (output[n, 2] + output[n, 4] - imageHeight) / 2,
-                width = output[n, 3] - output[n, 1],
-                height = output[n, 4] - output[n, 2],
+                centerX = ((output[n, 1] + output[n, 3])*scaleX - displayWidth) / 2,
+                centerY = ((output[n, 2] + output[n, 4])*scaleY - displayHeight) / 2,
+                width = (output[n, 3] - output[n, 1])*scaleX,
+                height = (output[n, 4] - output[n, 2])*scaleY,
                 label = labels[(int)output[n, 5]],
                 confidence = Mathf.FloorToInt(output[n, 6] * 100 + 0.5f)
             };