增加了数据增强离线的程序(可扩充数据集),将图片压缩从32*32,调整为224*224.

This commit is contained in:
zhanghuan
2025-09-15 17:26:14 +08:00
parent d73331010c
commit 264d01e9b9
5 changed files with 80 additions and 14 deletions
+2 -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', '12')
MODEL_DIR = os.path.join(BASE_DIR, 'model', '15')
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')
@@ -26,6 +26,7 @@ INFERENCE_BEST_MODEL_PATH = BEST_MODEL_PATH
# 训练参数
NUM_EPOCHS = 100
BATCH_SIZE = 32
# BATCH_SIZE = 128
LEARNING_RATE = 0.001
WEIGHT_DECAY = 1e-4