Research of object detection method based on DCGAN data-set enhancement technique

被引:0
|
作者
Shi Dunhuang [1 ]
Yu Yanan [1 ,2 ]
Li Huiping [1 ]
机构
[1] Tianjin Univ Technol & Educ, Sch Informat Technol Engn, Tianjin 300222, Peoples R China
[2] Tianjin Univ, Key Lab Micro Optoelectro Mech Syst Technol, Minist Educ, Tianjin 300072, Peoples R China
关键词
Deep learning; DCGAN; object detection; data enhancement;
D O I
10.1117/12.2606531
中图分类号
P1 [天文学];
学科分类号
0704 ;
摘要
With the rise of the new generation of artificial intelligence technology, the object detection method based on deep learning has achieved remarkable results. In this paper, the detection accuracy of three popular object detection algorithms such as You Only Look Once (YOLO V3), Region-CNN (Faster R-CNN) and Single Shot MultiBox Detector (SSD) has been compared. Aiming at the actual detection problems of building block parts with irregular shape and different sizes, a method that combines deep convolutional generative adversarial networks (DCGAN) with deep learning based object detection algorithm is proposed to solve the problems of over fitting or weak generalization ability in the case of limited datasets, and to improve the detection accuracy of object detection algorithm. Experimental results show that: 1. Using public datasets, when the training data is reduced, the mean average precision (mAP) values of the above three algorithms are reduced respectively. Among those, SSD algorithm has the smallest change, which decreases 7.81%. 2. The control variable method is used to train the building block parts. In the case of insufficient training data, the detection accuracy of three object detection algorithms is low. 3. After combining SSD algorithm with DCGAN algorithm and applying it into the detection task of building block parts, the mAP value is improved from 79.63% to 83.32%, and the detection accuracy is obviously improved.
引用
收藏
页数:6
相关论文
共 50 条
  • [21] Corn Ear Quality Recognition Based on DCGAN Data Enhancement and Transfer Learning
    Shi, Benzheng
    Zhou, Xiaolei
    Qin, Zhikang
    Sun, Lu
    Xu, Yan
    ICECC 2021: 4TH INTERNATIONAL CONFERENCE ON ELECTRONICS, COMMUNICATIONS AND CONTROL ENGINEERING, 2021, : 62 - 68
  • [22] Data Augmentation of Raman Spectral and Its Application Research Based on DCGAN
    Li Ling-qiao
    Li Yan-hui
    Yin Lin-lin
    Yang Hui-hua
    Feng Yan-chun
    Yin Li-hui
    Hu Chang-qin
    SPECTROSCOPY AND SPECTRAL ANALYSIS, 2021, 41 (02) : 400 - 407
  • [23] Data augmentation in network intrusion detection based on S-DCGAN
    Su, Mengwei
    Chai, SenChun
    2021 PROCEEDINGS OF THE 40TH CHINESE CONTROL CONFERENCE (CCC), 2021, : 8616 - 8621
  • [24] Research on large data set clustering method based on MapReduce
    Wei, Pengcheng
    He, Fangcheng
    Li, Li
    Shang, Chuanfu
    Li, Jing
    NEURAL COMPUTING & APPLICATIONS, 2020, 32 (01): : 93 - 99
  • [25] Research on large data set clustering method based on MapReduce
    Pengcheng Wei
    Fangcheng He
    Li Li
    Chuanfu Shang
    Jing Li
    Neural Computing and Applications, 2020, 32 : 93 - 99
  • [26] Classification on Diabetes Mellitus Data-set Based-on Artificial Neural Networks and ANFIS
    Vosoulipour, A.
    Teshnehlab, M.
    Moghadam, H. A.
    4TH KUALA LUMPUR INTERNATIONAL CONFERENCE ON BIOMEDICAL ENGINEERING 2008, VOLS 1 AND 2, 2008, 21 (1-2): : 27 - 30
  • [27] Classification-based symbolic indoor positioning over the Miskolc IIS Data-set
    Tamas, Judit
    Toth, Zsolt
    JOURNAL OF LOCATION BASED SERVICES, 2018, 12 (01) : 2 - 18
  • [28] Object Detection Method Based on Shallow Feature Fusion and Semantic Information Enhancement
    Luo, Huilan
    Wang, Pei
    Chen, Hongkun
    Xu, Min
    IEEE SENSORS JOURNAL, 2021, 21 (19) : 21839 - 21851
  • [29] WCE-DCGAN: A data augmentation method based on wireless capsule endoscopy images for gastrointestinal disease detection
    Xiao, Zhiguo
    Lu, Jia
    Wang, Xiaokun
    Li, Nianfeng
    Wang, Yuying
    Zhao, Nan
    IET IMAGE PROCESSING, 2023, 17 (04) : 1170 - 1180
  • [30] A Variational Level Set Method for Multiple Object Detection
    Pan, Zhenkuan
    Li, Hua
    Wei, Weibo
    Xu, Shuhua
    ADVANCES IN VISUAL COMPUTING, PT II, PROCEEDINGS, 2008, 5359 : 733 - 742