Joint compression and acceleration based on YOLOv3-MobilnetV1

Konferenz: ICMLCA 2021 - 2nd International Conference on Machine Learning and Computer Application
17.12.2021 - 19.12.2021 in Shenyang, China

Tagungsband: ICMLCA 2021

Seiten: 7Sprache: EnglischTyp: PDF

Persönliche VDE-Mitglieder erhalten auf diesen Artikel 10% Rabatt

Autoren:
Li, Xiaopeng; Ma, Yuchen; Li, Shuqin (School of Information Engineering, Northwest A&F University, Yangling, China)

Inhalt:
Traditional object detection algorithms have higher space complexity and longer reasoning time. This problem limits the deployment of deep neural networks on resource-constrained mobile devices and embedded devices. In order to solve this problem, this paper studies several model compression techniques based on the YOLOv3-MobilnetV1 object detection algorithm. By using the COCO data set, we compared the differences in the applicability of model compression and fast reasoning between pruning, quantification, knowledge distillation and their combination of object detection algorithms. The experimental results show that the optimization model using multiple compression methods reasonably achieves a compression rate of nearly 10 times without errors.