DebuggerCafe - Deep Learning, Machine Learning, Artificial Intelligence

Fine-Tune Mask RCNN PyTorch on Custom Dataset

In this article, we fine-tune the PyTorch Mask RCNN instance segmentation model on a microcontroller segmentation dataset. ...

Multi-Class Semantic Segmentation Training using PyTorch

Sovit Ranjan Rath July 31, 2023 10 Comments

This article the training of PyTorch DeepLabV3 ResNet101 model on a multi-class semantic segmentation dataset. ...

Train S3D Video Classification Model using PyTorch

Sovit Ranjan Rath July 24, 2023 4 Comments

In this tutorial, we train the S3D video classification model on a workout recognition dataset and run inference on new videos. ...

Training a Video Classification Model from Torchvision

Sovit Ranjan Rath July 17, 2023 0 Comment

In this article, we will be training a video classification model from Torchvision on the UCF50 dataset and running inference on unseen videos from the internet. ...

Traffic Light Detection Using RetinaNet and PyTorch

Sovit Ranjan Rath July 10, 2023 2 Comments

This article explores the training of PyTorch RetinaNet model for traffic light detection in real-life scenes. ...