Meshed-memory transformer自有数据集的文件预处理
作者:安静到无声 个人主页
目录
- Meshed-memory transformer自有数据集的文件预处理
- 生成与coco_detections.hdf5相似的特征文件
- 生成训练、测试和验证对应的JSON和字幕ID
- 推荐专栏
生成与coco_detections.hdf5相似的特征文件
coco数据集训练 Meshed-memory transformer的时候特征文件为coco_detections.hdf5,它是通过BUTD提取的特征,为了了解提取的内容主要是什么,我们对这个文件进行了加载,发现主要包括了boxes,cls_prob和features三个类型的数据,其中的前缀100000代表了图片的ID编号,具体的情况如下所示:
# 导入文件
import<