最后选择了kaggle网站中的火灾数据集,加上flame跟flame2里面选取了一些具有代表性图像进行标注,
选取的图像也进行数据增强,放到训练数据集里
接下来以YOLOv8s为baseline进行改进,改进3点,然后与YOLOv9,v10,v5 进行算法对比
注意力机制还可以选择两三个进行对比然后挑选出效果最好的一个再进行进一步的消融实验
生成曲线对比图
最后选择了kaggle网站中的火灾数据集,加上flame跟flame2里面选取了一些具有代表性图像进行标注,
选取的图像也进行数据增强,放到训练数据集里
接下来以YOLOv8s为baseline进行改进,改进3点,然后与YOLOv9,v10,v5 进行算法对比
注意力机制还可以选择两三个进行对比然后挑选出效果最好的一个再进行进一步的消融实验
生成曲线对比图