软件学报

作者：陈健，白琮，马青，郝鹏翼，陈胜勇

单位：陈健，浙江工业大学 计算机科学与技术学院, 浙江 杭州 31002311,白琮，浙江工业大学 计算机科学与技术学院, 浙江 杭州 31002302,马青，浙江工业大学 计算机科学与技术学院, 浙江 杭州 310023;浙江工业大学 理学院, 浙江 杭州 31002303,郝鹏翼，浙江工业大学 计算机科学与技术学院, 浙江 杭州 31002304,陈胜勇，天津理工大学 计算机科学与工程学院, 天津 30038405

关键词：基于草图的图像检索;细粒度检索;三元组网络;对抗训练

基金：国家重点研发计划（2018YFB1305200）；浙江省自然科学基金（LY18F020032，LY18F020034）；浙江省教育厅项目（Y201839922）

将草图作为检索示例用于图像检索称为基于草图的图像检索，简称草图检索.其中，细粒度检索问题或类内检索问题是2014年被研究者提出并快速成为广受关注的研究方向.目前研究者通常用三元组网络来解决类内检索问题，且取得了不错的效果.但是三元组网络的训练非常困难，很多情况下很难收敛甚至不收敛，且存在着容易过拟合的风险.借鉴循环生成对抗训练的思想，设计了SketchCycleGAN帮助提高三元组网络训练过程的效率，以对抗训练的方式使其参与到三元组网络的训练过程中，通过充分挖掘数据集自身信息的方式取代了利用其他数据集进行预训练的过程，在简化训练步骤的基础上取得了更好的检索性能.通过在常用的细粒度草图检索数据集上的一系列对比实验，证明了所提方法的有效性和优越性.

来源：2020年第7期

《软件学报》期刊编辑部

查看软件学报杂志2020年第7期