Add YOLOX object detection

laggui commented 8 months ago

Closes issue #21

This PR adds a Burn implementation for the YOLOX object detection models. Implementation and weights are based on the official implementation.

TODO

[x] YOLOX-Nano support
- Requires enum deserializer support to switch between regular conv blocks and depthwise separable conv blocks
[ ] ~NMS tensor implementation~
- ~Requires argsort~
- ~We should compare timings w/ the current naive implementation to see if it is worth it since we don't have the nms kernel to perform the computations even more efficiently~
I scratched the NMS tensor implementation. Not required at this time.

antimora commented 7 months ago

@laggui ready to merge?

laggui commented 7 months ago

@laggui ready to merge?

Actually I think we're waiting for the Burn release so we can use the crates version and announce some things.

I'll let @nathanielsimard decide when we merge this.

tracel-ai / models