YOLOv7实验测试之一:遥感图像检测应用(YOLOv7.yaml)

      前期一直没有时间跑YOLOv7源码,今天对YOLOv7在NWPU-10遥感图像数据集上进行实验测试,现将实验训练以及评估过程分享如下,希望对大家有帮助。为了客观分享,我对整个过程进行了截图,能够让大家看到我的实验参数以及实验设置。

需要更多程序资料以及答疑欢迎大家关注——微信公众号:人工智能AI算法工程师 

一、训练过程:选择YOLOv7.yaml配置文件,具体超参数以及实验过程如下所示

YOLOv7实验测试之一:遥感图像检测应用(YOLOv7.yaml)_第1张图片

YOLOv7实验测试之一:遥感图像检测应用(YOLOv7.yaml)_第2张图片

YOLOv7实验测试之一:遥感图像检测应用(YOLOv7.yaml)_第3张图片

YOLOv7实验测试之一:遥感图像检测应用(YOLOv7.yaml)_第4张图片

YOLOv7实验测试之一:遥感图像检测应用(YOLOv7.yaml)_第5张图片

YOLOv7实验测试之一:遥感图像检测应用(YOLOv7.yaml)_第6张图片

YOLOv7实验测试之一:遥感图像检测应用(YOLOv7.yaml)_第7张图片

二、评估过程:评估将iou_thres=0.5,得到[email protected]的值。

YOLOv7实验测试之一:遥感图像检测应用(YOLOv7.yaml)_第8张图片

YOLOv7实验测试之一:遥感图像检测应用(YOLOv7.yaml)_第9张图片

YOLOv7实验测试之一:遥感图像检测应用(YOLOv7.yaml)_第10张图片

YOLOv7实验测试之一:遥感图像检测应用(YOLOv7.yaml)_第11张图片

YOLOv7实验测试之一:遥感图像检测应用(YOLOv7.yaml)_第12张图片

三、结论

    训练过程用了1.2个小时,得到的权值文件为285MB,mAP为89.3%,网络层数407,参数量3652891。与YOLOv5s对比来看,YOLOv5s(5.0版本)在nwpu-10数据集上mAP可以达到91.7%,实验设置几乎一致的前提下。可能数据集650张图片样本量过小,YOLOv7过于复杂,造成过拟合,反而效果不如YOLOv5s。

预告一下:该系列下一篇将分享YOLOv7其他的轻量化的配置文件。

如果觉得对大家有帮助,欢迎点赞收藏关注,我会继续给大家做实验提供参考。有问题也欢迎私信我。

需要更多程序资料以及答疑欢迎大家关注——微信公众号:人工智能AI算法工程师 

你可能感兴趣的:(YOLOv7,深度学习,机器学习,人工智能)