From 6d253439897b931a894aff5daec37430f3a2748d Mon Sep 17 00:00:00 2001
From: zhangpu <1250681871@qq.com>
Date: Thu, 11 Dec 2025 08:52:46 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A4=84=E7=90=86=E4=B8=AD=E6=96=87=E6=98=BE?=
 =?UTF-8?q?=E7=A4=BA=E5=BC=82=E5=B8=B8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 SegFormer/scripts/2_visualize_data.py | 3 +++
 SegFormer/training/config.py          | 3 ++-
 2 files changed, 5 insertions(+), 1 deletion(-)

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