decouples / Unet

A demo of Unet to detect edges!
106 stars 74 forks source link

If the training images are in RGB channels #1

Open rangerli opened 6 years ago

rangerli commented 6 years ago

I'm glad to see this project. In my dataset, the images are in RGB channels (.tiff), what should i do to change the code adapting my dataset.

decouples commented 6 years ago

The label needs to be one channel grayscale image. It is not important whether the suffix is tiff or not. The input picture for training can be RGB. You only need to change all train-related tiffs in your program to the format of your picture.