search

yunjey / stargan

StarGAN - Official PyTorch Implementation (CVPR 2018)
MIT License
5.21k stars 968 forks source link

Custom evaluation dataset #14

Open ser0t0nin opened 6 years ago

ser0t0nin commented 6 years ago

Hello, thank you for your code. I performed the training stage and got the model - works perfectly on the default datasets. Is it possible to cast the model on my custom imageset? I created a proper folder structure, created valid metafile, filled all the paths (as i think) and got error 

Traceback (most recent call last):
  File "main.py", line 106, in <module>
    main(config)
  File "main.py", line 46, in main
    solver.test()
  File "/home/username/Documents/StarGAN/solver.py", line 679, in test
    target_c_list = self.make_celeb_labels(org_c)
  File "/home/username/Documents/StarGAN/solver.py", line 181, in make_celeb_labels
    c[i] = 0 if c[i] == 1 else 1   # opposite value
IndexError: index 4 is out of range for dimension 0 (of size 4)

Is it possible in general? Can you provide a kind of instruction, so people can use it without tough reverse engineering? Thanks

ghost commented 6 years ago

You have tested this model successfully. Can you please help me? I want to generate different expressions from neutral image...

Test time it generate only one image multiple times 1

iamzhangyunlu commented 4 years ago

You have tested this model successfully. Can you please help me? I want to generate different expressions from neutral image...

Test time it generate only one image multiple times 1

Have you solve the problem yet? Have you find the reason of the same expression?