计算机与现代化 ›› 2020, Vol. 0 ›› Issue (12): 116-122.

• 图像处理 • 上一篇    

基于孪生网络结构的单样本图例检测方法

  

  1. (中国石油大学(华东)计算机科学与技术学院,山东青岛266580)
  • 出版日期:2021-01-07 发布日期:2021-01-07
  • 作者简介:王超奇(1994—),男,浙江杭州人,硕士研究生,研究方向:深度学习,目标检测,E-mail: 470640825@qq.com; 宫法明(1970—),男,教授,硕士,研究方向:机器视觉,大数据,知识图谱,E-mail: 565917954@qq.com。
  • 基金资助:
    科技部创新方法工作专项资助项目(2015IM010300)

Detection Method of One-shot Legend Based on Siamese Neural Networks

  1. (College of Computer Science and Technology, China University of Petroleum, Qingdao 266580, China)
  • Online:2021-01-07 Published:2021-01-07

摘要: 针对现有深度学习方法训练难、检测慢、训练数据难以获取等问题,在图例检测问题上,提出一种新的解决方法。以高效的卷积神经网络为骨干网络,并根据图例宽高比固定、具有个体独立性等特点,使用一种新的SiameseSSD检测框架进行目标检测。该框架包含了用于特征提取的孪生网络结构子网和用于分类和回归的改良SSD子网。同时利用数据增强技术和特殊的图片配对算法训练模型,通过解决单样本问题、调整网络结构和检测方法以检测大分辨率施工图。该方法在施工图数据集上的实验结果表明,该图例检测方法是一种新的解决单样本学习任务的方法,准确率达到91.3%,检测速度达到61帧/s,相比于其他现有的目标检测方式有一定的优势,几乎能够满足实际工程的工作需求。

关键词: 图例检测, 孪生网络, 数据增强, 单样本学习

Abstract: In view of the problems such as the difficulty of training, the slow detection and the difficulty of obtaining the training data in the existing deep learning methods, a new solution is proposed for the single sample learning problem. Based on the structure of convolutional neural network, combined with the characteristics of fixed aspect ratio and independence of legend, a new SiameseSSD detection frame is used for target detection. The framework includes a siamese subnet for feature extraction and an improved SSD subnet for classification and regression. At the same time, we use the data enhancement technology to expand the sample, then make the data set and train the model and adjust network structure and detection method to detect large-resolution construction drawings. The experimental results of this method on the construction drawing data set show that this method is a new method to solve the single sample learning task, with an accuracy of 91.3%, the detection speed reached 61 fps. Compared with the existing top level, it has certain advantages and meets the actual work needs.

Key words: legend detection, siamese network, data enhancement, one-shot learning