一种增强小目标检测与特征提取能力的图像识别方法

本发明公开了一种增强小目标检测与特征提取能力的图像识别方法，该方法包括下述步骤：获取苹果图像数据，对苹果图像数据进行数据增强；对数据增强后的苹果图像数据进行数据预处理；将数据预处理后的苹果图像数据划分为训练集、验证集和测试集；构建苹果图像识别网络模型，将YOLOv5网络模型SPP模块之前的C3模块为C2f模块；获取训练集，通过NWD损失函数对苹果图像识别网络模型进行训练，得到训练后的苹果图像识别网络模型；基于测试集对苹果图像识别网络模型进行测试，输出图像识别的准确率；基于训练后的苹果图像识别网络模型得到预测的图像识别结果。本发明提升了对小目标的检测精度，同时增强了特征提取能力。