返回搜索
档案
一种小图像识别模型的生成方法、装置、设备及介质
发明专利有效
10权利要求 · 4 独立
§ Ⅰ
卷宗概要
发明人
庄思成; 张健伟; 付清喜; 李涛; 张立振; 乔阳; 赵扬波; 黄恩
IPC 分类
G06V 10/774 (2022.01)G06V 10/82 (2022.01)G06V 10/764 (2022.01)G06V 10/32 (2022.01)G06N 3/0464 (2023.01)
本发明公开了一种小图像识别模型的生成方法、装置、设备及介质。包括:对原始样本图像进行尺寸调整,生成标准图像,并根据原始样本图像的数量以及标准图像,生成训练样本集;确定当前轮次的训练批次大小,并根据训练批次大小,在训练样本集中选取至少一个目标分组,作为当前训练集合;在当前轮次训练完成之后,获取当前轮次的关键参数集合,并根据关键参数集合以及预设的容忍度系数,判断是否满足提前停止条件;若不满足提前停止条件,则根据卷积神经网络的结构参数以及当前轮次的关键参数集合,对模型参数集合进行更新,并将更新后的模型参数集合用于下一轮次训练。采用上述技术方案,能够实现在样本数量有限的情况下,高效训练小图像识别模型。