基于稳定扩散模型的苗绣图像分类研究
张银建1
杨邦勤1
陈研2
吴菁1
1.贵州民族大学 贵州贵阳 5500002.华南理工大学 广东广州 510641
摘要：针对苗绣图像分类中图片数据量少导致特征提取质量较低的问题,引入稳定扩散模型(stable diffusion)的"图生图"方法作为一种数据增强手段.在此数据增强的基础上,针对卷积神经网络在图像分类中特征提取不充分且贡献度不明确的问题,基于残差网络ResNet-18 结构引入多维度动态卷积(omni-dimensional dynamic convolution,ODConv)和多维协同注意力模块(multidimensional collaborative attention module,MCA)构建了基于稳定扩散模型的苗绣图像分类模型.通过在自建数据集上进行对比,实验结果表明,与不进行数据增强且单纯使用原始残差网络相比,文章所提方法可有效提高模型分类性能.
关键词：苗绣扩散模型数据增强残差网络图像分类
资助基金：贵州民族大学博士科研启动项目(GZMUZK(2024)QD11)
论文发表日期：2024-12-28
在线出版日期：2026-05-22（本平台首次上网日期，不代表文献的发表时间）
页数：8( 5-12 )