From ffe11b5255d600f88e07ab9cd42613069fda049f Mon Sep 17 00:00:00 2001 From: lee <770918727@qq.com> Date: Tue, 22 Jul 2025 13:30:42 +0800 Subject: [PATCH] =?UTF-8?q?=E9=A2=84=E8=AE=AD=E7=BB=83=E8=B0=83=E6=95=B4?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- demo.py | 2 +- ultralytics/cfg/default.yaml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/demo.py b/demo.py index b16f052..c17f78a 100644 --- a/demo.py +++ b/demo.py @@ -6,6 +6,6 @@ import numpy as np 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('/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[0].boxes.conf) diff --git a/ultralytics/cfg/default.yaml b/ultralytics/cfg/default.yaml index d8a2ecb..40b03a5 100644 --- a/ultralytics/cfg/default.yaml +++ b/ultralytics/cfg/default.yaml @@ -68,7 +68,7 @@ embed: # (list[int], optional) return feature vectors/embeddings from given laye # Visualize settings --------------------------------------------------------------------------------------------------- show: False # (bool) show predicted images and videos if environment allows 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_crop: False # (bool) save cropped images with results show_labels: True # (bool) show prediction labels, i.e. 'person'