From 243536e43e10498db36695fa5ff13d4d3283a130 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 9 Aug 2022 20:29:25 +0200 Subject: [PATCH] Scope `torchvision.utils.draw_bounding_boxes` import (#8915) * Scope `torchvision.utils.draw_bounding_boxes` import Quick fix for https://github.com/ultralytics/yolov5/issues/8898#issuecomment-1208851874 until a more permanent fix is implemented * Update clearml_utils.py --- utils/loggers/clearml/clearml_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/loggers/clearml/clearml_utils.py b/utils/loggers/clearml/clearml_utils.py index 0c30d2dc278d..fb9889172562 100644 --- a/utils/loggers/clearml/clearml_utils.py +++ b/utils/loggers/clearml/clearml_utils.py @@ -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):