search

Onr / Council-GAN

Implementation for our paper Breaking the Cycle - Colleagues are all you need (CVPR 2020)
Other
264 stars 32 forks source link

Focus_loss ? #12

Closed roger-cv closed 3 years ago

roger-cv commented 3 years ago

Hey, thanks for the code.

What part of the code is Focus_loss in? Would you please explain "the value of the 4th channel for pixel k"?

Onr commented 3 years ago

Hey Roger the Focus_loss is in the trainer_council.py search for "focus_loss" and you will find all of its parts. you can also look in networks.py at Decoder_V2_atten to see how the masks is being used. here is an illustration of how the 4th channel is used. mask

roger-cv commented 3 years ago

Thanks for you reply.