Laser & Optoelectronics Progress, Volume. 57, Issue 22, 221505(2020)

Real-Time Object Detection Based on Improved YOLOv3 Network

Jia Sun, Dabo Guo*, Tiantian Yang, and Shitu Ma
Author Affiliations
  • College of Physics and Electronic Engineering, Shanxi University, Taiyuan, Shanxi 030006, China
  • show less
    Figures & Tables(13)
    YOLOv3 network structure diagram. (a) Overall structure diagram of YOLOv3 network; (b) structure diagram of set conv layer and YOLO layer
    Dataset analysis results. (a) Target width and height distribution of the dataset; (b) k-means clustering analysis result
    Average IOU. (a) Relationship between Avg IOU and Mean IOU; (b) Avg IOU of k-means and k-thresh
    Image of object detection
    Video-YOLOv3 network structure. (a) Overall structure diagram of video-YOLOv3 network; (b) structure diagram of splice-conv module
    Comparison of network structure. (a) YOLOv3 network structure; (b) video-YOLOv3 network structure
    Flow chart of predicting new image
    Loss function and Avg IOU curve of video-YOLOv3. (a) Loss function curve; (b) Avg IOU curve
    Comparison of test results. (a) YOLOv3 test results; (b) video-YOLOv3 test results
    Comparison of real-time detection. (a) Detection every 5 frames; (b) detection every 6 frames; (c) detection every 7 frames
    • Table 1. Number of images with each type of object in the dataset

      View table

      Table 1. Number of images with each type of object in the dataset

      ClassPersonTvmonitorChair
      Total number250310501958
    • Table 2. mAP values of different models on the dataset unit: %

      View table

      Table 2. mAP values of different models on the dataset unit: %

      MethodAPmAP
      PersonTvmonitorChair
      Faster R-CNN70746569.67
      SSD72735867.67
      YOLOv377746070.33
      Video-YOLOv377766472.33
    • Table 3. Comparison of CPU and GPU real-time detection time

      View table

      Table 3. Comparison of CPU and GPU real-time detection time

      ItemCPU(Intel i7-7700k)GPU( Tesla K40)
      OriginalCorrectedOriginalCorrected
      Time /ms231.2251.7867.5416.39
      Max /(frame/s)4.3319.4515.7364.26
    Tools

    Get Citation

    Copy Citation Text

    Jia Sun, Dabo Guo, Tiantian Yang, Shitu Ma. Real-Time Object Detection Based on Improved YOLOv3 Network[J]. Laser & Optoelectronics Progress, 2020, 57(22): 221505

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Machine Vision

    Received: Apr. 2, 2020

    Accepted: Apr. 27, 2020

    Published Online: Nov. 5, 2020

    The Author Email: Dabo Guo (dabo_guo@sxu.edu.cn)

    DOI:10.3788/LOP57.221505

    Topics