Skip to content

Releases: pytorch/vision

More models and some bug fixes

11 Sep 16:15
Compare
Choose a tag to compare
  • 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

03 Apr 04:53
Compare
Choose a tag to compare

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

Just a version bump

03 Apr 04:41
Compare
Choose a tag to compare

A small release, just needed a version bump because of PyPI.

Add models and modelzoo, some bugfixes

03 Apr 04:40
Compare
Choose a tag to compare

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

03 Apr 04:35
Compare
Choose a tag to compare
First release Pre-release
Pre-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.