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

img

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

我很久以前遇到过这个问题,下面是我的解决方案。
可以尝试使用 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() 函数中设置对应的字体名称。

试试楼上的这个字体设置吧,不然其他基本上都得用英文,类别名这个东西改下就行,对于网络影响不大。