Open genghisun opened 4 years ago
It is already commented in the code (at line 30) and it is the fusion module mentioned in the paper to recombined the style codes of different parts
Edit: Sorry, I have mistaken your question. I think that LinearBlock
is the affine layer in AdaIN (you can see this in StyleGANv1) to match the dimension of the feature in the main branch
https://github.com/menyifang/ADGAN/blob/4dd70649ad136829b92dd6a1a823af7594a0220f/models/model_adgen.py#L37-L40
I noticed that you add an
LinearBlock
after obtain the style code by encoding every part of an image and its segment, It seems that this part is not mentioned in the paper. Can you explain why you do that?