基本信息:
- 专利标题： 目标检测模型的生成方法、目标检测方法及其装置
- 申请号：CN202410386901.2 申请日：2024-04-01
- 公开(公告)号：CN117974991B 公开(公告)日：2024-06-14
- 发明人： 丁延超 , 魏方圆 , 刘玉敏 , 陈赛 , 郭俊超
- 申请人： 清华大学苏州汽车研究院（相城）
- 申请人地址： 江苏省苏州市相城区高铁新城太阳路2266号5幢11层
- 专利权人： 清华大学苏州汽车研究院（相城）
- 当前专利权人： 清华大学苏州汽车研究院（相城）
- 当前专利权人地址： 江苏省苏州市相城区高铁新城太阳路2266号5幢11层
- 代理机构： 苏州威世朋知识产权代理事务所(普通合伙)
- 代理人： 陈如建
- 主分类号： G06V10/25
- IPC分类号： G06V10/25 ; G06V10/82 ; G06V20/58 ; G06N3/096 ; G06N3/084 ; G06N3/0455 ; G06N3/0464 ; G06N3/0495
摘要：
本发明提供一种目标检测模型的生成方法、目标检测方法及其装置，该生成方法包括：获取用于交通信号灯检测的第一数据集，以及用于车辆检测的第二数据集；生成基于YOLO模型的第一教师模型和第二教师模型，并使用第一、第二数据集分别对第一、第二教师模型进行训练，并生成学生模型；之后，对第一教师模型和第二教师模型B的信息融合；之后，加入一个自编码器来实现特征图压缩的功能，从而实现保留关键信息的同时缩减模型体积，也缩减了模型的参数规模；之后，使用知识蒸馏将第一、第二教师模型蒸馏到学生模型中。从而实现了一个能够识别交通信号灯和车辆的模型，而且，在保证检测准确的前提下，缩小模型规模以提高运行效率。
公开/授权文献:
- CN117974991A 目标检测模型的生成方法、目标检测方法及其装置 公开/授权日：2024-05-03