search

zhixuhao / unet

unet for image segmentation
MIT License
4.6k stars 2k forks source link

Does your code use the aug dataset? #237

Open tphankr opened 2 years ago

tphankr commented 2 years ago

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. image Thanks.

alirezajavady commented 1 month ago

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.