时间戳

首页 > 解决方案 > 如何在 YOLO (V4) 中编辑或移除边界框标签文本？

computer-vision - 如何在 YOLO (V4) 中编辑或移除边界框标签文本？

问题描述

我想编辑边界框标签以仅显示检测概率而不显示类别标签，我该怎么做？

我发现了一个名为image.cdarknet/src 的文件，我认为这是我需要进行编辑的地方。但是其中有多个似乎与此任务相关的功能，我不确定要编辑哪个，以及如何编辑以获得我想要的。代码image.c很长，因此请参考这个链接（官方暗网回购），我所指的代码是可用的。

我尝试void draw_detections在第 465 行编辑函数，只需将第 511 行的代码更改为printf("%s: %.0f%%", " ", prob * 100);，运行!make命令，但标签仍然存在于检测中。

标签： computer-visionobject-detectionyolobounding-boxyolov4

解决方案

您在项目中使用 alexyAB 模型。转到 image_opencv.cpp 文件并在找到此行之后找到 draw_detections_cv_v3 函数：

 strcat(labelstr, names[j]);

改成：

strcat(labelstr, "");

推荐阅读