计算机应用 ›› 2018, Vol. 38 ›› Issue (11): 3211-3215.DOI: 10.11772/j.issn.1001-9081.2018041347

• 第七届中国数据挖掘会议(CCDM 2018) • 上一篇    下一篇

基于深度学习的虚拟边界检测方法

赖传滨1, 韩越兴1,2, 顾辉3,4, 王冰1   

  1. 1. 上海大学 计算机工程与科学学院, 上海 200444;
    2. 上海大学 上海先进通信与数据科学研究院, 上海 200444;
    3. 上海大学 材料基因组工程研究院, 上海 200444;
    4. 上海大学 材料科学与工程学院, 上海 200444
  • 收稿日期:2018-04-30 修回日期:2018-06-25 出版日期:2018-11-10 发布日期:2018-11-10
  • 通讯作者: 韩越兴
  • 作者简介:赖传滨(1994-),男,江西赣州人,硕士研究生,主要研究方向:材料图像处理、计算机视觉、机器学习;韩越兴(1975-),男,黑龙江海伦人,讲师,博士,主要研究方向:材料图像处理、分子机器人、计算机视觉、机器视觉;顾辉(1963-),男,江苏如皋人,教授,博士,主要研究方向:先进材料分层的相-微观结构-界面关系、微观结构的定量分析方法;王冰(1978-),女,山东掖县人,教授,博士,主要研究方向:复杂网络、鲁棒网络结构、网络同步性设计。
  • 基金资助:
    国家青年科学基金资助项目(61603237);上海市浦江人才计划项目(17PJ1402900);上海高校特聘教授(东方学者)岗位计划。

Novel virtual boundary detection method based on deep learning

LAI Chuanbin1, HAN Yuexing1,2, GU Hui3,4, WANG Bing1   

  1. 1. School of Computer Engineering and Science, Shanghai University, Shanghai 200444, China;
    2. Shanghai Institute for Advanced Communication and Data Science, Shanghai University, Shanghai 200444, China;
    3. Material Genome Institute, Shanghai University, Shanghai 200444, China;
    4. School of Materials Science and Engineering, Shanghai University, Shanghai 200444, China
  • Received:2018-04-30 Revised:2018-06-25 Online:2018-11-10 Published:2018-11-10
  • Supported by:
    This work is partially supported by the National Science Foundation for Young Scientists of China (61603237), the Shanghai Pujiang Program (17PJ1402900), the Program for Professor of Special Appointment (Eastern Scholar) at at Shanghai Institutions of Higher Learning.

摘要: 针对传统边缘检测方法无法对材料微观图像中不同区域间存在的"虚拟边界"(VB)进行准确检测的问题,提出一种基于卷积神经网络(CNN)的虚拟边界检测模型,称之为"虚拟边界网络"(VBN)。该模型对VGGNet深度学习模型进行了简化,并在模型训练过程中采用了dropout以及Adam算法等优化策略。VBN以图像中每个像素为中心所取的图像块作为输入,然后输出该图像块所属的类别并据此判断中心像素是否属于虚拟边界。在对两类材料图像进行虚拟边界检测的实验中,VBN的平均检测精度到达92.5%,平均召回率达到89.5%,证明该模型能够准确、有效地对图像中的虚拟边界进行检测,是一种替代低效率人工分析方法的有效手段。

关键词: 虚拟边界检测, 边缘检测, 卷积神经网络, 深度学习, 图像分割

Abstract: Traditional edge detection methods can not accurately detect the Virtual Boundary (VB) between different regions in materials microscopic images. In order to solve this problem, a virtual boundary detection model based on Convolutional Neural Network (CNN) called Virtual Boundary Net (VBN) was proposed. The VGGNet (Visual Geometry Group Net) deep learning model was simplified, and dropout and Adam algorithms were applied in the training process. An image patch centered on each pixel in the image was extracted as the input, and the class of the image patch was output to decide whether the center pixel belongs to the virtual boundary or not. In the experiment of virtual boundary detection for two kinds of material images, the average detection precision of this method reached 92.5%, and the average recall rate reached 89.5%. The experimental results prove that the VBN can detect the virtual boundary in the image accurately and effectively, which is an alternative method to low-efficient manual analysis.

Key words: virtual boundary detection, edge detection, Convolutional Neural Network (CNN), deep learning, image segmentation

中图分类号: