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

训练指标曲线图配置成setting文件中的变量。

Browse Source
This commit is contained in:
zhanghuan
2025-09-10 15:19:29 +08:00
parent 2412c26342
commit 5a3fc59573
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
settings/settings.py
+1 -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') TEST_DATA_DIR = os.path.join(DATASET_DIR, 'test')
# 模型保存路径 # 模型保存路径
MODEL_DIR = os.path.join(BASE_DIR, 'model', '03') MODEL_DIR = os.path.join(BASE_DIR, 'model', '04')
BEST_MODEL_PATH = os.path.join(MODEL_DIR, 'best_food_model.pth') BEST_MODEL_PATH = os.path.join(MODEL_DIR, 'best_food_model.pth')
TRAINING_CURVES_PATH = os.path.join(MODEL_DIR, 'training_curves.png') TRAINING_CURVES_PATH = os.path.join(MODEL_DIR, 'training_curves.png')
train/train_food_classifier.py
+1 -1
View File
@@ -246,7 +246,7 @@ if __name__ == '__main__':
plt.grid(True) plt.grid(True)
plt.tight_layout() plt.tight_layout()
plt.savefig('../model/02/training_curves.png', dpi=300, bbox_inches='tight') plt.savefig(settings.TRAINING_CURVES_PATH, dpi=300, bbox_inches='tight')
plt.show() plt.show()
print(f'\n最终结果:') print(f'\n最终结果:')