Laser & Optoelectronics Progress, Volume. 58, Issue 20, 2010005(2021)

Lightweight Target Detection Network Integrating Scene Context

Tingting Liu, Hua Miao*, Lin Li, Yang Xiang, Qi Li, and Qi Meng
Author Affiliations
  • School of Opto-Electronic Engineering, Changchun University of Science and Technology, Changchun, Jilin 130022, China
  • show less
    Figures & Tables(12)
    Structure diagram of SS-YOLOv3
    Structure diagram of MobileNet-YOLOv3
    Test results of YOLOv3
    Scene context module
    Typical aerial images. (a) Large scale variation; (b) complex background; (c) most small goals
    Comparison of visual test results and enlarged view of some areas. (a) YOLOv3 test results; (b) SS-YOLOv3 test results
    Visualization results of SS-YOLOv3
    • Table 1. Hyperparameters of pre-trained model

      View table

      Table 1. Hyperparameters of pre-trained model

      Weight decayBatch sizeLearning rateMomentumMaximum iteration
      0.0005160.001 (M<1000)0.0001(M>1000)0.910000
    • Table 2. Results of different target detection algorithms

      View table

      Table 2. Results of different target detection algorithms

      AlgorithmBackbonemAP /%FPS /(frame·s-1)
      Fast R-CNNVGG1670.06
      Faster R-CNNVGG1675.88
      YOLO64.318
      SSD300VGG1673.439
      SSD500VGG1674.922
      YOLOv2Darknet-1974.334
      YOLOv3Darknet-5375.927.6
      SS-YOLOv3MobileNetV384.333.4
    • Table 3. Detection results of YOLOv3 and SS-YOLOv3 unit: %

      View table

      Table 3. Detection results of YOLOv3 and SS-YOLOv3 unit: %

      ClassYOLOV3SS-YOLOv3Subtract
      Aeroplane74.684.910.3
      Car75.984.08.1
      Boat75.785.69.9
      Cycle76.382.46.1
      House74.384.610.3
      Electrocar76.982.75.8
      Person77.685.98.3
    • Table 4. Results of ablation learning

      View table

      Table 4. Results of ablation learning

      YOLOv3MobileNetV3Scene SRUCIOUmAP /%FPS /(frame·s-1)
      75.927.6
      74.236.4
      83.533.4
      84.333.4
    • Table 5. Detect results of YOLOv3 and SS-YOLOv3 unit: %

      View table

      Table 5. Detect results of YOLOv3 and SS-YOLOv3 unit: %

      AlgorithmPedestrianPersonBicycleCarVanTruckTricyleAwnBusMotor
      YOLOv320.758.047.4345.8230.5622.8913.049.2733.8411.46
      SS-YOLOv322.457.939.6551.0334.5424.1315.029.1336.8913.73
    Tools

    Get Citation

    Copy Citation Text

    Tingting Liu, Hua Miao, Lin Li, Yang Xiang, Qi Li, Qi Meng. Lightweight Target Detection Network Integrating Scene Context[J]. Laser & Optoelectronics Progress, 2021, 58(20): 2010005

    Download Citation

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

    Category: Image Processing

    Received: Nov. 12, 2020

    Accepted: Jan. 2, 2021

    Published Online: Oct. 12, 2021

    The Author Email: Miao Hua (ilev24@163.com)

    DOI:10.3788/LOP202158.2010005

    Topics