Ankle/yolov5_ZWgra
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Scope torchvision.utils.draw_bounding_boxes import (#8915)

Browse Source 
* 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
Download Patch File Download Diff File Expand all files Collapse all files
utils/loggers/clearml/clearml_utils.py
+1 -1
View File
@@ -5,11 +5,11 @@ from pathlib import Path
import yaml import yaml
from torchvision.transforms import ToPILImage from torchvision.transforms import ToPILImage
from torchvision.utils import draw_bounding_boxes
try: try:
import clearml import clearml
from clearml import Dataset, Task 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 assert hasattr(clearml, '__version__') # verify package import not local dir
except (ImportError, AssertionError): except (ImportError, AssertionError):