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:
1 parent
d81bc47823
commit
201bafc7cf
4 files changed
+9
-10
No files matched your search
@@ -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
|
||||
|
||||
Reference in new issue
Block a user