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
This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-08-09 20:29:25 +02:00
1 parent b551098d59
commit 7ab1f50633
1 file changed
+1 -1
+1 -1
View File
@@ -5,11 +5,11 @@ from pathlib import Path
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):