Simplify autoshape() post-process (#1653)

* Simplify autoshape() post-process

* cleanup

* cleanup
This commit is contained in:
Glenn Jocher authored and GitHub committed 2020-12-09 07:44:06 -08:00
1 parent 84f9bb5d92
commit fa8f1fb0e9
4 files changed
+9 -10

No files matched your search

+1 -1
View File
@@ -108,7 +108,7 @@ def yolov5x(pretrained=False, channels=3, classes=80):
if __name__ == '__main__':
model = create(name='yolov5s', pretrained=True, channels=3, classes=80) # example
model = model.fuse().autoshape() # for PIL/cv2/np inputs and NMS
model = model.autoshape() # for PIL/cv2/np inputs and NMS
# Verify inference
from PIL import Image