TorchVision

:fire: Computer vision datasets, transforms, and models for Ruby

This gem is currently experimental. There may be breaking changes between each release. Please report any issues you experience.

Installation

Add this line to your application’s Gemfile:

gem 'torchvision'

Getting Started

This library follows the Python API. Many methods and options are missing at the moment. PRs welcome!

Datasets

MNIST dataset

trainset = TorchVision::Datasets::MNIST.new("./data", train: true, download: true)
trainset.size

Disclaimer

This library downloads and prepares public datasets. We don’t host any datasets. Be sure to adhere to the license for each dataset.

If you’re a dataset owner and wish to update any details or remove it from this project, let us know.

History

View the changelog

Contributing

Everyone is encouraged to help improve this project. Here are a few ways you can help:

• Report bugs
• Fix bugs and submit pull requests
• Write, clarify, or fix documentation
• Suggest or add new features

To get started with development:

git clone https://github.com/ankane/torchvision.git
cd torchvision
bundle install
bundle exec rake test