Training issue

[chiawenhuanghuang]

Thank you for providing the code, you did a great work. However, I am still a bit confused after reading the README, as it only mentions the training method for Segmentation. I want to ask, in the code, are both the classification branch and segmentation branch trained using train_single.py? train_single.py seems like it's for segmentation training, so how is the classification part trained if I use my own dataset?

[yuhuan-wu]

Thanks a lot for your enquiry. train_single.py is to train the segmentation branch. During the joint training stage, you could load the trained weights of segmentation branch (pretrained on segmentation maps) and the classification branch (pretrained on image-level maps) separately, then conducting the joint training (finetuning).

[chiawenhuanghuang]

Thank you for your response. I need assistance with two issues:

1. How can I train a classification branch using techniques like image mixing and pixel-level supervision for activation mapping, as mentioned in the paper?
2. Is the code for this specific part, especially for training a classification branch on a custom dataset, available in this GitHub repository?

[yuhuan-wu]

We have no plans to release the entire code of training the classification branch currently.
However, you can reimplement the training process as all details (process, hyper-params) have been explained clearly in our paper.
Thank you for your understanding.

[chiawenhuanghuang]

Thanks for your clarification. I appreciate your great work. Have a nice day!