Simplified PyTorch hub for custom models (#1677)
Browse files- hubconf.py +1 -1
hubconf.py
CHANGED
@@ -129,7 +129,7 @@ def custom(path_or_model='path/to/model.pt'):
|
|
129 |
|
130 |
if __name__ == '__main__':
|
131 |
model = create(name='yolov5s', pretrained=True, channels=3, classes=80) # pretrained example
|
132 |
-
# model = custom(
|
133 |
model = model.autoshape() # for PIL/cv2/np inputs and NMS
|
134 |
|
135 |
# Verify inference
|
|
|
129 |
|
130 |
if __name__ == '__main__':
|
131 |
model = create(name='yolov5s', pretrained=True, channels=3, classes=80) # pretrained example
|
132 |
+
# model = custom(path_or_model='path/to/model.pt') # custom example
|
133 |
model = model.autoshape() # for PIL/cv2/np inputs and NMS
|
134 |
|
135 |
# Verify inference
|