在YOLOv8中获取预测的类名

Question

我想在我的Python脚本中获取类数据， 我测试了这段代码，但我有一个问题：

from ultralytics.yolo.engine.model import YOLO

model = YOLO("path/to/best.pt")
results = model.predict(source=0, save=True, conf=0.5, show=True)
names = model.names

for r in results:
for c in r.boxes.cls:
print(names[int(c)])


**output:**

Traceback (most recent call last):
File "C:....project\\2.py", line 1, in \<module\>
from ultralytics.yolo.engine.model import YOLO
ModuleNotFoundError: No module named 'ultralytics.yolo'

Answer 1

请参阅https://stackoverflow.com/a/76196856/17233687，它似乎提供了解决方案：

检查您是否已安装 ultralytics 使用：

pip show ultralytics

或：

pip list

如果未安装，请使用以下命令继续安装：

pip install ultralytics

对于特定版本：

pip install ultralytics==X.X.X