search

lufficc / SSD

High quality, fast, modular reference implementation of SSD in PyTorch
MIT License
1.51k stars 385 forks source link

modify the color of the detection box #199

Open Rock-L21 opened 3 years ago

Rock-L21 commented 3 years ago

What should I do if I want to modify the color of the detection box in the demo.py output image?

SamSamhuns commented 3 years ago

The author has disabled the color parameter in the draw_boxes function from their Vizer library if you check here I'd say create your own function to draw the bounding boxes which should be pretty straightforward or use the _draw_single_box function from the vizer library to iterate through the boxes and use custom colors

Rock-L21 commented 3 years ago

The author has disabled the color parameter in the draw_boxes function from their Vizer library if you check here I'd say create your own function to draw the bounding boxes which should be pretty straightforward or use the _draw_single_box function from the vizer library to iterate through the boxes and use custom colors

Thank you very much！