pytorch/vision

toy video datasets for generative models

Open

#2 676 ouverte le 14 sept. 2020

Voir sur GitHub
 (4 commentaires) (0 réactions) (0 assignés)Python (15 050 stars) (6 858 forks)batch import
help wantedmodule: datasetsnew feature

Description

🚀 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

Guide contributeur

Stack technique
python
Domaine
datamachine learning
Type d'issue
feature
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
3
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
half day
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
stale
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
clear
Prérequis
PyTorch Dataset APIVideo data loading
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
60
Direction de recherche
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.