YuXin_Liu/丝束检测/predictTow.py

8 lines
390 B
Python

from ultralytics import YOLO
model = YOLO("/mnt/d/fiber/yolov8/ultralytics-main/ultralytics-main/runs/detect/train/weights/best.pt")
model.predict(source="/mnt/d/fiber/droneData/train/images", save=True, save_conf=True,
imgsz=1024,
conf=0.001, hide_labels=True, classes=0)
# D:\fiber\yolov8\ultralytics-main\ultralyti
# cs-main\runs\detect\train\weights\best.pt