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.

[alirezajavady]

As you can see, data_gen_arg is used in the trainGenerator function, which is defined in the data.py file. In lines 56 and 57 of the data.py file, the augmentation dictionary is used to augment the images and masks.