增加了数据增强离线的程序(可扩充数据集),将图片压缩从32*32,调整为224*224.
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user