facebookresearch / SlowFast

PySlowFast: video understanding codebase from FAIR for reproducing state-of-the-art video models.
Apache License 2.0
6.56k stars 1.21k forks source link

how to make a image custom dataset #619

Open linglingl635 opened 1 year ago

linglingl635 commented 1 year ago

i follow the discussion about custom dataset on AVA dataset. but i wnat to make a image dataset like cifar. so how can i do it. what should i do to make cifar dataset could be trained?

alpargun commented 1 year ago

Hi, please follow the Charades data set guide in DATASET.md file (https://github.com/facebookresearch/SlowFast/blob/main/slowfast/datasets/DATASET.md) as it is also a data set with RGB images. To summarize (copying from DATASET.md):

Please download the Charades RGB frames from dataset provider.

Download the frame list from the following links: (train, val).

Please set DATA.PATH_TO_DATA_DIR to point to the folder containing the frame lists, and DATA.PATH_PREFIX to the folder containing RGB frames.

So you just need to organize your images according to this folder structure, and prepare frame lists yourself. Finally, set the correct paths using the parameters: _DATA.PATH_TO_DATADIR and _DATA.PATHPREFIX in your config file.