DebuggerCafe - Deep Learning, Machine Learning, Artificial Intelligence

Human Action Recognition using 2D CNN with PyTorch

Human Action Recognition using a 2D CNN, specifically, fine tuning a pretrained ResNet50 model on a dataset with 15 action classes. ...

Using Custom Backbone for PyTorch SSD for Object Detection

Sovit Ranjan Rath June 5, 2023 52 Comments

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. ...