diff --git a/SegFormer/scripts/2_visualize_data.py b/SegFormer/scripts/2_visualize_data.py index 17ad20f..d4f0fbe 100644 --- a/SegFormer/scripts/2_visualize_data.py +++ b/SegFormer/scripts/2_visualize_data.py @@ -19,6 +19,9 @@ import matplotlib.pyplot as plt from pathlib import Path import random +# 设置matplotlib中文字体 +plt.rcParams['font.sans-serif'] = ['SimHei', 'Microsoft YaHei', 'Arial Unicode MS'] # 用来正常显示中文标签 +plt.rcParams['axes.unicode_minus'] = False # 用来正常显示负号 class DatasetVisualizer: """数据集可视化工具""" diff --git a/SegFormer/training/config.py b/SegFormer/training/config.py index 025fdf7..559d42d 100644 --- a/SegFormer/training/config.py +++ b/SegFormer/training/config.py @@ -272,7 +272,8 @@ def get_cpu_config() -> TrainConfig: config.image_size = 256 # 降低分辨率加快训练 # 快速验证配置 - config.num_epochs = 20 + # config.num_epochs = 20 + config.num_epochs = 50 config.eval_every_n_epochs = 5 config.save_every_n_epochs = 10