From 701873d540285e468cc4bb081af55ca0b792f3dc Mon Sep 17 00:00:00 2001 From: Zhiqiang Wang Date: Thu, 10 Mar 2022 02:27:03 +0800 Subject: [PATCH] Enable unittest of load_yolov5_model for voc --- test/test_utils.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/test/test_utils.py b/test/test_utils.py index d1d575424..953eb3f90 100644 --- a/test/test_utils.py +++ b/test/test_utils.py @@ -1,4 +1,5 @@ -# Copyright (c) 2021, Zhiqiang Wang. All Rights Reserved. +# Copyright (c) 2021, yolort team. All rights reserved. + import cv2 import numpy as np import pytest @@ -65,17 +66,11 @@ def test_load_from_ultralytics( assert len(model_info["strides"]) == 4 if use_p6 else 3 -@pytest.mark.skip(reason="Due to #235") @pytest.mark.parametrize( "arch, version, upstream_version, hash_prefix", [("yolov5s-VOC", "r4.0", "v5.0", "23818cff")], ) -def test_load_from_ultralytics_voc( - arch: str, - version: str, - upstream_version: str, - hash_prefix: str, -): +def test_load_from_ultralytics_voc(arch, version, upstream_version, hash_prefix): img_path = "test/assets/bus.jpg" base_url = "https://github.com/ultralytics/yolov5/releases/download/"