orobix / retina-unet

Retina blood vessel segmentation with a convolutional neural network

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

regarding the values of prediction

wenouyang opened this issue · comments

Hi, I am running this model against a given data set. For curiosity, I just test the prediction result using the model weights after finished the first 12 epochs of training.

The prediction shape is (841,4096,2). I found that the values of [:,:,0] are almost all 1 while [:,:,1] are all very small values. Is this a strong signal that the model has not been able to learn anything during the first very few epochs. Thanks.