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:
1 parent
b551098d59
commit
7ab1f50633
1 file changed
+1
-1
@@ -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):
|
||||
|
||||
Reference in new issue
Block a user