microsoft / CoCosNet-v2

CoCosNet v2: Full-Resolution Correspondence Learning for Image Translation
https://arxiv.org/abs/2012.02047
MIT License
339 stars 36 forks source link

train Metfaces dataset error #24

Closed JcccKing closed 2 years ago

JcccKing commented 2 years ago

Hello friends, I use your model to train metfaces dataset and create paired label edge label After 100 rounds of training, classify the pictures according to the cosine similarity of VGG extracted features. The test results are as follows. What's the problem? Thank you and wish you success in your scientific research! image image

SystemErrorWang commented 1 year ago

hello JcccKing, do you have any update about this problem? I am facing the same problem and would like to know the reason

JcccKing commented 1 year ago

Hello I have solved part of the problem, but it does not reproduce the generalization effect in the paper completely. My solution is as follows

  1. First of all, we need to align the images, VGG calculate the similarity between external images and sample images, and choose the one with high similarity to generate. Try to avoid side faces and gender differences.
  2. You can refer to my paper(https://github.com/AiArt-HDU/MATEBIT), I subsequently improved v2; to improve the generalization of certain external data.