• Laser & Optoelectronics Progress
  • Vol. 57, Issue 22, 221505 (2020)
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
    DOI: 10.3788/LOP57.221505 Cite this Article Set citation alerts
    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 Copy Citation Text show less
    YOLOv3 network structure diagram. (a) Overall structure diagram of YOLOv3 network; (b) structure diagram of set conv layer and YOLO layer
    Fig. 1. 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
    Fig. 2. 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
    Fig. 3. Average IOU. (a) Relationship between Avg IOU and Mean IOU; (b) Avg IOU of k-means and k-thresh
    Image of object detection
    Fig. 4. Image of object detection
    Video-YOLOv3 network structure. (a) Overall structure diagram of video-YOLOv3 network; (b) structure diagram of splice-conv module
    Fig. 5. 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
    Fig. 6. Comparison of network structure. (a) YOLOv3 network structure; (b) video-YOLOv3 network structure
    Flow chart of predicting new image
    Fig. 7. Flow chart of predicting new image
    Loss function and Avg IOU curve of video-YOLOv3. (a) Loss function curve; (b) Avg IOU curve
    Fig. 8. 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
    Fig. 9. 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
    Fig. 10. Comparison of real-time detection. (a) Detection every 5 frames; (b) detection every 6 frames; (c) detection every 7 frames
    ClassPersonTvmonitorChair
    Total number250310501958
    Table 1. Number of images with each type of object in the dataset
    MethodAPmAP
    PersonTvmonitorChair
    Faster R-CNN70746569.67
    SSD72735867.67
    YOLOv377746070.33
    Video-YOLOv377766472.33
    Table 2. mAP values of different models on the dataset unit: %
    ItemCPU(Intel i7-7700k)GPU( Tesla K40)
    OriginalCorrectedOriginalCorrected
    Time /ms231.2251.7867.5416.39
    Max /(frame/s)4.3319.4515.7364.26
    Table 3. Comparison of CPU and GPU real-time detection time
    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