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

Learn how to train Object Detection Transformers using DETR on custom datasets in this comprehensive tutorial. Set up a Python environment, perform DETR model inference on example images, and download custom datasets from Roboflow Universe. Build custom PyTorch COCO Detection datasets, visualize COCO dataset entries, and create custom PyTorch data loaders. Construct a custom PyTorch Lightning DETR module for training your own object detection model. Follow step-by-step instructions for training DETR on your custom dataset, performing custom DETR model inference, and evaluating the model's performance. Gain practical insights into implementing advanced object detection techniques using transformers and custom datasets.

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.