4 Commits

Author SHA1 Message Date
Nioolek
5708864d5b [Enhance] YOLOv6 model convert (#165)
* add yolov5 voc training

* fix bug

* fix bug

* fix yolov6 model convert

* update yolov5 modelconvert

* rollback yolov5 model convert
2022-11-03 19:03:06 +08:00
Haian Huang(深度眸)
95f60784a5 [Feature] Support YOLOv7 inference (#149)
* update

* update

* update

* update

* update

* add docstr

* fix comments

* update
2022-11-03 19:03:06 +08:00
Haian Huang(深度眸)
bb3aa48689 [Feature] Support RTMDet (#85)
* add cspnext

* add neck


* update

* align s,tiny test

* update

* update

* fix bug of UT

* fix typehint

* fix resize bug

* update s and tiny

* update s and tiny link

* update convert

* add cls config with rtmdet

* update config


* fix ci error

Co-authored-by: wanghonglie <wanghonglie@pjlab.org.cn>
2022-09-29 19:09:56 +08:00
wanghonglie
b3d405aa4c
add model convert (#8) 2022-09-18 11:12:08 +08:00