simple-vqa

• 一个简单的 vqa demo
• 网上代码比较乱，要么环境不对，要么要现在各种大数据集和模型权重
• 代码基于 Python3，推荐用 anaconda 维护纯洁环境

About the Dataset

• 4000 个训练图片
• 1000 张测试图片 所有图片是 64*64 的，包含计数 识别 颜色 判断等问题

Example Images

easy_vqa 和 use_keras 中的 zip 文件解压后可以看数据集 包括问题 答案 和 图片

Installing the Package

pip install simple-vqa

Generating the Dataset

安装必备的库

pip install -r gen_data/requirements.txt

使用如下命令生成数据，可自行调整

python gen_data/generate_data.py

Using the Package

通过 use_keras 下 requirements.txt 下的 安装必备的库

pip install -r use_keras/requirements.txt

• use_keras 下 train.py 训练模型
• windows.py 可视化界面
• analyze.py 分析结果
• model.py 模型的结构 可自行修改