基于Resnet101等CNN卷积神经网络,学习蝴蝶图像训练集,算法通过对蝴蝶翅膀纹理颜色等的自学习实现对蝴蝶图片中蝴蝶的种类识别。数据集大小为2248张图片,其中训练集,测试集各1124张,包括国内外34种蝴蝶种类。模型加入了通道注意力模块和空间注意力模块提升识别性能,最终图像识别准确率高达97.31%。
评论