Skip to content

Commit

Permalink
Enable unittest of load_yolov5_model for voc
Browse files Browse the repository at this point in the history
  • Loading branch information
zhiqwang committed Mar 9, 2022
1 parent 7082df5 commit 701873d
Showing 1 changed file with 3 additions and 8 deletions.
11 changes: 3 additions & 8 deletions test/test_utils.py
Original file line number Diff line number Diff line change
@@ -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
Expand Down Expand Up @@ -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/"
Expand Down

0 comments on commit 701873d

Please sign in to comment.