rachellea/pytorch-computer-vision

PyTorch tutorial for computer vision

pytorch-computer-vision

This repository contains code for two PyTorch tutorials available on my Glass Box blog.

Tutorial links:

Designing Custom 2D and 3D CNNs in PyTorch: Tutorial with Code

Building Custom Image Datasets in PyTorch: Tutorial with Code