yangxy / GPEN

2.43k stars 451 forks source link

MASK_COLORMAP颜色生成顺序不对应 #135

Open xuyu666 opened 2 years ago

xuyu666 commented 2 years ago

您好: 我最近在尝试复现您的“segmentation2face”的工作,但是在执行的时候遇到一些问题。

问题描述: 复现过程中,发现使用您提供制作mask的方式与模型权重的style不匹配,导致生成图像效果差。 请问您可以提供制作’/example/seg‘时,对应的MASK_COLORMAP嘛?

复现过程:

  1. 使用 example 提供的 Seg 测试了segmentation2face模型,可以复现成功。 image

  2. 提供自己的图片,使用‘’face_parse/face_parsing.py‘生成自己的mask,颜色与您提供的’face_parse/mask.png‘(不同于’example/seg‘)匹配 image image

  3. 使用生成好的mask复原,结果差距很大,经过查验,应该是 mask label不一致导致的。 image

yangxy commented 2 years ago

是的,你需要保证mask label的一直性。可以参考这里,这也是我所采用的