Commit Graph

11 Commits (2813e89f44a67d87fa79f7b363926b0eaabf7dd6)

Author SHA1 Message Date
Range King a631370cb3
Update model graphs (#442) 2023-01-07 17:44:19 +08:00
Haian Huang(深度眸) 884330108d
[BC Breaking] Keep the same input shape order as mmdet 3.0.0rc5 (#436)
* Keep the same input shape order as mmdet

* add desc

* update

* update
2023-01-06 19:00:30 +08:00
RangiLyu 5172805dae
fix rtmdet-s cfg and neck (#434) 2023-01-06 11:40:19 +08:00
RangiLyu 48f8896e84
[Feature] Implement fast version of RTMDet. (#425)
* Accelerate RTMDet

* update

* update

* update

* update1

* update2

* update pipeline

* update lr cudnnbenchmark

* revert batchsize

* fix batch inference

* refactor head

* update box

* bs=16

* update

* move reduce mean

* update head

* per img loss

* fix

* fix sum

* concat loss

* batch dsla

* sort topk

* bs 32

* clean code

* update readme

* update ut

* update checkpoint

* num_class

* clean code

* resolve comments

* fix readme

* fix ut

Co-authored-by: huanghaian <huanghaian@sensetime.com>
Co-authored-by: hha <1286304229@qq.com>
2023-01-04 16:06:44 +08:00
Xiaolin Wang 71e5c793e5 [Doc] Fix typo (#270)
* fix typo

* fix ci
2022-11-10 11:12:31 +08:00
Haian Huang(深度眸) 8fc8066b1b Add changelog of v0.1.2 (#226)
* Add changelog of v0.1.2

* update version

* fix comments

* fix comments

* update

* update version

* update version
2022-11-03 19:03:06 +08:00
Haian Huang(深度眸) b95bf63916 [Feature] Support P6 YOLOv5 (#168)
* add yolov5s config

* support train

* fix some comment

* add config

* add link

* UPDATE

* UPDATE

* remove
2022-11-03 19:03:06 +08:00
bin ze 15e0cfa4b5 [Fix] Update metafile for RTMDet (#188)
* update metafile

* update metafile

* update metafile
2022-11-03 19:03:06 +08:00
Haian Huang(深度眸) b18d5cf586 [Fix] Remove Albu version limit (#187)
* remove albu version limit

* fix lint
2022-11-03 19:03:06 +08:00
Haian Huang(深度眸) 3ea6edfeac Fix visualization bug in test_pipeline (#166) 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