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-07-22 13:30:42 +08:00
parent ed6a4144fa
commit ffe11b5255
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
demo.py
View File

@ -6,6 +6,6 @@ import numpy as np
model = YOLOv10('ckpts/20250630/best_gift_v10n.pt') model = YOLOv10('ckpts/20250630/best_gift_v10n.pt')
# result = model.predict('./data/bandage.jpg', save=True, imgsz=[224, 224], conf=0.1) # 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/gift', 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)
print(result) print(result)
# print(result[0].boxes.conf) # print(result[0].boxes.conf)

2
ultralytics/cfg/default.yaml
View File

@ -68,7 +68,7 @@ embed: # (list[int], optional) return feature vectors/embeddings from given laye
# Visualize settings --------------------------------------------------------------------------------------------------- # Visualize settings ---------------------------------------------------------------------------------------------------
show: False # (bool) show predicted images and videos if environment allows show: False # (bool) show predicted images and videos if environment allows
save_frames: False # (bool) save predicted individual video frames save_frames: False # (bool) save predicted individual video frames
save_txt: False # (bool) save results as .txt file save_txt: True # (bool) save results as .txt file
save_conf: False # (bool) save results with confidence scores save_conf: False # (bool) save results with confidence scores
save_crop: False # (bool) save cropped images with results save_crop: False # (bool) save cropped images with results
show_labels: True # (bool) show prediction labels, i.e. 'person' show_labels: True # (bool) show prediction labels, i.e. 'person'