处理中文显示异常

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:
     """数据集可视化工具"""

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