YOLOv5 release v6.0 (#5141)
* Update P5 models * Update P6 models * Update with GFLOPs and Params * Update with GFLOPs and Params * Update README * Update * Update README * Update * Update * Add times * Update README * Update results * Update results * Update results * Update hyps * Update plots * Update plots * Update README.md * Add nano models to hubconf.py
This commit is contained in:
1 parent
938efe5dcf
commit
956be8e642
15 files changed
+228
-82
No files matched your search
+10
@@ -70,6 +70,11 @@ def custom(path='path/to/model.pt', autoshape=True, verbose=True, device=None):
|
||||
return _create(path, autoshape=autoshape, verbose=verbose, device=device)
|
||||
|
||||
|
||||
def yolov5n(pretrained=True, channels=3, classes=80, autoshape=True, verbose=True, device=None):
|
||||
# YOLOv5-nano model https://github.com/ultralytics/yolov5
|
||||
return _create('yolov5n', pretrained, channels, classes, autoshape, verbose, device)
|
||||
|
||||
|
||||
def yolov5s(pretrained=True, channels=3, classes=80, autoshape=True, verbose=True, device=None):
|
||||
# YOLOv5-small model https://github.com/ultralytics/yolov5
|
||||
return _create('yolov5s', pretrained, channels, classes, autoshape, verbose, device)
|
||||
@@ -90,6 +95,11 @@ def yolov5x(pretrained=True, channels=3, classes=80, autoshape=True, verbose=Tru
|
||||
return _create('yolov5x', pretrained, channels, classes, autoshape, verbose, device)
|
||||
|
||||
|
||||
def yolov5n6(pretrained=True, channels=3, classes=80, autoshape=True, verbose=True, device=None):
|
||||
# YOLOv5-nano-P6 model https://github.com/ultralytics/yolov5
|
||||
return _create('yolov5n6', pretrained, channels, classes, autoshape, verbose, device)
|
||||
|
||||
|
||||
def yolov5s6(pretrained=True, channels=3, classes=80, autoshape=True, verbose=True, device=None):
|
||||
# YOLOv5-small-P6 model https://github.com/ultralytics/yolov5
|
||||
return _create('yolov5s6', pretrained, channels, classes, autoshape, verbose, device)
|
||||
|
||||
Reference in new issue
Block a user