diff --git a/fonts/simhei.ttf b/fonts/simhei.ttf new file mode 100644 index 0000000..5bd4687 Binary files /dev/null and b/fonts/simhei.ttf differ diff --git a/utils/common.py b/utils/common.py index 44051b6..ac4a984 100644 --- a/utils/common.py +++ b/utils/common.py @@ -10,13 +10,18 @@ from sklearn.metrics import precision_score, recall_score, f1_score from sklearn.utils.class_weight import compute_class_weight import logging import matplotlib.pyplot as plt +from matplotlib.font_manager import FontProperties # 控制是否打印的宏定义 PRINT_LOG = True -# 确保 matplotlib 支持中文 -plt.rcParams['font.sans-serif'] = ['SimHei'] # 使用黑体 -plt.rcParams['axes.unicode_minus'] = False # 解决保存图像时负号 '-' 显示为方块的问题 +# 指定字体路径 +font_path = os.path.join(os.path.dirname(__file__), 'fonts', 'simhei.ttf') +font_prop = FontProperties(fname=font_path) + +# 设置 matplotlib 支持中文 +plt.rcParams['font.sans-serif'] = [font_prop.get_name()] +plt.rcParams['axes.unicode_minus'] = False def log_print(message): logging.info(message)