MrGiovanni / UNetPlusPlus

[IEEE TMI] Official Implementation for UNet++
Other
2.26k stars 538 forks source link

How can I modify the network code for the gray-scale image patches? #35

Open beargolden opened 4 years ago

beargolden commented 4 years ago

Hello, Zongwei, Thank you for your wonderful UNetPlusPlus project. I understand that the network expects input channels of 3. But if I would input a gray-scale image patch into the network, how should I modify the network code? Could you give me some advice? Thanks again and Merry Christmas!

Fairydetail commented 4 years ago

In my experience, you can simply assign the three channels to the same value.

beargolden commented 4 years ago

In my experience, you can simply assign the three channels to the same value.

Thanks, I made it, and created the network model by myself. Your solution should be working, but I have never tried.