likyoo / change_detection.pytorch

Deep learning models for change detection of remote sensing images
MIT License
331 stars 59 forks source link

Question about the pretrained model #8

Closed l912237463 closed 2 years ago

l912237463 commented 2 years ago

Since the number of data channels is not 3, I cannot use the pre-trained model. How can I modify the code

likyoo commented 2 years ago

You just need to change in_channels in your model. For using pretrained model, refer here.