DebuggerCafe - Deep Learning, Machine Learning, Artificial Intelligence

Using Custom Backbone for PyTorch SSD for Object Detection

In this article, we use a PyTorch SSD model with custom ResNet34 backbone and train it on a person detection dataset. ...

Train DETR on Custom Dataset

Sovit Ranjan Rath May 29, 2023 2 Comments

In this article, we fine fine DETR (Detection Transformer) models on a custom aquarium dataset. After training, we also run inference on the test dataset and unseen videos. ...

DETR for Object Detection

Sovit Ranjan Rath May 22, 2023 9 Comments

In this article, we explore the DETR model for object detection. Along with a short discussion of the model architecture, we also carry out inference on videos. ...

Train PyTorch RetinaNet on Custom Dataset

Sovit Ranjan Rath May 15, 2023 29 Comments

In this article, we train the PyTorch RetinaNet model on a custom BCCD dataset. We go through the configuration of the RetinaNet model for custom training, train the model, and also carry out inference. ...

Train SSD300 VGG16 Model from Torchvision on Custom Dataset

Sovit Ranjan Rath May 8, 2023 59 Comments

In this article, we go through the training of SSD300 VGG16 from Torchvision on a custom license plate detection dataset. ...