一种基于改进的YOLOv5模型的船舱火灾图像识别方法

本发明公开了一种基于改进的YOLOv5模型的船舱火灾图像识别方法，属于火灾图像识别技术领域。方法包括采集和预处理火灾图像数据，增加小目标检测头和CBAM注意力模块，改进损失函数为WIoU。通过训练集的迭代训练，调整模型参数至最优，并在验证集和测试集上评估模型性能。评价指标包括平均准确率mAP、模型大小和复杂度。该方法提高了火灾图像识别的准确性和模型效率。