hanzhanggit / StackGAN-Pytorch

MIT License
491 stars 146 forks source link

How to match the generated images with the caption? #23

Closed Hanzy1996 closed 5 years ago

Hanzy1996 commented 5 years ago

I have finished the training of 2 stages on CUB and I can generate the samples during the test. During the test, I also use the .pickle file extracted from the char-CNN-RNN text embeddings file of CUB as the embedding. But I failed to match the caption with the generated images. I just use the corresponding description during test in [self.caption]( # self.captions = self.load_all_captions()), but they are not matched. How to get the correct caption of the generated images?

image

Much appreciation!