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
import mmcv
import numpy as np
from mmengine import mkdir_or_exist
from mmengine.dist import master_only
from torch import Tensor
......@@ -605,6 +606,8 @@ class Det3DLocalVisualizer(DetLocalVisualizer):
win_name=name,
wait_time=wait_time)
mkdir_or_exist(out_file)
if out_file is not None:
if drawn_img_3d is not None:
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