search

bigmb / Unet-Segmentation-Pytorch-Nest-of-Unets

Implementation of different kinds of Unet Models for Image Segmentation - Unet , RCNN-Unet, Attention Unet, RCNN-Attention Unet, Nested Unet
MIT License
1.87k stars 345 forks source link

Single Channel Input #10

Closed vidalmaxime closed 4 years ago

vidalmaxime commented 5 years ago

Which parameters do you have to change if you are working with 1-channel images instead of 3 ?

bigmb commented 5 years ago

In this line, change 3 to 1. model_test = model_unet(model_Inputs[0], 3, 1) ---> model_test = model_unet(model_Inputs[0], 1, 1)

You might have to change 3 to 1 in some places where 3 channel images are required.