Python:FileNotFoundError:[Errno 2]没有这样的文件或目录:'volov8m.pt'

问题描述 投票:0回答:1

我一直在尝试创建一个物体识别软件,并且一直从一个名为 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()

我不知道如何解决这个问题

python yolo
1个回答
0
投票

你打错字了,你应该使用

model = YOLO("yolov8m.pt)
而不是
model = YOLO("volov8m.pt")

© www.soinside.com 2019 - 2024. All rights reserved.