Unverified 提交 7ab1f506 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

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
上级 b551098d
...@@ -5,11 +5,11 @@ from pathlib import Path ...@@ -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):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论