更改YOLO Darkflow中的边界框厚度

时间:2018-03-18 07:51:09

标签: bounding-box yolo darkflow

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

我使用以下命令测试视频

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

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

谢谢:)

1 个答案:

答案 0 :(得分:2)

修改

对于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);