Deep Learning Archives - Page 6 of 43

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

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

Human Action Recognition using 2D CNN with PyTorch

Sovit Ranjan Rath June 12, 2023 0 Comment

Human Action Recognition using a 2D CNN, specifically, fine tuning a pretrained ResNet50 model on a dataset with 15 action classes. ...

Using Custom Backbone for PyTorch SSD for Object Detection

Sovit Ranjan Rath June 5, 2023 52 Comments

In this article, we use a PyTorch SSD model with custom ResNet34 backbone and train it on a person detection dataset. ...

DETR for Object Detection

Sovit Ranjan Rath May 22, 2023 9 Comments

In this article, we explore the DETR model for object detection. Along with a short discussion of the model architecture, we also carry out inference on videos. ...

Category: Deep Learning

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

UAV Small Object Detection using Deep Learning and PyTorch

Human Action Recognition using 2D CNN with PyTorch

Using Custom Backbone for PyTorch SSD for Object Detection

DETR for Object Detection