DebuggerCafe - Deep Learning, Machine Learning, Artificial Intelligence

Training a Video Classification Model from Torchvision

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

Export PyTorch Model to ONNX – Convert a Custom Detection Model to ONNX

Sovit Ranjan Rath July 3, 2023 2 Comments

In this article, we train a custom PyTorch RetinaNet model and export it to ONNX format. Further, we run inference using it on the CUDA device. ...

UAV Small Object Detection using Deep Learning and PyTorch

Sovit Ranjan Rath June 26, 2023 20 Comments

In this article, we use PyTorch Faster RCNN and fine-tune it for small object detection on UAV images. ...

Workout Recognition using CNN and Deep Learning

Sovit Ranjan Rath June 19, 2023 0 Comment

This article fine-tunes a pretrained ResNet50 CNN model for workout recognition in images and videos. ...