rpmcruz / detr

A simple implementation of DETR for educational and practical purposes.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

DETR

A simple implementation of DETR (End-to-End Object Detection with Transformers) for educational and practical purposes.

The implementation seems pretty faithful to the paper and it seems to produce results in the right direction. One difference is that I use much smaller images. The loss has been pretty optimized, but maybe it could be optimized further (I process each element of the batch separately). When training with PASCAL VOC, each epoch takes a little over 3 minutes (GPU A100).

About

A simple implementation of DETR for educational and practical purposes.


Languages

Language:Python 100.0%