Improved hubconf.py CI tests (#2251)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-02-19 12:35:38 -08:00
1 parent 9d87307784
commit db28ce61ac
2 files changed
+8 -3

No files matched your search

+7 -2
View File
@@ -133,9 +133,14 @@ if __name__ == '__main__':
# model = custom(path_or_model='path/to/model.pt') # custom example
# Verify inference
import numpy as np
from PIL import Image
imgs = [Image.open(x) for x in Path('data/images').glob('*.jpg')]
results = model(imgs)
imgs = [Image.open('data/images/bus.jpg'), # PIL
'data/images/zidane.jpg', # filename
'https://github.com/ultralytics/yolov5/raw/master/data/images/bus.jpg', # URI
np.zeros((640, 480, 3))] # numpy
results = model(imgs) # batched inference
results.print()
results.save()