/Detection-Transformer

Custom object detection Using Detection Transformer (DETR)

Primary LanguageJupyter Notebook

Detection-Transformer

Custom object detection Using Detection Transformer (DETR) .

To understand the code, check this video: https://youtu.be/xuh37qziXnw

Requirements:

!python -m pip install --upgrade pip

!pip install supervision==0.3.0

!pip install transformers

!pip install pytorch-lightning

!pip install timm

!pip install cython

!pip install pycocotools

!pip install scipy

image

image image

Reference: https://github.com/NielsRogge/Transformers-Tutorials/blob/master/DETR/Fine_tuning_DetrForObjectDetection_on_custom_dataset_(balloon).ipynb

Original DETR paper: https://arxiv.org/abs/2005.12872

Original DETR repo: https://github.com/facebookresearch/detr