YOLOV8显示结果坐标和实现类别计数

发布时间 2023-05-05 10:46:44作者: 不做签到员

YOLOV8显示结果坐标和实现类别计数

​ 本来5.1号的时候就想发出来的,结果阳了,躺了三天,今天好点了!!!5.1那天有个水友问我,YOLOV8如何实现结果检验框的坐标显示和类别计数,我说YOLOV8和YOLOV5一样都是ultralytics的产物,修改代码的原理是一样的,但是值得注意的是如果你直接利用的是pip install ultralytics进行安装的,此时你修改你在github上下载好的YOLOV8源码是没有任何效果的。

要到你的pip所对应的python环境下去寻找这里面去修改,例如我的ultralytics包在:

D:\python_Develop\anaconda3\Lib\site-packages\ultralytics-8.0.58-py3.10.egg\ultralytics

此时我想加入坐标显示和类别计数,只需要修改predict.py

D:\python_Develop\anaconda3\Lib\site-packages\ultralytics-8.0.58-py3.10.egg\ultralytics\yolo\v8\detect\predict.py

具体的修改如下:

结果保存,输入命令进行测试:

yolo 
task=detect 
mode=predict 
model=runs\detect\train2\weights\last.pt
source=D:\python_Develop\Yolo_Project\Data\Mask_Data\test\images\* 

效果展示: