Using Predefined and Pretrained CNNs in PyTorch

Tutorial with Code

This is a photo of Habitat 67 from Wikimedia Commons (license: CC)

At the end of this tutorial you should be able to:

• Load randomly initialized or pre-trained CNNs with PyTorch torchvision.models (ResNet, VGG, etc.)
• Select out only part of a pre-trained CNN, e.g. only the convolutional feature extractor
• Automatically calculate the number of…