Просмотр исходного кода

Uncomment OpenCV 4.5.4 requirement in detect.py (#5305)

modifyDataloader
Jebastin Nadar GitHub 3 лет назад
Родитель
Сommit
e0c3f42de7
Не найден GPG ключ соответствующий данной подписи Идентификатор GPG ключа: 4AEE18F83AFDEB23
1 измененных файлов: 1 добавлений и 1 удалений
  1. +1
    -1
      detect.py

+ 1
- 1
detect.py Просмотреть файл

@@ -89,7 +89,7 @@ def run(weights=ROOT / 'yolov5s.pt', # model.pt path(s)
modelc.load_state_dict(torch.load('resnet50.pt', map_location=device)['model']).to(device).eval()
elif onnx:
if dnn:
# check_requirements(('opencv-python>=4.5.4',))
check_requirements(('opencv-python>=4.5.4',))
net = cv2.dnn.readNetFromONNX(w)
else:
check_requirements(('onnx', 'onnxruntime-gpu' if torch.has_cuda else 'onnxruntime'))

Загрузка…
Отмена
Сохранить