摘要
本发明公开了一种小样本图像扩增方法。该方法的步骤包括:获取多种类原始图像数据集;获取小样本目标图像数据集;对所述多种类原始图像数据集进行预处理,得到种类相关的编辑向量;根据所述若干张小样本目标图像、种类相关的编辑向量,对所述离散码本进行一阶段训练,得到训练好的离散码本;对建立的代码预测模块训练,得到训练好的代码预测模块;带扩增的小样本图像和提示词输入所述训练好的代码预测模块,得到索引序列;将所述索引序列在训练好的离散码本中进行匹配,获得索引序列在训练好的离散码本中对应位置的重建向量;将所述重建向量生成对应小样本图像的扩增图像。该扩增方法可以同时生成多种类,每个种类多张相似的图像。