DebuggerCafe - Deep Learning, Machine Learning, Artificial Intelligence

Train PyTorch RetinaNet on Custom Dataset

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 61 Comments

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

Steel Surface Defect Detection using Object Detection

Sovit Ranjan Rath May 1, 2023 2 Comments

In this article, we train several Faster RCNN object detection models for to detect defects on the steel surface. ...

Disaster Tweet Classification using PyTorch

Sovit Ranjan Rath April 24, 2023 2 Comments

In this article, we train a simple embedding language model on disaster tweet classification dataset and then run inference on the test set. ...

Getting Started with Text Classification using Pytorch, NLP, and Deep Learning

Sovit Ranjan Rath April 17, 2023 9 Comments

In this article, we follow a code-first approach to text classification using PyTorch, NLP, and Deep Learning. ...