Simple-DETR-Implementation

This repo offers simple implementation of DETR (DEtection TRansformer) from scratch using PyTorch. Please refer to the implmentation details here.

Guidance

Please follow the insturction below.

git clone https://github.com/bskkimm/Simple-DETR-Implementation.git
conda create -n DETR python=3.10 -y
conda activate DETR
pip install -r requirements.txt

Then, implement DETR step by step using tutorial.ipynb

Result

Predicted Bounding boxes

Bounding boxs and correponding object queries