oneThousand1000 / HairMapper

(CVPR 2022) HairMapper: Removing Hair from Portraits Using GANs.
279 stars 39 forks source link

weird testing results: real images should be processed by image alignment #2

Closed Chopin68 closed 2 years ago

Chopin68 commented 2 years ago

Hi , Does this situation need to be solved through training? If so, how can I prepare a data set about her bald head C (181)

oneThousand1000 commented 2 years ago

Hi , Does this situation need to be solved through training? If so, how can I prepare a data set about her bald head C (181)

Hi, could you upload the input image and tell me the resolution of it?

Btw, did you process the input image with "image align code" that mentioned in readme?

oneThousand1000 commented 2 years ago

Hi , Does this situation need to be solved through training? If so, how can I prepare a data set about her bald head C (181)

Is it the result of testing? Or you get the bald child image during training?

Chopin68 commented 2 years ago

你好, 这种情况需要通过培训解决吗?如果是这样,我如何准备一个关于她光头的数据集 丙 (181)

嗨,你能上传输入图像并告诉我它的分辨率吗?

顺便说一句,您是否使用自述文件中提到的“图像对齐代码”处理输入图像?

The resolution is 512x512 C (181)

I didn't do image alignment, but I used 1024x1024 and succeeded in several cases, but sometimes bald child still appear.

Chopin68 commented 2 years ago

This is the result of the test. I haven't trained yet because I don't know how to collect some bald data sets to fit her.

oneThousand1000 commented 2 years ago

This is the result of the test. I haven't trained yet because I don't know how to collect some bald data sets to fit her.

As I explained in the readme, you should do image alignment first (the input image will be automatically clipped and resized to 1024x1024 ) then apply encoder4editing to obtain latent code, finally run our HairMapper.

The image alignment is only required in the testing stage, and there is no need to collect your own bald datasets because all the training images are generated by StyleGAN.

I applied image alignment to your input image: test_01

I get the result: test_01

Chopin68 commented 2 years ago

Thank you for your guidance! I already know how to get a normal result.