Skip to content

Latest commit

 

History

History
27 lines (17 loc) · 915 Bytes

File metadata and controls

27 lines (17 loc) · 915 Bytes

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

image

Bounding boxs and correponding object queries

image