An error occurred while using complex text，my prompt is a racoon detective using a microscope while riding in a train

wtliao / text2image

Text to Image Generation with Semantic-Spatial Aware GAN

179 stars 34 forks source link

An error occurred while using complex text，my prompt is a racoon detective using a microscope while riding in a train #28

Open ctxya1207 opened 9 months ago

ctxya1207 commented 9 months ago

Traceback (most recent call last): File "E:\Project\pythonProject\SSA_GAN\main.py", line 549, in gen_sample(text_encoder, netG, device, wordtoix, captions) # generate images with description from user File "E:\Project\pythonProject\SSA_GAN\main.py", line 189, in gen_sample caps[cl][cw] = wordtoix[word.lower()] KeyError: 'detective'