lichen/ieemoo-ai-gift
1
0
Fork 0
mirror of https://gitee.com/nanjing-yimao-information/ieemoo-ai-gift.git synced 2025-08-18 21:30:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

预训练调整

Browse Source
This commit is contained in:
lee
2025-08-05 14:27:30 +08:00
parent ffe11b5255
commit d5aee7b936
2 changed files with 78 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
demo.py
View File

@ -3,7 +3,7 @@ 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/20250630/best_gift_v10n.pt')
model = YOLOv10('ckpts/20250701/best_gift_v10n.pt')
# result = model.predict('./data/bandage.jpg', save=True, imgsz=[224, 224], conf=0.1)
result = model.predict('/home/lc/data_center/gift/trace_subimgs/predict_actual_test/test', save=True, imgsz=[224, 224], conf=0.1)