[发明专利]一种图像数据集扩展方法、系统及存储介质

摘要

本发明公开了一种图像数据集扩展方法、系统及存储介质，涉及图像数据集扩展技术领域，首先获取现有的图像‑文本数据集，输入stable‑diffusion模型中进行训练，得到训练好的stable‑diffusion模型；将目标文本输入训练好的stable‑diffusion模型中，得到生成图像集；对所述生成图像集进行筛选，得到正例扩充图像集和反例扩充图像集；将所述正例扩充图像集和反例扩充图像集与所述图像‑文本数据集进行合并，得到新的图像‑文本数据集。本发明仅需要少量的原始数据集，就可以创造性的生成新的图像数据集，可以应用在多个图像数据集扩充领域，如医学影像数据集扩充领域，食物图像数据集扩充领域等。