Add yolov5m test to mv_test

[Issue type] update

Even yolov5m isn't different from yolv5s but there is a requirement of
testing the both by mv_test.

Change-Id: I7fa8e432a18391bc975f121e17af9fed68b680e5
Signed-off-by: Tae-Young Chung <ty83.chung@samsung.com>

diff --git a/test/testsuites/machine_learning/inference/test_object_detection.cpp b/test/testsuites/machine_learning/inference/test_object_detection.cpp
index 13bbb29dba46f1563717374ce87b167c88a5e591..981128001c1f39f66f34a72600215aadebd813b2 100644 (file)
--- a/test/testsuites/machine_learning/inference/test_object_detection.cpp
+++ b/test/testsuites/machine_learning/inference/test_object_detection.cpp
@@ -46,9 +46,12 @@
 #define OD_TFLITE_WEIGHT_MOBILENET_V2_SSD_320_PATH \
        MV_CONFIG_PATH                                 \
        "/models/OD/tflite/od_mobilenet_v2_ssd_320x320.tflite"
-#define OD_SNPE_WEIGHT_QC_YOLO_V5_MULTIANCHOR_PATH \
+#define OD_SNPE_WEIGHT_QC_YOLO_V5s_MULTIANCHOR_PATH \
        MV_CONFIG_PATH                                 \
        "/models/OD/snpe/yolov5s_quantize.dlc"
+#define OD_SNPE_WEIGHT_QC_YOLO_V5m_MULTIANCHOR_PATH \
+       MV_CONFIG_PATH                                 \
+       "/models/OD/snpe/yolov5m_quantize.dlc"
 
 #define OD_SNPE_LABEL_QC_YOLO_V5_MULTIANCHOR_PATH \
        MV_CONFIG_PATH                                \
@@ -166,14 +169,22 @@ TEST_P(TestObjectDetectionSnpe, DISABLED_EFDLite2QC)
        inference("person", IMG_BASEBALL);
 }
 
-TEST_P(TestObjectDetectionSnpe, DISABLED_YoloV5MultiAnchor)
+TEST_P(TestObjectDetectionSnpe, DISABLED_YoloV5sMultiAnchor)
 
 {
        ASSERT_TRUE(_use_json_parser);
-       engine_config_hosted_snpe_model(engine_cfg, OD_SNPE_WEIGHT_QC_YOLO_V5_MULTIANCHOR_PATH,
+       engine_config_hosted_snpe_model(engine_cfg, OD_SNPE_WEIGHT_QC_YOLO_V5s_MULTIANCHOR_PATH,
                                                                        OD_SNPE_LABEL_QC_YOLO_V5_MULTIANCHOR_PATH, _use_json_parser, _target_device_type);
        inference("bus", IMG_BUS);
 }
 
+TEST_P(TestObjectDetectionSnpe, DISABLED_YoloV5mMultiAnchor)
+
+{
+       ASSERT_TRUE(_use_json_parser);
+       engine_config_hosted_snpe_model(engine_cfg, OD_SNPE_WEIGHT_QC_YOLO_V5m_MULTIANCHOR_PATH,
+                                                                       OD_SNPE_LABEL_QC_YOLO_V5_MULTIANCHOR_PATH, _use_json_parser, _target_device_type);
+       inference("bus", IMG_BUS);
+}
 INSTANTIATE_TEST_CASE_P(Prefix, TestObjectDetectionSnpe,
                                                ::testing::Values(ParamTypes(true, MV_INFERENCE_TARGET_DEVICE_CUSTOM)));