D-X-Y / AutoDL-Projects

Automated deep learning algorithms implemented in PyTorch.
MIT License
1.56k stars 281 forks source link

Custom training data #7

Closed ghost closed 5 years ago

ghost commented 5 years ago

@D-X-Y thank you for your hard work

How can I train on my own dataset, with a simple structure of:

D-X-Y commented 5 years ago

I define the dataset and data augmentation function in https://github.com/D-X-Y/GDAS/blob/master/lib/datasets/get_dataset_with_transform.py#L38 . You can add your own dataset in this function.

D-X-Y commented 5 years ago

Feel free to reopen it as necessary.