search

varunagrawal / tiny-faces-pytorch

Finding Tiny Faces in PyTorch
MIT License
164 stars 44 forks source link

Visualize result #30

Closed toti-bz closed 9 months ago

toti-bz commented 2 years ago

is there a way to visualize the predictions?

hamzakhalil798 commented 1 year ago

@toti-bz Did you find any solution?

varunagrawal commented 9 months ago

Torchvision has a method called draw_bounding_box that might be useful in this case. Else, OpenCV has lots of examples of drawing bounding boxes in images.