一种基于形状大模型的样本增强方法

本发明涉及一种基于形状大模型的样本增强方法，包括以下步骤，S1、将训练集的标注图像输入到形状大模型结合扩散网络构成的数据增强网络中；S2、迭代测试训练，训练完成后，使用测试集输入到网络中生成样本，计算生成的形状样本与测试集的输入图像IOU交并比；S3、测试成功后，则应用网络来作数据增强；S4、使用透视变换矩阵对标注图像对应的原图作透视变换，得到增强样本。本发明通过训练出基于形状的大模型，再结合文生图扩散模型，通过新的形状来生成新的样本，使生成的样本多样且逼真。获得了质量更高的增强样本，提升了训练数据的质量，从而提高了模型效果的上限。