我一直在尝试创建一个物体识别软件,并且一直从一个名为 PySource 的 Youtuber 那里复制过来(这里是视频:https://www.youtube.com/watch?v=kEcWUZ8unmc)。我对物体识别和 YOLO 非常陌生,所以我不知道为什么找不到 volov8m.pt。我已经从 ultralytics import YOLO 导入,问题发生在
model = YOLO("volov8m.pt")
我正在使用 M1 Macbook Air 2020 和 PyCharm(如果有帮助的话)。
以下是完整代码供参考:
import cv2
import numpy as np
from ultralytics import YOLO
import torch
print(torch.backends.mps.is_available())
# Opencv DNN
cap = cv2.VideoCapture("dogs.mp4")
model = YOLO("volov8m.pt")
while True:
ret, frame = cap.read()
if not ret:
break
results = model(frame, device="mps")
result = results[0]
bboxes = np.array(result.boxes.xyxy.cpu(), dtype="int")
classes = np.array(result.boxes.cls.cpu(),dtype="int")
for cls, bbox in zip(classes, bboxes):
(x,y,x2,y2) = bbox
cv2.rectangle(frame,(x,y),(x2,y2),(0,0,225),2)
cv2.putText(frame,str(cls),(x,y,-5),cv2.FONT_HERSHEY_PLAIN, 2,(0,0,225),2)
cv2.imshow("Img",frame)
cv2.waitKey(1)
if key == 27:
break
cap.release()
cv2.destroyAllWindows()
我不知道如何解决这个问题
你打错字了,你应该使用
model = YOLO("yolov8m.pt)
而不是model = YOLO("volov8m.pt")
。