New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[ERROR] DeepLabV3+DenseNet Backbone #44

Open

stigma0617 opened this issue May 21, 2019 · 1 comment

stigma0617 commented May 21, 2019 •

edited

Loading

Hi,

I have a question when I try to train DeepLabV3 with DenseNet backbone.

the loss term of DeepLabV3.

I wonder why you use this auxiliary loss term. As far as I know, original DeepLabV3 was not trained by the auxiliary cross-entropy loss.

As your code about deeplabv3 was hard-coded, it should be fixed for other backbone (i.g., DenseNet).

Thanks

Owner

donnyyou commented May 21, 2019

Original DeepLabV3 maybe not be trained with auxiliary, and I haven't tested the influence of it.
The code is really hard-coded. You could simply fix it by replacing 2048 with self.backbone.num_features.
Welcome your pull request!!!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment