Sync train and test iou_thresh (#1465)

* Sync train and test iou_thresh

* Sync train and test iou_thresh

* weights names .lower()

* Notebook update
This commit is contained in:
Glenn Jocher authored and GitHub committed 2020-11-21 12:38:35 +01:00
1 parent d81bc47823
commit 201bafc7cf
4 files changed
+9 -10

No files matched your search

+1 -1
View File
@@ -18,7 +18,7 @@ def gsutil_getsize(url=''):
def attempt_download(weights):
# Attempt to download pretrained weights if not found locally
weights = weights.strip().replace("'", '')
file = Path(weights).name
file = Path(weights).name.lower()
msg = weights + ' missing, try downloading from https://github.com/ultralytics/yolov5/releases/'
models = ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt'] # available models