YOLOV8显示结果坐标和实现类别计数
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\*
效果展示: