《计算机应用与软件杂志》发表论文赏析
作者:朱郝,杨世恩,
摘要:针对当前目标检测卷积神经网络参数量大、结构复杂、不善于提取全局特征信息的问题,基于MobileViT轻量化网络并结合YOLO系列算法的研究成果,设计一个轻量化的目标检测网络模型。将MobileViT作为骨干网络,增加提取全局特征信息的能力,而后使用PAN拓扑结构进行多尺度特征融合,再采用Decoupled Head结构提高检测精度。提出的网络模型以更小的参数量,在VOC07+12上的检测精度平均达到85.5%。在NVIDIA RTX3060 GPU上的检测速度为80.3FPS。实验结果表明,提出的算法与其他体量相当的目标检测算法相比,参数量更小,检测精度更高。
关键词:轻量化网络, 目标检测, YOLO, MobileViT