Изменить толщину ограничивающего прямоугольника в YOLO Darkflow
Я играю в Arround с темным потоком YOLO ( https://github.com/thtrieu/darkflow) и хочу узнать, как изменить толщину ограничивающего прямоугольника прогнозируемых объектов.
Я использую следующую команду, чтобы проверить видео
flow --model cfg/yolo-new.cfg --load bin/yolo-new.weights --demo videofile.avi
На самом деле ограничивающие рамки слишком толстые. Когда на изображении много объектов, я вижу только ограничивающие рамки и никаких объектов. Поэтому я хочу знать, как сделать коробки тоньше, например, толщиной всего в 1 или 2 пикселя.
Спасибо:)
1 ответ
Решение
РЕДАКТИРОВАТЬ
Для darflow: измените следующую строку в файле Forex.py в yolo и yolo2
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 box то
cvRectangle(show_img, pt1, pt2, color, 1, 8, 0);