diff --git a/README.md b/README.md index 79b4773fafc1..b9a090c0bf42 100755 --- a/README.md +++ b/README.md @@ -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) |
@@ -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** diff --git a/pretrained_models/modelartifacts/8bits/artifacts.list b/pretrained_models/modelartifacts/8bits/artifacts.list index 62ad30557caa..2642fecd0f0a 100644 --- a/pretrained_models/modelartifacts/8bits/artifacts.list +++ b/pretrained_models/modelartifacts/8bits/artifacts.list @@ -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 \ No newline at end of file +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 \ No newline at end of file diff --git a/pretrained_models/modelartifacts/8bits/artifacts.yaml b/pretrained_models/modelartifacts/8bits/artifacts.yaml index 367aceba20ae..23f803e4ed42 100644 --- a/pretrained_models/modelartifacts/8bits/artifacts.yaml +++ b/pretrained_models/modelartifacts/8bits/artifacts.yaml @@ -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 diff --git a/pretrained_models/modelartifacts/8bits/od-8100_onnxrt_weights_yolov5s6_640_ti_lite_37p4_56p0_onnx.tar.gz.link b/pretrained_models/modelartifacts/8bits/od-8100_onnxrt_weights_yolov5s6_640_ti_lite_37p4_56p0_onnx.tar.gz.link index f0e8370df809..804f6a8b8479 100644 --- a/pretrained_models/modelartifacts/8bits/od-8100_onnxrt_weights_yolov5s6_640_ti_lite_37p4_56p0_onnx.tar.gz.link +++ b/pretrained_models/modelartifacts/8bits/od-8100_onnxrt_weights_yolov5s6_640_ti_lite_37p4_56p0_onnx.tar.gz.link @@ -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 diff --git a/pretrained_models/modelartifacts/8bits/od-8110_onnxrt_weights_yolov5s6_384_ti_lite_32p8_51p2_onnx.tar.gz.link b/pretrained_models/modelartifacts/8bits/od-8110_onnxrt_weights_yolov5s6_384_ti_lite_32p8_51p2_onnx.tar.gz.link index a9a8b1e03b0b..46e35dd7a228 100644 --- a/pretrained_models/modelartifacts/8bits/od-8110_onnxrt_weights_yolov5s6_384_ti_lite_32p8_51p2_onnx.tar.gz.link +++ b/pretrained_models/modelartifacts/8bits/od-8110_onnxrt_weights_yolov5s6_384_ti_lite_32p8_51p2_onnx.tar.gz.link @@ -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 diff --git a/pretrained_models/modelartifacts/8bits/od-8120_onnxrt_weights_yolov5m6_640_ti_lite_44p1_62p9_onnx.tar.gz.link b/pretrained_models/modelartifacts/8bits/od-8120_onnxrt_weights_yolov5m6_640_ti_lite_44p1_62p9_onnx.tar.gz.link index d9261621adb6..6d37b2f9563d 100644 --- a/pretrained_models/modelartifacts/8bits/od-8120_onnxrt_weights_yolov5m6_640_ti_lite_44p1_62p9_onnx.tar.gz.link +++ b/pretrained_models/modelartifacts/8bits/od-8120_onnxrt_weights_yolov5m6_640_ti_lite_44p1_62p9_onnx.tar.gz.link @@ -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 diff --git a/pretrained_models/modelartifacts/8bits/od-8130_onnxrt_weights_yolov5l6_640_ti_lite_47p1_65p6_onnx.tar.gz.link b/pretrained_models/modelartifacts/8bits/od-8130_onnxrt_weights_yolov5l6_640_ti_lite_47p1_65p6_onnx.tar.gz.link index 0c4ccd312466..28ed1be00ce1 100644 --- a/pretrained_models/modelartifacts/8bits/od-8130_onnxrt_weights_yolov5l6_640_ti_lite_47p1_65p6_onnx.tar.gz.link +++ b/pretrained_models/modelartifacts/8bits/od-8130_onnxrt_weights_yolov5l6_640_ti_lite_47p1_65p6_onnx.tar.gz.link @@ -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 diff --git a/pretrained_models/modelartifacts/8bits/od-8140_onnxrt_edgeai-yolox_yolox-s-ti-lite_39p1_57p9_onnx.tar.gz.link b/pretrained_models/modelartifacts/8bits/od-8140_onnxrt_edgeai-yolox_yolox-s-ti-lite_39p1_57p9_onnx.tar.gz.link new file mode 100644 index 000000000000..9606bf8da067 --- /dev/null +++ b/pretrained_models/modelartifacts/8bits/od-8140_onnxrt_edgeai-yolox_yolox-s-ti-lite_39p1_57p9_onnx.tar.gz.link @@ -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