在 YOLO Darkflow 中更改边界框厚度

Question

我正在玩 YOLO darkflow (https://github.com/thtrieu/darkflow)，我想知道如何更改预测对象的边界框厚度。

我用下面的命令来测试一个视频

flow --model cfg/yolo-new.cfg --load bin/yolo-new.weights --demo videofile.avi

实际上边界框太粗了。当图像上有很多物体时，我只能看到边界框而看不到任何物体。所以我想知道如何使盒子更薄，例如只有 1 或 2 个像素厚。

谢谢:)

Answer 1

编辑

对于 darflow：在 yolo 和 yolo2

中更改文件 predict.py 中的以下行

cv2.rectangle(imgcv,(left, top), (right, bot),self.meta['colors'][max_indx], thick)

到

cv2.rectangle(imgcv,(left, top), (right, bot),self.meta['colors'][max_indx], 1)

对于暗网：您必须手动更改框的厚度。因此，在 image.c 文件

中搜索以下行

cvRectangle(show_img, pt1, pt2, color, width, 8, 0);

并将第 5 个参数 width 更改为像素值

例如，如果你想要 1px 的框那么

cvRectangle(show_img, pt1, pt2, color, 1, 8, 0);

Change Bounding Box thickness in YOLO Darkflow