search

biomag-lab / hypocotyl-UNet

MIT License
11 stars 2 forks source link

Train on greyscale images ? #4

Open darvida opened 4 years ago

darvida commented 4 years ago

Hello i have a large dataset of arabodopsis greyscale images. Do you think that this model is able to be trained on greyscale images and perform well ?

darvida commented 4 years ago

108 00 s

cosmic-cortex commented 4 years ago

Hi! Sure, grayscale should work as well, although I haven't experimented with this too extensively. One thing you need to make sure is that in spite of being greyscale, your images are saved as RGB. In the training and prediction script, I assume that there are three channels. Of course you can modify the scripts to use a single channel grayscale image, but saving the images as RGB is simpler.

Also, you need to make sure that the plants do not overlap in the images. This algorithm is unable to handle overlaps properly, so you might get false results.