search

chenxi116 / DeepLabv3.pytorch

PyTorch implementation of DeepLabv3
Other
253 stars 47 forks source link

Modify your network to train on Single channel Image #12

Open ghost opened 4 years ago

ghost commented 4 years ago

I'm trying to replicate your model, but I have a single-channel image as an input. So can you please help me on how to achieve it

chenxi116 commented 4 years ago

Sounds like the main change is replacing the first parameter "3" in https://github.com/chenxi116/DeepLabv3.pytorch/blob/master/deeplab.py#L144 and https://github.com/chenxi116/DeepLabv3.pytorch/blob/master/deeplab.py#L148 to "1".