modify gray2psedu_color.py (#117)

* gray2color

modify gray2psedu_color.py (#117)
* gray2color
0e1ba034 · LutaoChu · wuzewu · a62fd058 · 0e1ba034
隐藏空白更改
内联并排

Showing with 1 addition and 2 deletion

pdseg/tools/gray2pseudo_color.py pdseg/tools/gray2pseudo_color.py +1 -2

未找到文件。
--- 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)