DETR PyTorch Implementation | DETR Tutorial Part 2
In this video, we dive into the implementation of DETR (DEtection TRansformer) for object detection using PyTorch. This is Part 2 of the DETR tutorial, where using our understanding from Part 1 Video, we get to actually building and training the DETR model and see the an implementation of end-to-end object detection with transformers. We also visualize the model attention maps after training it on voc dataset. By looking through the voc dataset training code, one should get a sense of how to train DETR model on your own custom dataset and even implement your own detr from scratch.
The detr tu…
Watch on YouTube ↗
(saves to browser)
Chapters (11)
Intro
1:03
DETR Explanation Recap
7:33
Detection Transformer PyTorch Module Initialisation
17:05
Generation Prediction from DETR Layers
26:57
Matching Predictions and Target in DETR
37:52
DETR Loss Implementation
45:06
DETR Inference Code
46:57
Training DETR on VOC
52:24
Results of DETR Model Training
1:00:27
Visualisations of DETR Attention Maps
1:05:02
Outro
DeepCamp AI