在模型中增加预处理步骤!

This commit is contained in:
zhanghuan
2025-09-11 13:52:36 +08:00
parent adefc3ecb2
commit ee829dc8c9
2 changed files with 44 additions and 2 deletions
+1 -1
View File
@@ -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', '06')
MODEL_DIR = os.path.join(BASE_DIR, 'model', '07')
BEST_MODEL_PATH = os.path.join(MODEL_DIR, 'best_food_model.pth')
TRAINING_CURVES_PATH = os.path.join(MODEL_DIR, 'training_curves.png')
TRAINING_RESULTS_PATH = os.path.join(MODEL_DIR, 'training_results.txt')