PyTorch Archives - Page 4 of 33

Multi-Class Semantic Segmentation Training using PyTorch

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

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

Category: PyTorch

Multi-Class Semantic Segmentation Training using PyTorch

Train S3D Video Classification Model using PyTorch

Training a Video Classification Model from Torchvision

Traffic Light Detection Using RetinaNet and PyTorch

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