qixiong-wang / mmsegmentation-mpcnet

Apache License 2.0
5 stars 0 forks source link

Question about the loss function #1

Open Daisy-7 opened 1 year ago

Daisy-7 commented 1 year ago

Thank you for releasing the code. It is a nice work.

Due to the lack of code comments, I cannot match the loss function in the code with the paper. Can you tell me which loss functions in the code represent the SPC and CPC losses in the paper?

qixiong-wang commented 1 year ago

Thanks for your interest in our work. We reorganized the code and put SPC and CPC losses into mmseg/models/decode_heads/mpc_loss.py, which is called in mmseg/models/decode_heads/decode_head_memory.py.