Category: Computer Vision

Computer Vision in AI encompasses various tasks including classical computer vision, deep learning based image classification, image classification, object detection etc.

Saving and Loading the Best Model in PyTorch

Learn about saving and loading the best model in PyTorch and how running tests on the best model gives better deep learning results. ...

Satellite Image Classification using PyTorch ResNet34

Sovit Ranjan Rath November 1, 2021 4 Comments

This tutorial explores satellite image classification using deep learning and PyTorch ResNet34 model. ...

Custom Object Detection using PyTorch Faster RCNN

Sovit Ranjan Rath October 25, 2021 62 Comments

In this tutorial, we explore how to write a simple training pipeline for custom object detection using PyTorch Faster RCNN model. ...

Transfer Learning using PyTorch ShuffleNetV2

Sovit Ranjan Rath October 18, 2021 0 Comments

Learn how to carry out transfer learning using the PyTorch ShuffleNetV2 model pre-trained on the ImageNet dataset. ...

PyTorch ImageFolder for Training CNN Models

Sovit Ranjan Rath October 11, 2021 15 Comments

Learn how to use PyTorch ImageFolder class for easier training of CNN models. Train a CNN on a very interesting Butterfly images dataset. ...