将YOLOv5的检测结果在PyCharm终端中以中文形式输出

Question

目前用PyCharm在跑YOLOv5代码，目前训练权重都有了，该怎么在将检测结果在PyCharm的终端中输出这个图里的结果啊

Answer 1

引自免费微信小程序：皆我百晓生

要实现在PyCharm终端中以中文形式输出YOLOv5的检测结果，你需要修改YOLOv5的推理脚本，使其不仅显示图像上的检测结果，还将其打印到控制台。这里提供一种方法，通过修改YOLOv5的源代码（通常位于detect.py文件）来实现这一目标。

首先，确保你的环境支持中文输出，这通常涉及到编码设置。对于Python，确保你的脚本使用utf-8编码，并且你的终端也支持这一编码。

接下来，打开YOLOv5中的detect.py文件，找到处理检测结果的部分。在YOLOv5中，这部分通常是在save_one_box()函数被调用之后，检测结果被收集并可能存储或可视化的地方。你需要在这个部分添加一些逻辑来统计每个类别的数量，并将这些信息输出到控制台。

下面是一个示例代码片段，展示了如何修改detect.py来实现这一功能：

# 在detect.py中找到处理检测结果的部分，通常是在save_one_box()调用之后

# 假设predictions是模型的预测结果，im0s是原始图像
for i, det in enumerate(predictions):  # detections per image
    if len(det):
        # Rescale boxes from img_size to im0 size
        det[:, :4] = scale_coords(img.shape[2:], det[:, :4], im0s[i].shape).round()

        # Print results
        for c in det[:, -1].unique():
            n = (det[:, -1] == c).sum()  # detections per class
            s = f"{n} {names[int(c)]}{'s' * (n > 1)}, "  # add to string
            print(s)  # 输出到控制台

        # 统计每个类别的数量
        counts = {}
        for *xyxy, conf, cls in reversed(det):
            key = names[int(cls)]
            if key not in counts:
                counts[key] = 0
            counts[key] += 1

        # 打印图片名称和类别统计
        print(f"图片\"{img_files[i]}\"中人群总数{n}人；")
        for color, count in counts.items():
            print(f"{color}{count}人，")

# 确保在所有循环和条件之外关闭输出
print("进程已结束，退出代码为0")

这段代码首先遍历检测结果，统计每个类别的数量，然后输出到控制台。注意，names是YOLOv5中用来存储类别名的列表，img_files是包含图像文件名的列表，这两个变量通常在detect.py的其他地方定义。

确保你的detect.py文件中有对img_files和names的引用，如果没有，你可能需要从模型配置或数据集配置中获取它们。此外，如果你的中文输出在控制台中显示乱码，确保你的终端支持utf-8编码，可以通过终端设置或在Python脚本开始处添加# -*- coding: utf-8 -*-来解决。

以上步骤应该可以帮助你在PyCharm的终端中以中文形式输出YOLOv5的检测结果。如果有任何具体问题或需要进一步的帮助，请随时提问。