DebuggerCafe - Deep Learning, Machine Learning, Artificial Intelligence

Train S3D Video Classification Model using PyTorch

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

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