jakeret / tf_unet

Generic U-Net Tensorflow implementation for image segmentation
GNU General Public License v3.0
1.9k stars 748 forks source link

Very confused about the result of prediction! #214

Open sanersbug opened 6 years ago

sanersbug commented 6 years ago

I have two class include water and background ,and my masks include 0 and 1. but the result has three class include 0,1,2???? what's the problem? any one meet it? Thanks a lot! result_

jakeret commented 6 years ago

What is the value of n_class that you pass into the unet?

sanersbug commented 6 years ago

@jakeret I am sure the n_class is 2