摘要
本发明公开了一种小样本图像扩增方法。该方法的步骤包括:获取多种类原始图像数据集;获取小样本目标图像数据集;对所述多种类原始图像数据集进行预处理,得到种类相关的编辑向量;根据所述若干张小样本目标图像、种类相关的编辑向量,对所述离散码本进行一阶段训练,得到训练好的离散码本;对建立的代码预测模块训练,得到训练好的代码预测模块;带扩增的小样本图像和提示词输入所述训练好的代码预测模块,得到索引序列;将所述索引序列在训练好的离散码本中进行匹配,获得索引序列在训练好的离散码本中对应位置的重建向量;将所述重建向量生成对应小样本图像的扩增图像。该扩增方法可以同时生成多种类,每个种类多张相似的图像。
技术关键词
图像扩增方法
原始图像数据
图像嵌入
样本
编辑
词嵌入向量
输入端
索引
图像编码器
模块
图像生成器
序列
解码
注意力
阶段
扩增系统
系统为您推荐了相关专利信息
图像特征提取
样本
深度学习模型
矩阵
图像分类方法
铁路
钢箱梁桥顶板
钢箱梁顶板
应力
概率密度曲线
图像去雾方法
三次样条函数
曲线
样本
三次样条插值法