search

ShemoonX / Chinese-image-caption

Image Chinese Description Generation Based on Multi-level Selective Visual Semantic Attributes
12 stars 5 forks source link

What is the word_att.pth in multi_label_extract_imgeai_block.py? #4

Closed DANG-ai closed 4 years ago

DANG-ai commented 4 years ago

Hi, I could not find the word_att.pth when I run the multi_label_extract_imgeai_block.py. Where could I get this pth file? Thanks!

ShemoonX commented 4 years ago

thanks for your question,Thank you for your question. This is my negligence. I didn't specify to save this file in the code. You need to save it during data processing. In data preprocess pic2att.py， you can get and save the word_att.pth 。