Live Online Classes in Design, Coding & AI — Small Classes, Free Retakes
Learn AI, Data Science & Business — Earn Certificates That Get You Hired
Overview
Google, IBM & Meta Certificates – 40% Off
One Coursera Plus subscription covers most Professional Certificates on Coursera.
Unlock All Certificates
This tutorial shows how to train a DETR object-detection model on a custom dataset. It covers COCO dataset preparation, PyTorch data loaders, a PyTorch Lightning training module, inference, and evaluation.
Syllabus
Introduction
Setting up the Python environment
DETR model inference on example images
Download custom dataset from Roboflow Universe
Building custom PyTorch COCO Detection datasets
Visualising COCO datasets entry
Building custom PyTorch Data Loaders
Building custom PyTorch Lightning DETR Module
Training DETR on custom dataset
Custom DETR model inference
Evaluating custom DETR model
Outro
Taught by
Roboflow
Reviews
4.0 rating, based on 1 Class Central review
Showing Class Central Sort
-
A great introduction on how to use DETR on custom dataset.
But could have given a more thorough description of the internal workings or components of the model.