search

likyoo / change_detection.pytorch

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

Can you train STANet? When I train it,Imeet a error. #3

Closed wangjin1234567891 closed 3 years ago

wangjin1234567891 commented 3 years ago

image

likyoo commented 3 years ago

@LinoSun

LinoSun commented 3 years ago

Could plz show more code.

wangjin1234567891 commented 3 years ago

I trian Unet and it's ok .I just change the model as following picture shows image

LinoSun commented 3 years ago

We still need debug STANet later.If you want use it now. You can try to use BCLLoss or other loss function for STANet because of STANet model only forward one channel distance map.

wangjin1234567891 commented 3 years ago

Ok, thans a lot.