Efficient Yolo: A Lightweight Model For Embedded Deep Learning Object Detection

2020 IEEE International Conference on Multimedia & Expo Workshops (ICMEW)(2020)

引用 20|浏览193
暂无评分
摘要
It is essential to pursue efficiency for on-road object detection task. To incorporate deep model into embedded devices while maintaining high accuracy, in this paper, an Efficient YOLO framework is rebuilt based on traditional YOLOv3. Firstly, an iterative initialization strategy is designed to ensure the network sparsity in the initial training. Then comprehensive pruning schemes including layer-level and channel-wise pruning are proposed to lighten the model parameters.With the support of external dataset, the detection accuracy remains at a high level. Compared with the orignal version, our model shrinks the model size by 96.93% and calculation amount by 84.36%. The inference speed is improved 2.23 times on NVIDIA Jetson TX2 platform. Finally, we achieve a mAP of 0.492 on the testing dataset, and rank the top accuracy of ICME competition.
更多
查看译文
关键词
Object detection,YOLO,embedded system,model pruning
AI 理解论文
溯源树
样例
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要