刚开始使用 Python / YoloV8。我设法创建了一个自定义模型,现在正在用它进行测试。你能帮一个n00b吗?
我在图像大小 1024 上训练它:
model.train(data='data.yaml', epochs=200, imgsz=1024, device=0)
我使用此代码开始预测(Windows 上的 Python):
import os
import torch
from ultralytics import YOLO
model = YOLO(r"C:\yolo\License Plates.v3i.yolov8-v3\best.pt")
#For video
model.predict(source=r"C:\yolo\testvideo.mp4", show=True)
#For image
model.predict(source=r"C:\yolo\testimag.jpg", show=True)
我遇到的问题:
我希望你能回答我的基本问题!预先感谢。
你找到答案了吗?你能分享吗?