From 5a3fc595735f85ffc59f165342005ea54796114a Mon Sep 17 00:00:00 2001 From: zhanghuan <1262329256@qq.com> Date: Wed, 10 Sep 2025 15:19:29 +0800 Subject: [PATCH] =?UTF-8?q?=E8=AE=AD=E7=BB=83=E6=8C=87=E6=A0=87=E6=9B=B2?= =?UTF-8?q?=E7=BA=BF=E5=9B=BE=E9=85=8D=E7=BD=AE=E6=88=90setting=E6=96=87?= =?UTF-8?q?=E4=BB=B6=E4=B8=AD=E7=9A=84=E5=8F=98=E9=87=8F=E3=80=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- settings/settings.py | 2 +- train/train_food_classifier.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/settings/settings.py b/settings/settings.py index 8e9337c..489c792 100644 --- a/settings/settings.py +++ b/settings/settings.py @@ -15,7 +15,7 @@ VAL_DATA_DIR = os.path.join(DATASET_DIR, 'val') TEST_DATA_DIR = os.path.join(DATASET_DIR, 'test') # 模型保存路径 -MODEL_DIR = os.path.join(BASE_DIR, 'model', '03') +MODEL_DIR = os.path.join(BASE_DIR, 'model', '04') BEST_MODEL_PATH = os.path.join(MODEL_DIR, 'best_food_model.pth') TRAINING_CURVES_PATH = os.path.join(MODEL_DIR, 'training_curves.png') diff --git a/train/train_food_classifier.py b/train/train_food_classifier.py index 968eaf3..7cdfb6b 100644 --- a/train/train_food_classifier.py +++ b/train/train_food_classifier.py @@ -246,7 +246,7 @@ if __name__ == '__main__': plt.grid(True) plt.tight_layout() - plt.savefig('../model/02/training_curves.png', dpi=300, bbox_inches='tight') + plt.savefig(settings.TRAINING_CURVES_PATH, dpi=300, bbox_inches='tight') plt.show() print(f'\n最终结果:')