Upload RunYOLO.cs

RunYOLO.cs

@@ -40,6 +40,7 @@ public class RunYOLO : MonoBehaviour
 
     private VideoPlayer video;
 
+    List<GameObject> boxPool = new List<GameObject>();
     //bounding box data
     public struct BoundingBox
     {
@@ -60,7 +61,7 @@ public class RunYOLO : MonoBehaviour
         labels = labelsAsset.text.Split('\n');
 
         //Load model
-        model = ModelLoader.Load(Application.streamingAssetsPath + "/" + modelName);
+        model = ModelLoader.Load(Application.streamingAssetsPath +"/"+ modelName);
 
         targetRT = new RenderTexture(imageWidth, imageHeight, 0);
 
@@ -123,51 +124,75 @@ public class RunYOLO : MonoBehaviour
                 label = labels[(int)output[n, 5]],
                 confidence = Mathf.FloorToInt(output[n, 6] * 100 + 0.5f)
             };
-            DrawBox(box);
+            DrawBox(box, n);
         }
     }
 
-    public void DrawBox(BoundingBox box)
+    public void DrawBox(BoundingBox box , int id)
     {
-        Color color = Color.yellow;
+        //Create the bounding box graphic or get from pool
+        GameObject panel;
+        if (id < boxPool.Count)
+        {
+            panel = boxPool[id];
+            panel.SetActive(true);
+        }
+        else
+        {
+            panel = CreateNewBox(Color.yellow);
+        }
+        //Set box position
+        panel.transform.localPosition = new Vector3(box.centerX, -box.centerY);
+
+        //Set box size
+        RectTransform rt = panel.GetComponent<RectTransform>();
+        rt.sizeDelta = new Vector2(box.width, box.height);
+        
+        //Set label text
+        var label = panel.GetComponentInChildren<Text>();
+        label.text = box.label + " (" + box.confidence + "%)";
+    }
+
+    public GameObject CreateNewBox(Color color)
+    {
+        //Create the box and set image
 
-        //Create the bounding box graphic
         var panel = new GameObject("ObjectBox");
         panel.AddComponent<CanvasRenderer>();
         Image img = panel.AddComponent<Image>();
         img.color = color;
-        img.sprite  = boxTexture;
+        img.sprite = boxTexture;
+        img.type = Image.Type.Sliced;
         panel.transform.SetParent(displayLocation, false);
-        panel.transform.localPosition = new Vector3(box.centerX, -box.centerY);
-        RectTransform rt = panel.GetComponent<RectTransform>();
-        rt.sizeDelta = new Vector2(box.width, box.height);
 
-        //Add the class label
+        //Create the label
+
         var text = new GameObject("ObjectLabel");
         text.AddComponent<CanvasRenderer>();
-        Text txt = text.AddComponent<Text>();
         text.transform.SetParent(panel.transform, false);
+        Text txt = text.AddComponent<Text>();
         txt.font = font;
-        txt.text = box.label + " (" + box.confidence + "%)";
         txt.color = color;
         txt.fontSize = 40;
         txt.horizontalOverflow = HorizontalWrapMode.Overflow;
+
         RectTransform rt2 = text.GetComponent<RectTransform>();
         rt2.offsetMin = new Vector2(20, rt2.offsetMin.y);
         rt2.offsetMax = new Vector2(0, rt2.offsetMax.y);
-        rt2.offsetMax = new Vector2(rt2.offsetMax.x, 30);
         rt2.offsetMin = new Vector2(rt2.offsetMin.x, 0);
-        rt2.anchorMin = new Vector2(0,0);
+        rt2.offsetMax = new Vector2(rt2.offsetMax.x, 30);
+        rt2.anchorMin = new Vector2(0, 0);
         rt2.anchorMax = new Vector2(1, 1);
 
-        img.sprite = boxTexture;
-        img.type = Image.Type.Sliced;
+        boxPool.Add(panel);
+        return panel;
     }
 
     public void ClearAnnotations()
     {
-        foreach (Transform child in displayLocation) {
-            Destroy(child.gameObject);
+        foreach(var box in boxPool)
+        {
+            box.SetActive(false);
         }
     }