search

bcmi / DIRL-Inharmonious-Region-Localization

[ICME2021]The first work on Deep Inharmonious Region Localization, which can help image harmonization in an adversarial way.
24 stars 1 forks source link

Unable to change backbone when training #4

Closed alishan2040 closed 1 year ago

alishan2040 commented 2 years ago

Hello @jimleungjing,

Thanks for providing the implementation. I'm trying to re-train DIRL network by changing the backbone from Resnet34 to Resnet50 but it didn't work. Could you please comment on this on how to change the backbone. It seems we have to make some changes in the code in order for it to work?

image

Thanks

jimleungjing commented 2 years ago

It seems that the input tensor doesn't match the convolution weight along the channel dimension, and you should check the output channel of each resnet50 layer.