yixiong/FoodClassifier
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

菜品识别程序，增加向量输出到控制台的功能！

Browse Source
This commit is contained in:
zhanghuan
2025-09-11 13:39:49 +08:00
parent d9057edbd9
commit adefc3ecb2
4 changed files with 44 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
settings/settings.py
+3 -4
View File
@@ -15,17 +15,16 @@ 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', '05')
MODEL_DIR = os.path.join(BASE_DIR, 'model', '06')
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')
# INFERENCE_BEST_MODEL_PATH = os.path.join(BASE_DIR, 'model', '03','best_food_model.pth')
# INFERENCE_BEST_MODEL_PATH = os.path.join(BASE_DIR, 'model', '05','best_food_model.pth')
INFERENCE_BEST_MODEL_PATH = BEST_MODEL_PATH
# 训练参数
# NUM_EPOCHS = 100
NUM_EPOCHS = 13
NUM_EPOCHS = 100
BATCH_SIZE = 32
LEARNING_RATE = 0.001
WEIGHT_DECAY = 1e-4