目录
❗❗❗ 兄弟姐妹们,如果看习惯了运行train.py
时打印0.551
这种的不超过1的小数点值
,一定要把以下操作全部还原哟!
❗❗❗ 不然就会像我一样,当看到打印台那一行行的P、R、mAP50、mAP50-95
的超过1的值
,心慌的不得了!错以为是自己数据集的问题,毁我心情,耽误我好几天不跑代码!
❗❗❗ 不过它只是影响的打印台中的显示值,不影响tensorboard中画曲线的值
哈,也不影响result.txt
中的值!
1. 最终打印效果
2. 做两处更改
2.1 修改map显示,在ultralytics-main/ultralytics/yolo/v8/detect/val.py
中操作
首先,按下Ctrl+F
检索定位到print_results
处,然后如下操作:
- 如图标号1,将
11.3
改为11.4
- 如图标号2,注释掉源代码,加入以下代码
LOGGER.info(pf % ('all', self.seen, self.nt_per_class.sum(), self.metrics.mean_results()[0] * 100, self.metrics.mean_results()[1] * 100, self.metrics.mean_results()[1] * 100, self.metrics.mean_results()[3] * 100))
- 如图标号3,注释掉源代码,加入以下代码
LOGGER.info(pf % (self.names[c], self.seen, self.nt_per_class[c], self.metrics.class_result(i)[0] * 100, self.metrics.class_result(i)[1] * 100, self.metrics.class_result(i)[2] * 100, self.metrics.class_result(i)[3] * 100))
2.2 打印FPS,在ultralytics-main/ultralytics/yolo/engine/validator.py
中操作
首先,按下Ctrl+F
检索定位到LOGGER.info('Speed
处,然后如下操作:
然后,如图添加以下代码:
speedvalues_list = [round(x, 1) for x in list(self.speed.values())]
print(f'FPS: {round(1000 / sum(speedvalues_list), 2)} 帧/s')
3. (更新)解决更改后达不到本博客相同效果
- 另一篇关于v8打印的文章【YOLO系列PR、F1绘图】更改v5、v7、v8(附v8训练、验证方式),实现调用val.py或者test.py后生成pr.csv,然后再整合绘制到一张图上(使用matplotlib绘制),也有同学反应这个情况,用下图的方式去更改应该有用
- 关注的是那篇博客的这个位置👉2.1.3 YOLOv8的更改(附训练、验证方式)
- 为什么用这种验证方式呢?是因为v8的一些bug吧,哈哈哈,反正我训练、验证都无法用命令直接启动,都是参考的博客YoloV8的python启动用
py文件
运行的
# 参考博客:https://blog.csdn.net/ljlqwer/article/details/129175087
from ultralytics import YOLO
model = YOLO("/opt/data/private/user_LWD/train_result/yolov8s/yolov8s-best.pt") # 权重地址
results = model.val(data="ultralytics/datasets/RDD.yaml", imgsz=640, split='test', batch=1, conf=0.001, iou=0.5, name='yolov8s-from-ultralytics-main-bs1', optimizer='Adam') # 参数和训练用到的一样