科研之友
微信
新浪微博
Facebook
分享链接
摘要
本发明公开了一种基于知识蒸馏和半监督学习的船舶快速检测方法,包括：1)利用已有船舶数据与需要部署的海域采集到的船舶数据构建船舶数据集,并对数据集中的部分数据进行船舶目标的标注,根据先验框滑动生成训练样本并进行数据增强；2)构建教师网络,先利用标注样本进行预训练,接着联合无标注样本进行半监督训练以实现船舶检测；3)构建轻量级学生网络,利用已训练好的教师网络指导学生网络对标注样本与无标注样本实现知识蒸馏；4)将训练好的学生网络模型转化为ONNX格式,采用OpenVINO进一步优化和部署学生网络,实现快速的船舶检测。本发明将精度高容量大的教师网络所学习到的知识迁移到轻量级学生网络中,在保证检测精度同时,实现快速的船舶检测。
- 单位
