Fixing test_load_from_yolov5

zhiqwang · Oct 10, 2021 · 6dd2cc7 · 6dd2cc7
1 parent 6c21a1c
commit 6dd2cc7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/test/test_models.py b/test/test_models.py
@@ -316,7 +316,8 @@ def test_load_from_yolov5(arch, version, hash_prefix):
             hash_prefix=hash_prefix,
         )
 
-    model_yolov5 = YOLOv5.load_from_yolov5(yolov5s_r40_path, version=version)
+    up_version = version.replace("v", "r")
+    model_yolov5 = YOLOv5.load_from_yolov5(yolov5s_r40_path, version=up_version)
     model_yolov5.eval()
     out_from_yolov5 = model_yolov5.predict(img_path)
     assert isinstance(out_from_yolov5[0], dict)