如何在 Yolov3 暗网中更改视频输出中的边界框颜色

Question

我已经在我的自定义数据集上训练了YOLOv3暗网模型。首先，我在谷歌colab中克隆了https://github.com/kriyeng/darknet/存储库，然后在谷歌colab中运行整个代码。但是我想更改输出视频中边界框的颜色。请建议如何做。提前谢谢。

Answer 1

这是C++的解决方案，但您也可以将其应用于python。只需更改标量中的值即可。在下面的示例中，它设置为红色。

//Draw a rectangle displaying the bounding box
rectangle(frame, Point(left, top), Point(right, bottom), Scalar(0, 0, 255));