Train a custom object detection model on your own dataset using DETR as a clean, pipeline-free PyTorch starting point.
Run panoptic segmentation on images using the pretrained DETR models included with the repo.
Experiment with Transformer-based vision architectures by studying DETR's roughly 50-line inference implementation.
Benchmark DETR against traditional detectors on the COCO dataset using the provided training scripts.
| facebookresearch/detr | pre-commit/pre-commit | encode/httpx | |
|---|---|---|---|
| Stars | 15,266 | 15,267 | 15,264 |
| Language | Python | Python | Python |
| Setup difficulty | hard | easy | easy |
| Complexity | 4/5 | 2/5 | 2/5 |
| Audience | researcher | developer | developer |
Figures from each repo's GitHub metadata at analysis time.
Requires a GPU and a conda environment, pretrained model weights must be downloaded separately before running inference.
DETR, short for Detection Transformer, is research code from Facebook AI for end-to-end object detection. Object detection is the computer vision task of looking at an image and not just saying what is in it, but also drawing a box around each thing and labelling it (a dog here, a car there). The README explains that DETR replaces the usual hand-crafted detection pipeline (lots of stages and special-purpose tricks) with a single Transformer model that produces the whole set of boxes in one shot. The README describes how it works in plain terms: DETR treats detection as a direct set prediction problem. A small fixed set of learned object queries are fed through a Transformer encoder-decoder, and the network reasons about how the objects relate to each other and to the global image content, then outputs the final predictions in parallel. A set-based global loss using bipartite matching makes sure each ground-truth object is matched with exactly one prediction during training. The repo states that this approach matches a well-known baseline detector on the COCO benchmark while using half the computation. It also notes that the inference logic can be written in about 50 lines of code. You would use DETR if you are training or experimenting with object detection or panoptic segmentation models and want a clean, library-free starting point. The code is written in PyTorch (a Python deep learning framework named in the README), is installed via conda, and ships with pretrained models, Colab notebooks, and an optional Detectron2 wrapper. The full README is longer than what was provided.
Research code from Facebook AI for end-to-end object detection using Transformers, detects objects and draws bounding boxes in one shot without a multi-stage pipeline, matching top detectors at half the compute.
Mainly Python. The stack also includes Python, PyTorch, conda.
Not specified in the explanation.
Setup difficulty is rated hard, with roughly 1h+ to a first successful run.
Mainly researcher.
This repo across BitVibe Labs
Verify against the repo before relying on details.