The datasets
folder is used to contain dataset code or environment code.
Don't store actual data like images here! For those, please use the data
folder instead of datasets
.
Create a folder to create your own pytorch dataset definition. Then, update the __init__.py
at every level to register all datasets.
Each dataset class takes in a DictConfig file cfg
in its __init__
, which allows you to pass in arguments via configuration file in configurations/dataset
or command line override.
This repo is forked from Boyuan Chen's research template repo. By its MIT license, you must keep the above sentence in README.md
and the LICENSE
file to credit the author.