公开(公告)号：CN117974991B

公开(公告)日：2024-06-14

申请号：CN202410386901.2

申请日：2024-04-01

Applicant: 清华大学苏州汽车研究院(相城)

Inventor： 丁延超 ,  魏方圆 ,  刘玉敏 ,  陈赛 ,  郭俊超

IPC: G06V10/25 ,  G06V10/82 ,  G06V20/58 ,  G06N3/096 ,  G06N3/084 ,  G06N3/0455 ,  G06N3/0464 ,  G06N3/0495

Abstract: 本发明提供一种目标检测模型的生成方法、目标检测方法及其装置，该生成方法包括：获取用于交通信号灯检测的第一数据集，以及用于车辆检测的第二数据集；生成基于YOLO模型的第一教师模型和第二教师模型，并使用第一、第二数据集分别对第一、第二教师模型进行训练，并生成学生模型；之后，对第一教师模型和第二教师模型B的信息融合；之后，加入一个自编码器来实现特征图压缩的功能，从而实现保留关键信息的同时缩减模型体积，也缩减了模型的参数规模；之后，使用知识蒸馏将第一、第二教师模型蒸馏到学生模型中。从而实现了一个能够识别交通信号灯和车辆的模型，而且，在保证检测准确的前提下，缩小模型规模以提高运行效率。

公开(公告)号：CN117974991A

公开(公告)日：2024-05-03

申请号：CN202410386901.2

申请日：2024-04-01

Applicant: 清华大学苏州汽车研究院(相城)

Inventor： 丁延超 ,  魏方圆 ,  刘玉敏 ,  陈赛 ,  郭俊超

IPC: G06V10/25 ,  G06V10/82 ,  G06V20/58 ,  G06N3/096 ,  G06N3/084 ,  G06N3/0455 ,  G06N3/0464 ,  G06N3/0495

Abstract: 本发明提供一种目标检测模型的生成方法、目标检测方法及其装置，该生成方法包括：获取用于交通信号灯检测的第一数据集，以及用于车辆检测的第二数据集；生成基于YOLO模型的第一教师模型和第二教师模型，并使用第一、第二数据集分别对第一、第二教师模型进行训练，并生成学生模型；之后，对第一教师模型和第二教师模型B的信息融合；之后，加入一个自编码器来实现特征图压缩的功能，从而实现保留关键信息的同时缩减模型体积，也缩减了模型的参数规模；之后，使用知识蒸馏将第一、第二教师模型蒸馏到学生模型中。从而实现了一个能够识别交通信号灯和车辆的模型，而且，在保证检测准确的前提下，缩小模型规模以提高运行效率。