Изменить толщину ограничивающего прямоугольника в 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);
Другие вопросы по тегам