摘要
本发明公开了一种基于主动学习训练样本的AI模型闭环迭代训练方法和系统。从样本图片集中选定部分图片标注得到标注数据集,将标注数据集划分为训练数据集和测试数据集;利用标注数据集对深度学习模型进行训练得到初始训练模型;利用初始训练模型从样本图片集中挑选部分图片标注并合入至训练数据集中;利用更新后的训练数据集及测试数据集对初始训练模型进行更新;反复执行更新训练数据集和更新初始训练模型的操作,直到训练数据集的规模达到预设数量或模型性能达到要求。本发明通过主动学习选择部分样本进行标注,避免对整个数据集进行标注造成的时间和资源浪费,适用于需大规模数据集进行训练的AI模型,能处理海量数据并快速提升模型性能。
技术关键词
迭代训练方法
图片
语义分割模型
训练系统
数据
矩阵
闭环
深度学习模型
样本
模块
像素点
模型更新
元素
规模
资源
参数