计算机工程

作者：徐守坤, 吉晨晨, 倪楚涵, 李宁

单位：常州大学 信息科学与工程学院 数理学院, 江苏 常州 213164

关键词：YOLOv3网络,施工场景,空间关系,图像描述,关系检测模型

基金：国家自然科学基金（61803050）。

为解决施工场景中缺少空间关系图像描述的问题，提出一种融合施工场景及空间关系的图像描述生成模型。采用YOLOv3网络进行目标检测，以TransE算法为基础在传统对象检测模型中加入特征提取层形成关系检测模型，结合对象坐标框信息得到对象之间的关系，并采用基于规则和模板的方法生成图像描述。实验结果表明，与m-RNN、NIC、Soft-Attention等模型相比，该模型能生成更准确的空间关系图像描述。

来源：2020年第6期

《计算机工程》期刊编辑部

查看计算机工程杂志2020年第6期