Class Central is learner-supported. When you buy through links on our site, we may earn an affiliate commission.

YouTube

How to Train Object Detection Transformer on Custom Dataset

Roboflow via YouTube

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

Start your review of How to Train Object Detection Transformer on Custom Dataset

  • Profile image for Avnish Panwar
    Avnish Panwar
    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.

Never Stop Learning.

Get personalized course recommendations, track subjects and courses with reminders, and more.

Someone learning on their laptop while sitting on the floor.