pytorch/vision

toy video datasets for generative models

Open

Aperta il 14 set 2020

Vedi su GitHub
 (4 commenti) (0 reazioni) (0 assegnatari)Python (15.050 star) (6858 fork)batch import
help wantedmodule: datasetsnew feature

Descrizione

🚀 Feature

Adding toy video datasets for generative models such as Moving MNIST and Robonet

Motivation

The currently available video datasets are suitable for recognition and not synthesis. It would be nice if you add simple toy video datasets such as Moving MNIST.

cc @pmeier

Guida contributor

Tech stack
python
Dominio
datamachine learning
Tipo issue
feature
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
3
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
half day
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
stale
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
clear
Prerequisiti
PyTorch Dataset APIVideo data loading
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
60
Direzione di ricerca
Examine existing dataset implementations in torchvision/datasets/, especially the MNIST class. Implement a MovingMNIST class that generates synthetic videos of moving digits with random trajectories. Follow the pattern of other datasets, including proper registration in init .py and adding unit tests. For Robonet, explore the original dataset format and adapt it similarly. Consider reaching out to maintainers (cc @pmeier) for guidance on integration.