Commit Graph

2 Commits (main)

Author SHA1 Message Date
Nioolek e966ce4e60
[Improve] Beauty RTMDet config (#531)
* beauty rtmdet config

* format
2023-02-09 19:08:21 +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