arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2506.09299cs.CVcs.LG

面向航空影像应急响应的量化YOLOv4-Tiny轻量化目标检测

Lightweight Object Detection Using Quantized YOLOv4-Tiny for Emergency Response in Aerial Imagery

Department of Electrical

Sindhu Boddu, Arindam Mukherjee

AI总结:

针对航空应急影像公开数据集缺失及边缘部署的轻量化需求,本文构建了含10820张标注图像的专用数据集,通过训练后INT8量化优化YOLOv4-Tiny,在保持检测性能的同时大幅压缩模型体积、提升推理速度,适配低功耗边缘设备的实时应急检测。

摘要:

本文提出了一种轻量化且高能效的目标检测方案,用于应急响应场景下拍摄的航空影像。我们重点部署YOLOv4-Tiny模型——一种紧凑的卷积神经网络,通过训练后量化将其优化至INT8精度。该模型在自定义构建的航空应急数据集上训练,该数据集包含10820张标注图像,覆盖多种关键应急场景。与依赖公开数据集的现有研究不同,由于缺乏公开的无人机视角应急影像,我们自行构建了该数据集,这也是本研究的核心贡献之一。我们将量化后的模型与YOLOv5-small在多个指标上进行对比评估,包括平均精度均值(mAP)、F1分数、推理时间和模型大小。实验结果表明,量化后的YOLOv4-Tiny检测性能与基线相当,同时模型大小从22.5 MB降至6.4 MB,推理速度提升44%。量化YOLOv4-Tiny模型大小减少71%、推理速度提升44%,非常适合在低功耗边缘设备上进行实时应急检测。

英文摘要:

This paper presents a lightweight and energy-efficient object detection solution for aerial imagery captured during emergency response situations. We focus on deploying the YOLOv4-Tiny model, a compact convolutional neural network, optimized through post-training quantization to INT8 precision. The model is trained on a custom-curated aerial emergency dataset, consisting of 10,820 annotated images covering critical emergency scenarios. Unlike prior works that rely on publicly available datasets, we created this dataset ourselves due to the lack of publicly available drone-view emergency imagery, making the dataset itself a key contribution of this work. The quantized model is evaluated against YOLOv5-small across multiple metrics, including mean Average Precision (mAP), F1 score, inference time, and model size. Experimental results demonstrate that the quantized YOLOv4-Tiny achieves comparable detection performance while reducing the model size from 22.5 MB to 6.4 MB and improving inference speed by 44\%. With a 71\% reduction in model size and a 44\% increase in inference speed, the quantized YOLOv4-Tiny model proves highly suitable for real-time emergency detection on low-power edge devices.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑