Model Compression Based on YOLOv3 Object Detection Algorithm from the Perspective of UAV

Recently, unmanned aerial vehicles are widely used in surveillance, aerial photography, power grid line inspections and other places. In order to deploy the YOLOv3 [1] algorithm on drones, it is necessary to adopt the YOLOv3 algorithm with fewer parameters and a simpler structure. This paper impleme...

Ausführliche Beschreibung

Gespeichert in:
Bibliographische Detailangaben
Veröffentlicht in:Chinese Control Conference S. 8439 - 8444
Hauptverfasser: Wu, Bing, Xue, Zhijun, Chen, Wenjie
Format: Tagungsbericht
Sprache:Englisch
Veröffentlicht: Technical Committee on Control Theory, Chinese Association of Automation 26.07.2021
Schlagworte:
Drones
Inspection
Model Compress
Object detection
Photography
Power grids
RK3399pro
Simulation
Surveillance
YOLOv3
ISSN:1934-1768
Online-Zugang:Volltext
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
Beschreibung
Zusammenfassung:Recently, unmanned aerial vehicles are widely used in surveillance, aerial photography, power grid line inspections and other places. In order to deploy the YOLOv3 [1] algorithm on drones, it is necessary to adopt the YOLOv3 algorithm with fewer parameters and a simpler structure. This paper implements the model compression of YOLOv3 based on methods such as sparseness, pruning, and knowledge distillation. This paper implements the sparseness of the network by adding L1 regular expressions on the convolutional layer. After that, redundant channels and layers are removed through channel pruning and layer pruning. After sparse and pruning, the mAP lost a lot. By using knowledge distillation after pruning, it attempts to recover mAP lost in sparseness and pruning. By this method, the YOLOv3 algorithm can be deployed on embedded platforms such as RK3399pro. We evaluate the model on the visdrone2019 dataset. The experimental results show that after model compression, YOLOv3 is more suitable for deployment on embedded platforms.
ISSN:1934-1768
DOI:10.23919/CCC52363.2021.9550707