Skip to content

Commit ab74414

Browse files
author
Tomasz Stanislawski
committed
fixup! make YOLO work with pyAIO
1 parent 06c79e6 commit ab74414

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

  • computer_vision/object_detection/yolo_v8

computer_vision/object_detection/yolo_v8/run.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -122,6 +122,7 @@ def run_single_pass(pytorch_runner, coco):
122122
anno_path, pre_processing="PyTorch_objdet", sort_ascending=True, order="NCHW")
123123

124124
from ultralytics import YOLO
125+
import numpy as np
125126
model = YOLO(model_path)
126127

127128
# make sure that model.predictor exists

0 commit comments

Comments
 (0)