search

lufficc / SSD

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

How to change the thickness of prediction box #194

Open 18718277909 opened 3 years ago

18718277909 commented 3 years ago

Hi，Thanks for your job Because the feature box of the predicted image output from demo is too thin, it is difficult to see the box clearly，So I would like to ask how to change the thickness of the box. Thank you for your help

SamSamhuns commented 3 years ago

Just pass a width parameter with the desired value in the draw_boxes function. Like this in the demo.py file.

90       drawn_image = draw_boxes(image, boxes, labels,
91                                scores, class_names, width=20).astype(np.uint8)