YOLO V5的metircs.py没法显示中文图例怎么办

Question

这个问题怎么解决呢，中文标签的，就是这个显示有问题，yolov5源代码

Answer 1

我很久以前遇到过这个问题，下面是我的解决方案。
可以尝试使用 Matplotlib 中文显示的方式。步骤如下：

1. 安装 SimSun 字体：打开 Windows 的字体文件夹，复制 simsun.ttc 字体文件到 ~/.matplotlib/fonts/ 目录下（如果该目录不存在，则需要手动创建）。
2. 在 metrics.py 文件中添加以下代码：

   import matplotlib.pyplot as plt
   plt.rcParams['font.family'] = ['SimSun']  # 设置中文字体
   

3. 修改 plot_results() 函数中的图例标签，将英文标签改为中文标签，例如：

   plt.plot(pr, rc, linewidth=2, label='PR')
   plt.plot(fp, tp, linewidth=2, label='TP')
   plt.xlabel('召回率', fontsize=15)
   plt.ylabel('精确率', fontsize=15)
   plt.title('PR 曲线', fontsize=20)
   plt.legend(['精确率/召回率'], loc='upper right', fontsize=12)  # 设置中文标签
   

这里需要注意的是如果使用的是 Linux 或 macOS 等操作系统，需要根据对应操作系统的字体文件名修改第一步中的字体文件名，并在 plot_results() 函数中设置对应的字体名称。