如何在AlexeyAB版YOLO Darknet中显示所有类别的概率？

可以实现，具体操作分以下场景：

1. 核心配置修改（必做） #

• 打开你的YOLO模型配置文件（比如yolov3-custom.cfg），定位到所有[yolo]层的配置段
• 在每个[yolo]段内添加一行配置：output_all=1
  这个参数会强制模型输出每个检测框对应的所有类别概率，而非仅保留置信度最高的类别

2. 命令行检测的补充设置 #

如果用命令行执行检测，可配合以下参数确保所有概率都能完整显示：

• 添加-thresh 0参数，避免默认阈值过滤低置信度结果，确保所有类别概率都被输出
• 示例命令：

./darknet detector test cfg/your-data.cfg cfg/your-model.cfg your-model.weights test-image.jpg -thresh 0

3. Python API调用场景 #

如果用Python调用Darknet，无需修改配置，直接访问检测结果中每个目标的prob数组即可，该数组原生包含了所有类别的概率值。

内容的提问来源于stack exchange，提问作者Alex Zaitsev