search

avalonstrel / GatedConvolution

An reimplement version of inpainting model in Free-Form Image Inpainting with Gated Convolution
Other
115 stars 27 forks source link

How to train the User-Guided model? #6

Open JinMengKang opened 5 years ago

JinMengKang commented 5 years ago

Dear author, I am very interested in the user - guide image in painting part of your work. However, I found that if I train directly, there is no image of EDG. How should I train such a model? What part of the code should I change. Another question, is your HED a pre-training model? Can you tell me where to modify the code that has been converted to TensorFlow in your code Looking forward to your reply.

avalonstrel commented 5 years ago

Thank you for your question, I have transfer the code into pytorch version, and you can refer to https://github.com/avalonstrel/GatedConvolution_pytorch for a better implementation. I am sorry to say I just abandon this version.