公开(公告)号：CN116630918A

公开(公告)日：2023-08-22

申请号：CN202310514016.3

申请日：2023-05-09

Applicant: 苏州大学

Inventor： 姚望舒 ,  彭思源 ,  薛一帆

IPC: G06V20/56 ,  G06N3/08 ,  G06V10/82 ,  G06V10/762 ,  G06V10/46 ,  G06N3/0464

Abstract: 本发明涉及图像处理和车辆智能驾驶领域，提供了一种基于矩形注意力机制的车道线检测方法，步骤包括：获取车道线图像，将所述车道线图像输入到已训练收敛的基于矩形注意力机制的车道线检测模型中进行下采样，将所述下采样后的特征图通过注意力机制网络得到注意力加权后的特征图，将所述注意力加权后的特征图上采样，将上采样后的特征图通过关键点检测网络得到包含车道线起始点在内的所有关键点坐标，采用基于距离的聚类网络，通过关键点坐标与其所属起始点间的偏移量来将关键点聚类为车道线实例。本发明解决了自动驾驶中存在的难检测车道线场景，提高了模型的推理速度和预测准确度。

公开(公告)号：CN117808071A

公开(公告)日：2024-04-02

申请号：CN202410073197.5

申请日：2024-01-18

Applicant: 苏州大学

Inventor： 姚望舒 ,  薛一帆 ,  姚士佑 ,  彭思源

IPC: G06N3/082 ,  G06N3/0464

Abstract: 本发明涉及一种基于通道关系感知的分层全局剪枝方法和系统，其中，方法包括：骤S1：在CNN模型训练中，计算特征图中通道的关联特征注意力权重和原始特征注意力权重；步骤S2：根据所述关联特征注意力权重和原始特征注意力权重得到关系感知注意力权重，并根据CNN模型训练数据对所述关系感知注意力权重取平均，并将取平均结果作为特征图中通道的重要性分数；步骤S3：根据所述特征图中通道的重要性分数对特征图的通道进行剪枝。本发明在剪枝过程中考虑了通道之间的相互依赖关系，能够对通道进行有效剪枝。