Zhangjinso / PISE

123 stars 28 forks source link

How to test the texture transfer model? #26

Closed confifu closed 3 years ago

confifu commented 3 years ago

I want to test only the texture transfer model. It is mentioned that the model needs to be changed. How should I change it? What inputs are needed to test the texture transfer model?

Zhangjinso commented 3 years ago

As far as I can remember, you can do it with some steps.

  1. You need to change this line to 'model_file_name = "model." + model_name + "_model_transfer" ' to import the texture transfer model.
  2. You also need to change this line to ' filename = 'generator_texture_transfer' if filename is None else filename ' to import generator.
  3. You need to change the dataloader and your test_pair.csv to input two reference images according to this line.
  4. You need to uncomment this line and modified the region index and 'alpha' (maybe change it to 1).
  5. Then you can test it.
happyday521 commented 3 years ago

@Zhangjinso Hi, in line 223 of generator_texture_transfer.py, what is the meaning of "if False"? Don't we need to execute indented codes under if?(line 223-265)Thanks!

Zhangjinso commented 3 years ago

Yes. The texture transfer model does not used spatio-aware normalization module for the convenience of training. If you want to get better results, you can retrain the model with this.

confifu commented 3 years ago

Pose transfer results using the pre-trained checkpoint are not as good as shown in the paper. Were different checkpoints used to get the results shown in the paper? fashionWOMENDressesid0000713204_4full_fashionWOMENDressesid0000713204_3back_all fashionWOMENDressesid0000715402_4full_fashionWOMENDressesid0000715402_2side_all fashionWOMENJackets_Coatsid0000450202_1front_fashionWOMENJackets_Coatsid0000450202_7additional_all fashionWOMENShortsid0000513803_4full_fashionWOMENShortsid0000513803_1front_all

Zhangjinso commented 3 years ago

You can refer to this issue.

confifu commented 3 years ago

What should be self.input_SPL3 ? SPL2_onehot has shape [8, 256, 256]. Do we also need to input one hot vector in input_SPL3 as well? Do we only have to leave the region for which we want the texture transfer?

confifu commented 3 years ago

If anyone wants to do this, To keep the pose same and only transfer texture, I did the following:

mlyarthur commented 3 years ago

image After following your instruction for texture transfer model testing, I cannot get the desired result in the paper @eternitjl @Zhangjinso Do you know what step I miss or make a mistake

ghost commented 2 years ago

image After following your instruction for texture transfer model testing, I cannot get the desired result in the paper @eternitjl @Zhangjinso Do you know what step I miss or make a mistake

@mlyarthur Hi,have you solved it? I have a similar problem image