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

Спасибо :)


yolo bounding-box darkflow
person HKC72    schedule 18.03.2018    source источник

Ответы (1)


arrow_upward
2
arrow_downward

ИЗМЕНИТЬ

Для darflow: измените следующую строку в файле predic.py в yolo и yolo2

cv2.rectangle(imgcv,(left, top), (right, bot),self.meta['colors'][max_indx], thick)

to 

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 на значение пикселя

например, если вам нужен блок размером 1 пиксель, тогда

cvRectangle(show_img, pt1, pt2, color, 1, 8, 0);
person Smit Shilu    schedule 19.03.2018
comment
Большое спасибо за уделенное время, Смит Шилу. Я уже искал эту деталь, но не нашел. Вы нашли его в репозитории darflow? - person HKC72; 19.03.2018
comment
Вы найдете cv2.rectangle в репозитории darkflow в файлах pred.py в yolo и yolo2. Просто измените параметр толщиной на 1 - person Smit Shilu; 19.03.2018
comment
Как это сделать в даркнете - person Trect; 28.11.2018