I think the code line 402-403 in openpsg/utils/utils.py is error ```python if out_file is not None: mmcv.imwrite(output_viz_graph, osp.join(out_dir, '{}.jpg'.format(i))) ``` It myabe ```python if out_dir is not None: mmcv.imwrite(output_viz_graph, osp.join(out_dir, '{}.jpg'.format(i))) ```