arXiv reaDer
YOLObile:圧縮コンパイル共同設計によるモバイルデバイスでのリアルタイムオブジェクト検出
YOLObile: Real-Time Object Detection on Mobile Devices via Compression-Compilation Co-Design
物体検出技術の急速な発展と幅広い利用により、物体検出器の精度と速度の両方に注目が集まっています。ただし、現在の最先端のオブジェクト検出作業は、大きなモデルを使用して精度を重視しているが待ち時間が長くなるか、軽量モデルを使用して速度を重視しているが精度を犠牲にしている。この作業では、圧縮コンパイルの共同設計によるモバイルデバイスでのリアルタイムオブジェクト検出であるYOLObileフレームワークを提案します。新しいブロックパンチプルーニングスキームが、任意のカーネルサイズに対して提案されています。モバイルデバイスでの計算効率を向上させるために、GPU-CPUコラボレーションスキームが、高度なコンパイラ支援最適化とともに採用されています。実験結果は、私たちの剪定スキームが49.0mAPでYOLOv4の14倍の圧縮率を達成することを示しています。 YOLObileフレームワークでは、Samsung GalaxyS20でGPUを使用して17FPSの推論速度を達成しています。提案されたGPU-CPUコラボレーションスキームを組み込むことにより、推論速度は19.1 FPSに向上し、元のYOLOv4より5倍高速化されます。ソースコードはhttps://github.com/nightsnack/YOLObileにあります。
The rapid development and wide utilization of object detection techniques have aroused attention on both accuracy and speed of object detectors. However, the current state-of-the-art object detection works are either accuracy-oriented using a large model but leading to high latency or speed-oriented using a lightweight model but sacrificing accuracy. In this work, we propose YOLObile framework, a real-time object detection on mobile devices via compression-compilation co-design. A novel block-punched pruning scheme is proposed for any kernel size. To improve computational efficiency on mobile devices, a GPU-CPU collaborative scheme is adopted along with advanced compiler-assisted optimizations. Experimental results indicate that our pruning scheme achieves 14× compression rate of YOLOv4 with 49.0 mAP. Under our YOLObile framework, we achieve 17 FPS inference speed using GPU on Samsung Galaxy S20. By incorporating our proposed GPU-CPU collaborative scheme, the inference speed is increased to 19.1 FPS, and outperforms the original YOLOv4 by 5× speedup. Source code is at: https://github.com/nightsnack/YOLObile.
updated: Wed Dec 30 2020 15:55:43 GMT+0000 (UTC)
published: Sat Sep 12 2020 01:41:08 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト