mirror of
https://gitee.com/nanjing-yimao-information/ieemoo-ai-gift.git
synced 2025-08-18 21:30:25 +00:00
update
This commit is contained in:
10
demo.py
Normal file
10
demo.py
Normal file
@ -0,0 +1,10 @@
|
||||
from ultralytics import YOLOv10
|
||||
import numpy as np
|
||||
# model = YOLOv10.from_pretrained('jameslahm/yolov10{n/s/m/b/l/x}')
|
||||
# or
|
||||
# wget https://github.com/THU-MIG/yolov10/releases/download/v1.1/yolov10{n/s/m/b/l/x}.pt
|
||||
model = YOLOv10('ckpts/20250514/best_gift_v10n.pt')
|
||||
|
||||
result = model.predict('./data/bandage.jpg', save=False, imgsz=[224, 224], conf=0.1)
|
||||
print(result)
|
||||
print(result[0].boxes.conf)
|
Reference in New Issue
Block a user