search

biomag-lab / hypocotyl-UNet

MIT License
11 stars 2 forks source link

Image augmentation #5

Open darvida opened 4 years ago

darvida commented 4 years ago

What kind of image augmentation did you use for the model that was trained on the Arabidopsis seedlings?

cosmic-cortex commented 4 years ago

Hi!

Sorry for the late reply, there was an issue with my GitHub email notifications, they are working now. I have used the same augmentation which can be found in the train.py: https://github.com/biomag-lab/hypocotyl-UNet/blob/master/src/train.py#L39-L41

However, augmentation almost always needs to be changed from dataset to dataset. This required the most experimentation from our part, it took me a while to find the augmentation which was working well for our images.