基于SSD-Mobilenet模型的目标检测
刘颜
朱志宇
张冰
1.江苏科技大学电子信息学院 镇江 2120032.江苏科技大学电子信息学院 镇江 2120033.江苏科技大学电子信息学院 镇江 212003
摘要：为缩短在目标检测时模型的训练时间,加快网络的收敛速度,论文采取了一种卷积神经网络模型与迁移学习相结合的方法.SSD-Mobilenet模型结合了SSD网络检测速度快和Mobilenet轻量型网络占用内存少的优点.首先,使用COCO数据集将SSD-Mobilenet模型进行预训练,得到模型参数、瓶颈描述因子,然后用Pets数据集重新训练网络的全连接层,运用迁移学习的思想,可以实现在较少数据集的情况下,通过短时间的训练模型即可收敛.实验结果显示总训练时长约为11个小时,检测准确率达到74.5%.由此表明,采用SSD-Mobilenet模型与迁移学习相结合的方法可以缩短模型训练时间、加速模型的收敛速度且检准率比较高.
关键词：目标检测卷积神经网络SSD-Mobilenet迁移学习
分类号：TP242.6(自动化技术及设备)
论文发表日期：2019-01-01
在线出版日期：2025-08-15（本平台首次上网日期，不代表文献的发表时间）
页数：5( 52-56 )
英文信息