/image_processing_with_pytorch

Deep learning architectures for image processing using the COCO database.

Primary LanguageJupyter Notebook

Image Processing with Pytorch

This repository contains code for the following:

  1. Deep learning architectures for image classification in Pytorch
  2. Deep learning architectures for semantic segmentation in Pytorch
  3. Python class for reading COCO images with their API and generating batches for deep learining algorithm training