Add YOLO v5m 640x640 (open-edge-platform#185)

* Add YOLO v5m 640x640 as an object detection model choice.
* Rename other models with size label.

Fixes ITEP-67471

Author: dnoliver

tools/visual-pipeline-and-platform-evaluation-tool/app.py

@@ -487,10 +487,11 @@ def create_interface():
         label="Object Detection Model",
         choices=[
             "SSDLite MobileNet V2",
-            "YOLO v5m",
-            "YOLO v5s",
+            "YOLO v5m 416x416",
+            "YOLO v5s 416x416",
+            "YOLO v5m 640x640"
         ],
-        value="YOLO v5s",
+        value="YOLO v5s 416x416",
     )
 
     # Object detection device

tools/visual-pipeline-and-platform-evaluation-tool/compose.yml

@@ -45,7 +45,8 @@ services:
           cp -r pipeline-zoo-models/storage/ssdlite_mobilenet_v2_INT8/ /output/pipeline-zoo-models &&
           cp -r pipeline-zoo-models/storage/resnet-50-tf_INT8/ /output/pipeline-zoo-models/ &&
           cp -r pipeline-zoo-models/storage/yolov5m-416_INT8/ /output/pipeline-zoo-models/ &&
-          cp -r pipeline-zoo-models/storage/yolov5s-416_INT8/ /output/pipeline-zoo-models/
+          cp -r pipeline-zoo-models/storage/yolov5s-416_INT8/ /output/pipeline-zoo-models/ &&
+          cp -r pipeline-zoo-models/storage/yolov5m-640_INT8/ /output/pipeline-zoo-models/
         ) &&
         touch /output/.done &&
         tail -f /dev/null

tools/visual-pipeline-and-platform-evaluation-tool/utils.py

@@ -67,14 +67,21 @@ def prepare_video_and_constants(
             constants["OBJECT_DETECTION_MODEL_PROC"] = (
                 f"{MODELS_PATH}/pipeline-zoo-models/ssdlite_mobilenet_v2_INT8/ssdlite_mobilenet_v2.json"
             )
-        case "YOLO v5m":
+        case "YOLO v5m 416x416":
             constants["OBJECT_DETECTION_MODEL_PATH"] = (
                 f"{MODELS_PATH}/pipeline-zoo-models/yolov5m-416_INT8/FP16-INT8/yolov5m-416_INT8.xml"
             )
             constants["OBJECT_DETECTION_MODEL_PROC"] = (
                 f"{MODELS_PATH}/pipeline-zoo-models/yolov5m-416_INT8/yolo-v5.json"
             )
-        case "YOLO v5s":
+        case "YOLO v5m 640x640":
+            constants["OBJECT_DETECTION_MODEL_PATH"] = (
+                f"{MODELS_PATH}/pipeline-zoo-models/yolov5m-640_INT8/FP16-INT8/yolov5m-640_INT8.xml"
+            )
+            constants["OBJECT_DETECTION_MODEL_PROC"] = (
+                f"{MODELS_PATH}/pipeline-zoo-models/yolov5m-640_INT8/yolo-v5.json"
+            )
+        case "YOLO v5s 416x416":
             constants["OBJECT_DETECTION_MODEL_PATH"] = (
                 f"{MODELS_PATH}/pipeline-zoo-models/yolov5s-416_INT8/FP16-INT8/yolov5s.xml"
             )