taki0112 / SPADE-Tensorflow

Simple Tensorflow implementation of "Semantic Image Synthesis with Spatially-Adaptive Normalization" a.k.a. GauGAN, SPADE (CVPR 2019 Oral)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

The segmap_label of this project

Andyzzz opened this issue · comments

Hi,
I tested the model with the segmap_label.txt in the README and the segmap images in ./dataset/spade_celebA/segmap_test:

segmap_label.txt for CelebAMask-HQ
{(0, 0, 0): 0, (0, 0, 255): 1, (255, 0, 0): 2, (150, 30, 150): 3, (255, 65, 255): 4, (150, 80, 0): 5, (170, 120, 65): 6, (125, 125, 125): 7, (255, 255, 0): 8, (0, 255, 255): 9, (255, 150, 0): 10, (255, 225, 120): 11, (255, 125, 125): 12, (200, 100, 100): 13, (0, 255, 0): 14, (0, 150, 80): 15, (215, 175, 125): 16, (220, 180, 210): 17, (125, 125, 255): 18}

but the output images are not correct, I think the reason is the segmap_label.txt does not correspond to the images the project provide, so what is the correct segmap_label?
Thanks!