rtdetr-pytorch
rtdetr-pytorch copied to clipboard
pranavdurai10
This repository provides a PyTorch implementation of RT-DeTR, a state-of-the-art Realtime Detection Transformer for object detection tasks.
Thanks for sharing! I'd like to do some research based on this library. Compared with the original paper, what modules do we need to complete this model, in addition to...
AIFI layer implementation `detr_layers/aifi.py` needs to be validated with actual architecture. Please feel free to submit a PR!
Dino head needs to be built for both L and X models in `models` directory. Please feel free to submit a PR!
HGBlock implementation `detr_layers/hgblock.py` needs to be validated with actual architecture. Please feel free to submit a PR!
HGStem implementation `detr_layers/hgstem.py` needs to be validated with actual architecture. Please feel free to submit a PR!
DWConv implementation `detr_layers/dwconv.py` needs to be validated with actual architecture. Please feel free to submit a PR!
RepC3 layer implementation `detr_layers/repc3.py` needs to be validated with actual architecture. Please feel free to submit a PR!
Metadata
Owner
Metadata
This repository provides a PyTorch implementation of RT-DeTR, a state-of-the-art Realtime Detection Transformer for object detection tasks.