18 lines
444 B
Python
18 lines
444 B
Python
from ultralytics import YOLO
|
|
|
|
# 加载训练好的模型
|
|
model = YOLO('run/train/weights/best.pt')
|
|
|
|
# 在验证集上进行推理
|
|
results = model.predict(
|
|
source='datasets/valid/images/-10-1_mp4-1_jpg.rf.fb00c51bb30dded64f20f66f13047b9d.jpg', # 替换为您的验证集图像路径或单张图片路径
|
|
imgsz=640,
|
|
conf=0.25,
|
|
iou=0.45,
|
|
save=True,
|
|
save_txt=True,
|
|
save_conf=True,
|
|
project='run',
|
|
name='train'
|
|
)
|