Usage on a custom dataset

LiJunnan1992 / DivideMix

Code for paper: DivideMix: Learning with Noisy Labels as Semi-supervised Learning

MIT License

529 stars 83 forks source link

Usage on a custom dataset #7

Closed Rishu31 closed 4 years ago

Rishu31 commented 4 years ago

I intend to use this repo on a custom dataset of fashion images with weak labels (stored in a CSV file, row wise).. can you suggest which files would be the best choice to edit and use?

LiJunnan1992 commented 4 years ago

Hi! You can modify Train_clothing1M with your custom dataloader. Thanks!

Rishu31 commented 4 years ago

Thanks!