Which file used for training?

[xuyifeng-nwpu]

Thank you share your codes.

Can you tell me which lua file used for training the dataset.

[leehomyc]

I did not include that to train the content network. But it is similar to the Context Encoders. You can use their code to train on ImageNet. (https://people.eecs.berkeley.edu/~pathak/context_encoder/)

[xuyifeng-nwpu]

@leehomyc The joint loss function of your paper is differ from that of context_encoder. Can I directly use the train code(https://people.eecs.berkeley.edu/~pathak/context_encoder/)?

Or should I modify the training lua code in the section of joint loss function?
Thanks.

[leehomyc]

No, you don't need to. The model for the content constraint is trained on ImageNet using Context Encoders.