diff --git a/hubconf.py b/hubconf.py index f8e7cd7268e6..96d6b12b8505 100644 --- a/hubconf.py +++ b/hubconf.py @@ -8,7 +8,6 @@ from pathlib import Path import torch -from PIL import Image from models.yolo import Model from utils.general import set_logging @@ -112,6 +111,8 @@ def yolov5x(pretrained=False, channels=3, classes=80): model = model.fuse().autoshape() # for PIL/cv2/np inputs and NMS # Verify inference + from PIL import Image + imgs = [Image.open(x) for x in Path('data/images').glob('*.jpg')] results = model(imgs) results.show()