Multi-Branch Distance-Sensitive Self-Attention Network for Image Captioning(MDSANet)在自定义数据上的训练与测试调试
1. 环境设置
我们采用和Lstnet一样的环境即可,所以我们直接克隆环境。我们执行如下命令,进行克隆:
conda create --name MDSANet --clone lstnet
激活进入:
conda activate MDSANet
退出环境:
conda deactivate
删除虚拟环境:
conda env remove --name MDSANet
2. 查看所用数据
主要使用的特征包括:coco_grid_features
.hdf5和annotation_folder
所使用的网格特征和rstnet
代码是一致的,所以我们可以复制所对应的文件夹,并将其重命名为MDSANet
即可:
cp -r ./rstnet ./MDSANet
至此完成了数据的准备: