search

ShemoonX / Chinese-image-caption

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

库版本, 中间生成文件的相关问题 #5

Open BobM-DS opened 3 years ago

BobM-DS commented 3 years ago

您好! 我在运行你的代码过程中遇到了一些问题 首先是一些库的版本 能否附上主要包的版本 (包括 tf keras imageai) 在跑代码的过程中遇到了疑似版本的问题: ' python multi_label_extract_imageai_block.py ' 中 抛出 AttributeError: module 'keras.backend' has no attribute 'get_session' 问题 我的本机包版本如下 tf=2.2.0 keras=2.4.3 imageai=2.1.5 我查了一下大多都是因为 tf 是 1.x 版本和 2.x 版本的差异问题 所以我想请问 你当时运行的具体版本是什么样子的. 其次是在文件 ' multi_label_extract_imageai_block.py ' 中第 122 行的 t.load('word_att.pth') 这个文件是在哪里生成的? 另外 这个文件中 第 24 行的 yolo.h5 的文件需要自己生成还是下载好使用?

希望您能解答 非常感谢

ShemoonX commented 3 years ago

您好，感谢你的提问。 （1）关于第一个问题，由于长时间没有维护代码，所以已记不清具体的软件版本，不好意思。但应该和说明中的版本一致，具体来说：Python 3.5 PyTorch 1.0 visdom jieba 0.38 imageai。至于你说的tf和keras错误，由于项目是基于pytorch实现的，所以有可能是一些以tf和keras作为后端的第三方库的问题，你还需要仔细排查一下。 （2）关于第二个问题，word_att.pth是需要根据你自己的数据由项目代码生成，具体生成代码应该在data_preprocess.py和data_preprocess_pic2att.py中可以找到。 （3）关于第三个问题，是的，项目使用下载好的预训练yolo模型文件进行目标检测。

BobM-DS commented 3 years ago

非常感谢您的回答! 我再检查一下, 希望能够运行出预期的结果