Hugging Face's logo

Spaces:
xiang-wuu
/
yolov5
Runtime error

App Files Community
glenn-jocher commited on
Commit
2062765
•
1 Parent(s): ca290dc

auto-download v3.1 models (#1259)

Browse files
Files changed (2) hide show
  1. README.md +6 -6
  2. utils/google_utils.py +1 -1
README.md CHANGED
@@ -21,14 +21,14 @@ This repository represents Ultralytics open-source research into future object d
21
 
22
  | Model | AP<sup>val</sup> | AP<sup>test</sup> | AP<sub>50</sub> | Speed<sub>GPU</sub> | FPS<sub>GPU</sub> || params | FLOPS |
23
  |---------- |------ |------ |------ | -------- | ------| ------ |------ | :------: |
24
- | [YOLOv5s](https://github.com/ultralytics/yolov5/releases/tag/v3.0) | 37.0 | 37.0 | 56.2 | **2.4ms** | **416** || 7.5M | 13.2B
25
- | [YOLOv5m](https://github.com/ultralytics/yolov5/releases/tag/v3.0) | 44.3 | 44.3 | 63.2 | 3.4ms | 294 || 21.8M | 39.4B
26
- | [YOLOv5l](https://github.com/ultralytics/yolov5/releases/tag/v3.0) | 47.7 | 47.7 | 66.5 | 4.4ms | 227 || 47.8M | 88.1B
27
- | [YOLOv5x](https://github.com/ultralytics/yolov5/releases/tag/v3.0) | **49.2** | **49.2** | **67.7** | 6.9ms | 145 || 89.0M | 166.4B
28
  | | | | | | || |
29
- | [YOLOv5x](https://github.com/ultralytics/yolov5/releases/tag/v3.0) + TTA|**50.8**| **50.8** | **68.9** | 25.5ms | 39 || 89.0M | 354.3B
30
  | | | | | | || |
31
- | [YOLOv3-SPP](https://github.com/ultralytics/yolov5/releases/tag/v3.0) | 45.6 | 45.5 | 65.2 | 4.5ms | 222 || 63.0M | 118.0B
32
 
33
  ** AP<sup>test</sup> denotes COCO [test-dev2017](http://cocodataset.org/#upload) server results, all other AP results in the table denote val2017 accuracy.
34
  ** All AP numbers are for single-model single-scale without ensemble or test-time augmentation. **Reproduce** by `python test.py --data coco.yaml --img 640 --conf 0.001`
 
21
 
22
  | Model | AP<sup>val</sup> | AP<sup>test</sup> | AP<sub>50</sub> | Speed<sub>GPU</sub> | FPS<sub>GPU</sub> || params | FLOPS |
23
  |---------- |------ |------ |------ | -------- | ------| ------ |------ | :------: |
24
+ | [YOLOv5s](https://github.com/ultralytics/yolov5/releases) | 37.0 | 37.0 | 56.2 | **2.4ms** | **416** || 7.5M | 13.2B
25
+ | [YOLOv5m](https://github.com/ultralytics/yolov5/releases) | 44.3 | 44.3 | 63.2 | 3.4ms | 294 || 21.8M | 39.4B
26
+ | [YOLOv5l](https://github.com/ultralytics/yolov5/releases) | 47.7 | 47.7 | 66.5 | 4.4ms | 227 || 47.8M | 88.1B
27
+ | [YOLOv5x](https://github.com/ultralytics/yolov5/releases) | **49.2** | **49.2** | **67.7** | 6.9ms | 145 || 89.0M | 166.4B
28
  | | | | | | || |
29
+ | [YOLOv5x](https://github.com/ultralytics/yolov5/releases) + TTA|**50.8**| **50.8** | **68.9** | 25.5ms | 39 || 89.0M | 354.3B
30
  | | | | | | || |
31
+ | [YOLOv3-SPP](https://github.com/ultralytics/yolov5/releases) | 45.6 | 45.5 | 65.2 | 4.5ms | 222 || 63.0M | 118.0B
32
 
33
  ** AP<sup>test</sup> denotes COCO [test-dev2017](http://cocodataset.org/#upload) server results, all other AP results in the table denote val2017 accuracy.
34
  ** All AP numbers are for single-model single-scale without ensemble or test-time augmentation. **Reproduce** by `python test.py --data coco.yaml --img 640 --conf 0.001`
utils/google_utils.py CHANGED
@@ -36,7 +36,7 @@ def attempt_download(weights):
36
  # return
37
 
38
  try: # GitHub
39
- url = 'https://github.com/ultralytics/yolov5/releases/download/v3.0/' + file
40
  print('Downloading %s to %s...' % (url, weights))
41
  torch.hub.download_url_to_file(url, weights)
42
  assert os.path.exists(weights) and os.path.getsize(weights) > 1E6 # check
 
36
  # return
37
 
38
  try: # GitHub
39
+ url = 'https://github.com/ultralytics/yolov5/releases/download/v3.1/' + file
40
  print('Downloading %s to %s...' % (url, weights))
41
  torch.hub.download_url_to_file(url, weights)
42
  assert os.path.exists(weights) and os.path.getsize(weights) > 1E6 # check