/rtdetr-pytorch

This repository provides a PyTorch implementation of RT-DeTR, a state-of-the-art Realtime Detection Transformer for object detection tasks.

Primary LanguagePythonMIT LicenseMIT

Realtime Detection Transformer using PyTorch

License

Description

This repository contains a PyTorch implementation of the cutting-edge RT-DeTR (Realtime Detection Transformer).

Contents

The repository currently includes implementations of two RT-DeTR models:

  1. RT-DeTR-L
  2. RT-DeTR-X

Note

This is a reference implementation of RT-DeTR model using PyTorch. At the time of creating this repository, RT-DeTR is officially available only through the PaddlePaddle framework.

Kindly use this as a starting point to implement a fully-functional RT-DeTR model, using PyTorch framework.

License

This repository is licensed under the MIT License, allowing you to freely use and modify the code.