Releases: pytorch/vision
Releases · pytorch/vision
More models and some bug fixes
- Ability to switch image backends between PIL and accimage
- Added more tests
- Various bug fixes and doc improvements
Models
- Fix for inception v3 input transform bug #144
- Added pretrained VGG models with batch norm
Datasets
- Fix indexing bug in LSUN dataset (#177)
- enable
~
to be used in dataset paths ImageFolder
now returns the same (sorted) file order on different machines (#193)
Transforms
- transforms.Scale now accepts a tuple as new size or single integer
Utils
- can now pass a pad value to make_grid and save_image
More models and datasets. Some bugfixes
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 exposenum_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
andsave_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
- Having files like
- LSUN test set loads now
Just a version bump
A small release, just needed a version bump because of PyPI.
Add models and modelzoo, some bugfixes
New Features
- Add
torchvision.models
: Definitions and pre-trained models for common vision models- ResNet, AlexNet, VGG models added with downloadable pre-trained weights
- adding padding to RandomCrop. Also add
transforms.Pad
- Add MNIST dataset
Performance Fixes
- Fixing performance of LSUN Dataset
Bug Fixes
- Some Python3 fixes
- Bug fixes in save_image, add single channel support
First release
Introduced Datasets and Transforms.
Added common datasets
-
COCO (Captioning and Detection)
-
LSUN Classification
-
ImageFolder
-
Imagenet-12
-
CIFAR10 and CIFAR100
-
Added utilities for saving images from Tensors.