This commit is contained in:
Glenn Jocher 2019-11-27 14:59:46 -10:00
parent 9c1d7d5248
commit 413afab11c
1 changed files with 8 additions and 22 deletions

View File

@ -139,30 +139,16 @@ Success: converted 'weights/yolov3-spp.pt' to 'converted.weights'
- `test.py --weights weights/yolov3.weights` tests official YOLOv3 weights. - `test.py --weights weights/yolov3.weights` tests official YOLOv3 weights.
- `test.py --weights weights/last.pt` tests latest checkpoint. - `test.py --weights weights/last.pt` tests latest checkpoint.
- Compare to darknet published results https://arxiv.org/abs/1804.02767. - mAPs on COCO2014 using pycocotools.
- mAP@0.5 run at --iou-thres 0.5, mAP@0.5 run at --iou-thres 0.65
- YOLOv3-SPP ultralytics is `ultralytics68.pt` with `yolov3-spp.cfg`.
- Darknet results published in https://arxiv.org/abs/1804.02767.
<!-- mAPs@0.5:0.95 obtained at --conf-thres 0.65 --> <i></i> |resolution |COCO mAP<br>@0.5...0.95 |COCO mAP<br>@0.5
<!-- ultralytics model is last68.pt -->
<i></i> |320<br>mAP@0.5:0.95 |416<br>mAP@0.5:0.95 |608<br>mAP@0.5:0.95
--- | --- | --- | --- --- | --- | --- | ---
darknet `YOLOv3-tiny` | 14.0 | 16.0 | 16.6 YOLOv3-tiny<br>YOLOv3<br>YOLOv3-SPP<br>**YOLOv3-SPP ultralytics** |320 |14.0<br>28.7<br>30.5<br>**35.2** |29.0<br>51.5<br>52.3<br>**53.9**
darknet `YOLOv3` | 28.7 | 31.1 | 33.0 YOLOv3-tiny<br>YOLOv3<br>YOLOv3-SPP<br>**YOLOv3-SPP ultralytics** |416 |16.0<br>31.1<br>33.9<br>**38.8** |32.9<br>55.3<br>56.8<br>**58.7**
darknet `YOLOv3-SPP` | 30.5 | 33.9 | 37.0 YOLOv3-tiny<br>YOLOv3<br>YOLOv3-SPP<br>**YOLOv3-SPP ultralytics** |608 |16.6<br>33.0<br>37.0<br>**40.4** |35.5<br>57.9<br>**60.6**<br>60.1
**ultralytics** `YOLOv3-SPP` | **35.2** | **38.8** | **40.4**
<!-- mAPs@0.5 obtained at --conf-thres 0.5 -->
<i></i> |320<br>mAP@0.5 |416<br>mAP@0.5 |608<br>mAP@0.5
--- | --- | --- | ---
darknet `YOLOv3-tiny` | 29.0 | 32.9 | 35.5
darknet `YOLOv3` | 51.5 | 55.3 | 57.9
darknet `YOLOv3-SPP` | 52.3 | 56.8 | **60.6**
**ultralytics** `YOLOv3-SPP` | **53.9** | **58.7** | 60.1
<i></i> |resolution |mAP<br>0.5:0.95 |mAP<br>0.5
--- | --- | --- | ---
YOLOv3-tiny<br>YOLOv3<br>YOLOv3-SPP<br>YOLOv3-SPP ultralytics |320 |14.0<br>28.7<br>30.5<br>**35.2** |29.0<br>51.5<br>52.3<br>**53.9**
YOLOv3-tiny<br>YOLOv3<br>YOLOv3-SPP<br>YOLOv3-SPP ultralytics |416 |16.0<br>31.1<br>33.9<br>**38.8** |32.9<br>55.3<br>56.8<br>**58.7**
YOLOv3-tiny<br>YOLOv3<br>YOLOv3-SPP<br>YOLOv3-SPP ultralytics |608 |16.6<br>33.0<br>37.0<br>**40.4** |35.5<br>57.9<br>**60.6**<br>60.1
```bash ```bash
$ python3 test.py --save-json --img-size 608 --weights ultralytics68.pt $ python3 test.py --save-json --img-size 608 --weights ultralytics68.pt