src/yolo_v2_class.hpp
@@ -453,7 +453,14 @@ #ifdef OPENCV cv::Scalar obj_id_to_color(int obj_id); cv::Scalar obj_id_to_color(int obj_id) { int const colors[6][3] = { { 1,0,1 },{ 0,0,1 },{ 0,1,1 },{ 0,1,0 },{ 1,1,0 },{ 1,0,0 } }; int const offset = obj_id * 123457 % 6; int const color_scale = 150 + (obj_id * 123457) % 100; cv::Scalar color(colors[offset][0], colors[offset][1], colors[offset][2]); color *= color_scale; return color; } class preview_boxes_t { enum { frames_history = 30 }; // how long to keep the history saved