Skip to content

Commit

Permalink
Scope torchvision.utils.draw_bounding_boxes import (ultralytics#8915)
Browse files Browse the repository at this point in the history
* Scope `torchvision.utils.draw_bounding_boxes` import

Quick fix for ultralytics#8898 (comment) until a more permanent fix is implemented

* Update clearml_utils.py
  • Loading branch information
glenn-jocher authored and Clay Januhowski committed Sep 8, 2022
1 parent c9c1ea4 commit 243536e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion utils/loggers/clearml/clearml_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,11 +5,11 @@

import yaml
from torchvision.transforms import ToPILImage
from torchvision.utils import draw_bounding_boxes

try:
import clearml
from clearml import Dataset, Task
from torchvision.utils import draw_bounding_boxes # WARNING: requires torchvision>=0.9.0

assert hasattr(clearml, '__version__') # verify package import not local dir
except (ImportError, AssertionError):
Expand Down

0 comments on commit 243536e

Please sign in to comment.