pytorch_datasets

Repo to keep datasets organized in one location.

Also for keeping loading functions for images, videos, etc.

Installation

Install into a virtual environment for ease of use.

workon <virtualenv name>
cd pytorch_datasets
pip install -e .

Getting Started

Once installed, can import any dataset like so:

import pytorch_datasets
dset = pytorch_datasets.EPFL()
print(dset.__getitem__(0))

Datasets in repo

EPFL (Multi-View Car Dataset)

• 20 sequences of cars as they rotate by 360 degrees. 2299 images total.
• Website, Paper

WCVP (Weizmann Cars ViewPoint)

• Images circling around 22 cars outside. 1530 images total.
• Website, Paper

ObjectNet3D

• 3D object locations and orientation in images. 100 object categories, 90,127 images, 201,888 objects total in these images.
• Website, Paper

MISTIC-SL

• RGB videos of surgeons performing a running suture on a phantom.
• Website

JIGSAWS

• RGB videos + kinematics of eight surgeons with different levels of skill performing five repetitions of three elementary surgical tasks on a bench-top model.
• Website, Paper