一种通用的嵌入式平台的行人检测模型训练方法

基本信息

申请号 CN202110883375.7 申请日 -
公开(公告)号 CN113343949A 公开(公告)日 2021-09-03
申请公布号 CN113343949A 申请公布日 2021-09-03
分类号 G06K9/00;G06K9/62;G06T1/20;G06N3/08 分类 计算;推算;计数;
发明人 王建兴;王纵驰;刘健;臧媛媛;王培;于佳耕 申请(专利权)人 航天神舟智慧系统技术有限公司
代理机构 北京中知法苑知识产权代理有限公司 代理人 李明;赵吉阳
地址 100088 北京市海淀区马甸路2号中航油大厦
法律状态 -

摘要

摘要 一种通用的基于嵌入式平台的行人检测模型的训练方法,包括:数据集的准备,进行图像标准化;训练浮点模型,采用降低浮点数进行深度学习模型训练;微调量化卷积层,使用上一步骤得到的浮点模型,并将模型的路径添加到微调量化卷积脚本中进行微调,得到微调量化卷积层;应用线性变换微调模型,利用已标注的样本对预训练参数进行调整,以针对行人检测任务来修正模型网络;模型效果测试,利用测试集测试行人检测模型的准确率和召回率;在实时性和检测的灵活性上有较大的提高,同时具有高准确率和召回率等良好性能。