Nota-NetsPresso / netspresso-trainer

A library for training, compressing and deploying computer vision models (including ViT) with edge devices
https://nota-netspresso.github.io/netspresso-trainer/
Apache License 2.0
62 stars 6 forks source link

[Sprint] Detection head re-implementation for PyNetsPresso compatibility #177

Closed illian01 closed 1 year ago

illian01 commented 1 year ago

Description

Checklists

Common

TODO

illian01 commented 1 year ago

184

There are two issues to solve.

illian01 commented 1 year ago

194

To complete this issue, we have to solve these issues.

illian01 commented 1 year ago

We decided followings through internal discussion:

Thus, this issue will be closed after refactoring 2-stage detector, and re-open when we need TensorRT of the model.