"git@developer.sourcefind.cn:OpenDAS/apex.git" did not exist on "fceec07dfc58f28d61fdf77447feda1d78f1cf47"
Unverified Commit 677954ba authored by tpoisonooo's avatar tpoisonooo Committed by GitHub
Browse files

feat(local_visualizer.py): add check dir exist (#1828)

* feat(local_visualizer.py): add check dir exist

* fix(local_visualizer.py): use mmengine API

* format(local_visualizer.py): remove useless

* style(local_visualizer.py): format
parent 82fd2d92
...@@ -5,6 +5,7 @@ from typing import Dict, List, Optional, Tuple, Union ...@@ -5,6 +5,7 @@ from typing import Dict, List, Optional, Tuple, Union
import mmcv import mmcv
import numpy as np import numpy as np
from mmengine import mkdir_or_exist
from mmengine.dist import master_only from mmengine.dist import master_only
from torch import Tensor from torch import Tensor
...@@ -605,6 +606,8 @@ class Det3DLocalVisualizer(DetLocalVisualizer): ...@@ -605,6 +606,8 @@ class Det3DLocalVisualizer(DetLocalVisualizer):
win_name=name, win_name=name,
wait_time=wait_time) wait_time=wait_time)
mkdir_or_exist(out_file)
if out_file is not None: if out_file is not None:
if drawn_img_3d is not None: if drawn_img_3d is not None:
mmcv.imwrite(drawn_img_3d[..., ::-1], out_file + '.jpg') mmcv.imwrite(drawn_img_3d[..., ::-1], out_file + '.jpg')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment