pytorch/vision v0.1.8

More models and datasets. Some bugfixes

on GitHub

New Features

Models

• SqueezeNet 1.0 and 1.1 models added, along with pre-trained weights
• Add pre-trained weights for VGG models
  • Fix location of dropout in VGG
• torchvision.models now expose num_classes as a constructor argument
• Add InceptionV3 model and pre-trained weights
• Add DenseNet models and pre-trained weights

Datasets

• Add STL10 dataset
• Add SVHN dataset
• Add PhotoTour dataset

Transforms and Utilities

• transforms.Pad now allows fill colors of either number tuples, or named colors like "white"
• add normalization options to make_grid and save_image
• ToTensor now supports more input types

Performance Improvements

Bug Fixes

• ToPILImage now supports a single image
• Python3 compatibility bug fixes
• ToTensor now copes with all PIL Image types, not just RGB images
• ImageFolder now only scans subdirectories.
  • Having files like .DS_Store is now no longer a blocking hindrance
  • Check for non-zero number of images in ImageFolder
  • Subdirectories of classes have recursive scans for images
• LSUN test set loads now