修改测试

This commit is contained in:
lee
2025-07-01 13:39:34 +08:00
parent 96a9445761
commit ed6a4144fa
8 changed files with 55 additions and 14 deletions

2
val.py
View File

@ -1,4 +1,4 @@
from ultralytics import YOLOv10
model = YOLOv10('/home/lc/ieemoo-ai-gift/runs/detect/train/weights/best_gift_v10n.pt')
model = YOLOv10('./ckpts/20250701/best_gift_v10n.pt')
metrics = model.val(batch=1, data='gift.yaml', imgsz=224)