科创中国
旋转船只目标检测模型的训练方法、训练装置和存储介质
发布时间: 2023-11-01
基本信息
合作方式：
技术服务
成果类型：
发明专利
行业领域：
物理
成果介绍
本发明公开了一种旋转船只目标检测模型的训练方法、训练装置和存储介质。该训练方法包括：对原始旋转船只图像进行预处理和角度标注处理，获得船只标注图像；根据上一轮训练之后的旋转损失情况设定当前训练轮的旋转触发概率，按照所述船只标注图像进行数据增强处理，获得船只增强图像；将所述船只增强图像输入到待训练的旋转船只目标检测模型中，得到预测值；根据所述预测值和所述船只标注图像对应的真实值计算当前训练轮的损失函数值；根据所述损失函数值更新所述待训练的旋转船只目标检测模型的模型参数，完成当前轮训练。通过loss反馈在训练过程中进行动态数据增强，改善了样本不均衡问题，使得数据增强有了针对性，提高了数据增强的效率。
成果亮点
.一种旋转船只目标检测模型的训练方法，其特征在于，所述训练方法包括：
对原始旋转船只图像进行预处理和角度标注处理，获得船只标注图像；
根据上一轮训练之后的旋转损失情况设定当前训练轮的旋转触发概率，按照所述船只标注图像进行数据增强处理，获得船只增强图像；
将所述船只增强图像输入到待训练的旋转船只目标检测模型中，得到预测值；
根据所述预测值和所述船只标注图像对应的真实值计算当前训练轮的损失函数值；
根据所述损失函数值更新所述待训练的旋转船只目标检测模型的模型参数，完成当前轮训练。
团队介绍
中国科学院深圳先进技术研究院提升了粤港地区及我国先进制造业和现代服务业的自主创新能力，推动我国自主知识产权新工业的建立，成为国际一流的工业研究院。 深圳先进院目前已初步构建了以科研为主的集科研、教育、产业、资本为一体的微型协同创新生AC态系统，由九个研究平台，国科大深圳先进技术学院，多个特色产业育成基地、多支产业发展基金、多个具有独立法人资质的新型专业科研机构等组成。开展先进技术研究，促进科技发展。信息、电子、通讯技术研究新材料、新能源技术研究高性能计算、自动化、精密机械研究生物医学与医疗仪器研究相关学历教育、博士后培养与学术交流。
成果资料
产业化落地方案
下载app