将分类的类别也应用在训练程序中,更好的解耦。
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', '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
|
||||
|
||||
Reference in New Issue
Block a user