DeepTrial / Retina-VesselNet

A Simple U-net model for Retinal Blood Vessel Segmentation based on tensorflow2
303 stars 76 forks source link

Pretrained model does not achieve as high AUC as stated #4

Closed jzkay12 closed 6 years ago

jzkay12 commented 6 years ago

Hi,

I tried your pretrained model, and re-trained the model my myself, but the final AUC is only about 0.71. Can you check what caused the difference?

Thanks

DeepTrial commented 6 years ago

What dataset do you use? DRIVE dataset? Can you give me more detials? I will try to help you:)

jzkay12 commented 6 years ago

I used DRIVE dataset, downloaded with the link in your document. Thanks a lot.

DeepTrial commented 6 years ago

Ok,I will try to find out the problem.

DeepTrial commented 6 years ago

@Ralphyan Sorry,I find that I upload the wrong version that use a small unet network.Now I have fixed it, you can reclone it and test again.