电子科技 ›› 2025, Vol. 38 ›› Issue (6): 23-29.doi: 10.16180/j.cnki.issn1007-7820.2025.06.004

• • 上一篇    下一篇

基于深度学习的车位检测方

唐玉良, 张轩雄()   

  1. 上海理工大学 光电信息与计算机工程学院,上海 200093
  • 收稿日期:2023-11-17 修回日期:2023-12-12 出版日期:2025-06-15 发布日期:2025-06-24
  • 通讯作者: 张轩雄(1965-),男, E-mail:xuanxiongzhang@163.com,博士,教授。研究方向:微电子机械系统。
  • 作者简介:唐玉良(1998-),男,硕士研究生。研究方向:计算机视觉。
  • 基金资助:
    国家自然科学基金(62276167)

A Parking Space Detection Method Based on Deep Learning

TANG Yuliang, ZHANG Xuanxiong()   

  1. School of Optical-Electrical and Computer Engineering,University of Shanghai for Science and Technology,Shanghai 200093,China
  • Received:2023-11-17 Revised:2023-12-12 Online:2025-06-15 Published:2025-06-24
  • Supported by:
    National Natural Science Foundation of China(62276167)

摘要:

针对自动泊车系统中车位检测算法存在的任务单一、车位关键点遮挡导致检测效率下降等问题,文中结合目标检测和关键点检测提出了一种基于YOLO(You Only Look Once)的车位检测模型,可同时检测车位区域、车位使用情况以及车位关键点。在YOLOv7-tiny(You Only Look Once version7-tiny)模型输出端增加车位关键点回归分支并对其进行编码设计,通过改进损失函数提高模的识别和定位准确性。在主干网络融合注意力机制,提高网络的特征提取能力。实验结果表明,相较于YOLOv7-tiny,所提模型车位区域识别准确率均值提升了1.8百分点,车位使用情况判断准确率提升了1.1百分点,并且增加了车位关键点定位功能,定位成功率达到94.1%,具有较高的应用价值。

关键词: 自动泊车, 遮挡, 目标检测, YOLOv7-tiny, 关键点检测, 特征提取, 注意力机制, 损失函数

Abstract:

In view of the problems such as single task and occlusion of key points in parking detection algorithm in automatic parking system, a YOLO(You Only Look Once)-based parking detection model is proposed by combining target detection and key point detection, which can simultaneously detect parking area, parking use and key points. The key point regression branch of parking space is added to the output end of YOLOv7-tiny(You Only Look Once version7-tiny) model and its coding design is carried out to improve the accuracy of model identification and positioning by improving the loss function. The attention mechanism is integrated in the backbone network to improve the feature extraction capability of the network. The experimental results show that compared with YOLOv7-tiny, the average parking area recognition accuracy of the proposed model is increased by 1.8 percentage, and the parking space usage judgment accuracy is increased by 1.1 percentage. Moreover, the key point location function is added, and the success rate of location reaches 94.1%, indicating that the proposed method has high application value.

Key words: automatic parking, occlusion, target detection, YOLOv7-tiny, keypoints detection, feature extraction, attention mechanism, loss function

中图分类号: 

  • TP391