diff --git a/pdseg/tools/gray2pseudo_color.py b/pdseg/tools/gray2pseudo_color.py index 6128f3776138956757577fe9a415d21aadc57c62..3627db0b216175b04a50d9012999d441f4df69fb 100644 --- a/pdseg/tools/gray2pseudo_color.py +++ b/pdseg/tools/gray2pseudo_color.py @@ -61,8 +61,7 @@ def gray2pseudo_color(args): try: grt_path = osp.join(fpath, f) _output_dir = fpath.replace(input, '') - if _output_dir[0] == os.path.sep: - _output_dir = _output_dir.lstrip(os.path.sep) + _output_dir = _output_dir.lstrip(os.path.sep) im = Image.open(grt_path) lbl = np.asarray(im)