Skip to content

Commit

Permalink
modelartifacts links now point to 08_01 release
Browse files Browse the repository at this point in the history
  • Loading branch information
Manu Mathew committed Dec 16, 2021
1 parent cb47352 commit f1b50dc
Show file tree
Hide file tree
Showing 8 changed files with 31 additions and 26 deletions.
16 changes: 8 additions & 8 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -10,14 +10,14 @@ This repository is based on [ultralytics/yolov5](https://github.com/ultralytics/

## **Official Models from Ultralytics**

|Dataset |Model Name |Input Size |GigaMACS |AP[0.5:0.95]%| AP50%|Notes |
|Dataset |Model Name |Input Size |GFLOPS |AP[0.5:0.95]%| AP50%|Notes |
|--------|------------------------------- |-----------|----------|-------------|------|----- |
|COCO |Yolov5s6 |1280x1280 |**34.8** | 43.3 | 61.9 |
|COCO |Yolov5s6_640 |640x640 |**8.7** | 38.9 | 56.8 |(Train@ 1280, val@640) |
|COCO |Yolov5m6 |1280x1280 |**104.8**| 50.5 | 68.7 | |
|COCO |Yolov5m6_640 |640x640 |**26.2** | 45.4 | 63.6 |(Train@ 1280, val@640) |
|COCO |Yolov5l6 |1280x1280 |**235.4**| 53.4 | 71.1 | |
|COCO |Yolov5l6_640 |640x640 |**58.85**| 49.0 | 67.0 |(Train@ 1280, val@640) |
|COCO |Yolov5s6 |1280x1280 |**69.6** | 43.3 | 61.9 |
|COCO |Yolov5s6_640 |640x640 |**17.4** | 38.9 | 56.8 |(Train@ 1280, val@640) |
|COCO |Yolov5m6 |1280x1280 |**209.6**| 50.5 | 68.7 | |
|COCO |Yolov5m6_640 |640x640 |**52.4** | 45.4 | 63.6 |(Train@ 1280, val@640) |
|COCO |Yolov5l6 |1280x1280 |**470.8**| 53.4 | 71.1 | |
|COCO |Yolov5l6_640 |640x640 |**117.7**| 49.0 | 67.0 |(Train@ 1280, val@640) |

<br/>

Expand Down Expand Up @@ -91,7 +91,7 @@ python test.py --data coco.yaml --img 512 --conf 0.001 --iou 0.65 --weights pret
### **ONNX export including detection:**
* Run the following command to export the entire models including the detection part,
```
python models/export.py --weights pretrained_models/yolov5s6_640_ti_lite/weights/best.pt --img 640 --batch 1 --simplify --export-nms --opset 11 # export at 640x640 with batch size 1
python export.py --weights pretrained_models/yolov5s6_640_ti_lite/weights/best.pt --img 640 --batch 1 --simplify --export-nms --opset 11 # export at 640x640 with batch size 1
```

## **References**
Expand Down
10 changes: 5 additions & 5 deletions pretrained_models/modelartifacts/8bits/artifacts.list
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
task_type,session_name,run_dir,model_name,size,shortlisted
detection,onnxrt,od-8100_onnxrt_weights_yolov5s6_640_ti_lite_37p4_56p0_onnx,weights_yolov5s6_640_ti_lite_37p4_56p0.onnx,51596848,True
detection,onnxrt,od-8110_onnxrt_weights_yolov5s6_384_ti_lite_32p8_51p2_onnx,weights_yolov5s6_384_ti_lite_32p8_51p2.onnx,51397030,True
detection,onnxrt,od-8120_onnxrt_weights_yolov5m6_640_ti_lite_44p1_62p9_onnx,weights_yolov5m6_640_ti_lite_44p1_62p9.onnx,141679412,True
detection,onnxrt,od-8130_onnxrt_weights_yolov5l6_640_ti_lite_47p1_65p6_onnx,weights_yolov5l6_640_ti_lite_47p1_65p6.onnx,299842909,False
task_type,session_name,run_dir,model_name,size,shortlisted,recommended
detection,onnxrt,od-8100_onnxrt_weights_yolov5s6_640_ti_lite_37p4_56p0_onnx,ONR-OD-8100-yolov5-s6-ti-lite-coco-640x640,51608959,True,False
detection,onnxrt,od-8110_onnxrt_weights_yolov5s6_384_ti_lite_32p8_51p2_onnx,ONR-OD-8110-yolov5-s6-ti-lite-coco-384x384,51404863,True,False
detection,onnxrt,od-8120_onnxrt_weights_yolov5m6_640_ti_lite_44p1_62p9_onnx,ONR-OD-8120-yolov5-m6-ti-lite-coco-640x640,141703361,False,False
detection,onnxrt,od-8130_onnxrt_weights_yolov5l6_640_ti_lite_47p1_65p6_onnx,ONR-OD-8130-yolov5-l6-ti-lite-coco-640x640,299199158,False,False
22 changes: 13 additions & 9 deletions pretrained_models/modelartifacts/8bits/artifacts.yaml
Original file line number Diff line number Diff line change
@@ -1,28 +1,32 @@
od-8100_onnxrt:
model_name: weights_yolov5s6_640_ti_lite_37p4_56p0.onnx
model_name: ONR-OD-8100-yolov5-s6-ti-lite-coco-640x640
recommended: false
run_dir: od-8100_onnxrt_weights_yolov5s6_640_ti_lite_37p4_56p0_onnx
session_name: onnxrt
shortlisted: true
size: 51596848
size: 51608959
task_type: detection
od-8110_onnxrt:
model_name: weights_yolov5s6_384_ti_lite_32p8_51p2.onnx
model_name: ONR-OD-8110-yolov5-s6-ti-lite-coco-384x384
recommended: false
run_dir: od-8110_onnxrt_weights_yolov5s6_384_ti_lite_32p8_51p2_onnx
session_name: onnxrt
shortlisted: true
size: 51397030
size: 51404863
task_type: detection
od-8120_onnxrt:
model_name: weights_yolov5m6_640_ti_lite_44p1_62p9.onnx
model_name: ONR-OD-8120-yolov5-m6-ti-lite-coco-640x640
recommended: false
run_dir: od-8120_onnxrt_weights_yolov5m6_640_ti_lite_44p1_62p9_onnx
session_name: onnxrt
shortlisted: true
size: 141679412
shortlisted: false
size: 141703361
task_type: detection
od-8130_onnxrt:
model_name: weights_yolov5l6_640_ti_lite_47p1_65p6.onnx
model_name: ONR-OD-8130-yolov5-l6-ti-lite-coco-640x640
recommended: false
run_dir: od-8130_onnxrt_weights_yolov5l6_640_ti_lite_47p1_65p6_onnx
session_name: onnxrt
shortlisted: false
size: 299842909
size: 299199158
task_type: detection
Original file line number Diff line number Diff line change
@@ -1 +1 @@
http://software-dl.ti.com/jacinto7/esd/modelzoo/gplv3/08_00_00_05/edgeai-yolov5/pretrained_models/modelartifacts/8bits/od-8100_onnxrt_weights_yolov5s6_640_ti_lite_37p4_56p0_onnx.tar.gz
http://software-dl.ti.com/jacinto7/esd/modelzoo/gplv3/08_01_00_05/edgeai-yolov5/pretrained_models/modelartifacts/8bits/od-8100_onnxrt_weights_yolov5s6_640_ti_lite_37p4_56p0_onnx.tar.gz
Original file line number Diff line number Diff line change
@@ -1 +1 @@
http://software-dl.ti.com/jacinto7/esd/modelzoo/gplv3/08_00_00_05/edgeai-yolov5/pretrained_models/modelartifacts/8bits/od-8110_onnxrt_weights_yolov5s6_384_ti_lite_32p8_51p2_onnx.tar.gz
http://software-dl.ti.com/jacinto7/esd/modelzoo/gplv3/08_01_00_05/edgeai-yolov5/pretrained_models/modelartifacts/8bits/od-8110_onnxrt_weights_yolov5s6_384_ti_lite_32p8_51p2_onnx.tar.gz
Original file line number Diff line number Diff line change
@@ -1 +1 @@
http://software-dl.ti.com/jacinto7/esd/modelzoo/gplv3/08_00_00_05/edgeai-yolov5/pretrained_models/modelartifacts/8bits/od-8120_onnxrt_weights_yolov5m6_640_ti_lite_44p1_62p9_onnx.tar.gz
http://software-dl.ti.com/jacinto7/esd/modelzoo/gplv3/08_01_00_05/edgeai-yolov5/pretrained_models/modelartifacts/8bits/od-8120_onnxrt_weights_yolov5m6_640_ti_lite_44p1_62p9_onnx.tar.gz
Original file line number Diff line number Diff line change
@@ -1 +1 @@
http://software-dl.ti.com/jacinto7/esd/modelzoo/gplv3/08_00_00_05/edgeai-yolov5/pretrained_models/modelartifacts/8bits/od-8130_onnxrt_weights_yolov5l6_640_ti_lite_47p1_65p6_onnx.tar.gz
http://software-dl.ti.com/jacinto7/esd/modelzoo/gplv3/08_01_00_05/edgeai-yolov5/pretrained_models/modelartifacts/8bits/od-8130_onnxrt_weights_yolov5l6_640_ti_lite_47p1_65p6_onnx.tar.gz
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
http://software-dl.ti.com/jacinto7/esd/modelzoo/gplv3/08_01_00_05/edgeai-yolov5/pretrained_models/modelartifacts/8bits/od-8140_onnxrt_edgeai-yolox_yolox-s-ti-lite_39p1_57p9_onnx.tar.gz

0 comments on commit f1b50dc

Please sign in to comment.