Does your code use the aug dataset?

[tphankr]

Thank you for your code,
Please allows me ask, does your code use the aug dataset?
We ran finishly the prepare.ipynb file. At this time, we had the agument data in the "aug" folder. And in your main.py, the link data to the train folder ( label and image) not aug folder. I though, your main.py only use the original training dataset. Is it right? thanks.

Thanks.