将分类的类别也应用在训练程序中,更好的解耦。

This commit is contained in:
zhanghuan
2025-09-12 11:20:46 +08:00
parent da67cb9f4c
commit d73331010c
4 changed files with 10 additions and 9 deletions
+2 -2
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', '10')
MODEL_DIR = os.path.join(BASE_DIR, 'model', '12')
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')
@@ -47,7 +47,7 @@ COLOR_JITTER_SATURATION = 0.2
COLOR_JITTER_HUE = 0.1
# 模型参数
NUM_CLASSES = 3
NUM_CLASSES = 4
# 其他配置
NUM_WORKERS = 0 # Windows下建议设为0