-
Notifications
You must be signed in to change notification settings - Fork 7.2k
Closed
Description
PR #2785 introduced a new utility to draw bounding boxes on images. The code changes were merged but there are a few nice-to-have improvements missing.
More specifically:
- The unit-test
test_draw_boxes()
uses PIL instead ofwrite_image()
to save the image. We should use TorchVision's method instead. - The implementation of the
draw_bounding_boxes()
initializes the text font within the loop which is suboptimal. We should move it outside of the loop and initialize it once.