Simplify autoshape() post-process (#1653)
* Simplify autoshape() post-process * cleanup * cleanup
This commit is contained in:
1 parent
84f9bb5d92
commit
fa8f1fb0e9
4 files changed
+9
-10
No files matched your search
+1
-1
@@ -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
|
||||
|
||||
Reference in new issue
Block a user