提交 2d04181b 编写于 作者: L LutaoChu 提交者: wuzewu

Update check.py (#39)

上级 164ade2c
...@@ -417,7 +417,8 @@ def check_val_dataset(): ...@@ -417,7 +417,8 @@ def check_val_dataset():
img = cv2_imread(img_path, cv2.IMREAD_UNCHANGED) img = cv2_imread(img_path, cv2.IMREAD_UNCHANGED)
grt = cv2_imread(grt_path, cv2.IMREAD_UNCHANGED) grt = cv2_imread(grt_path, cv2.IMREAD_UNCHANGED)
except Exception as e: except Exception as e:
imread_failed.append((line, e.message)) imread_failed.append((line, str(e)))
continue
is_gray = is_label_gray(grt) is_gray = is_label_gray(grt)
if not is_gray: if not is_gray:
...@@ -470,7 +471,7 @@ def check_test_dataset(): ...@@ -470,7 +471,7 @@ def check_test_dataset():
img = cv2_imread(img_path, cv2.IMREAD_UNCHANGED) img = cv2_imread(img_path, cv2.IMREAD_UNCHANGED)
grt = cv2_imread(grt_path, cv2.IMREAD_UNCHANGED) grt = cv2_imread(grt_path, cv2.IMREAD_UNCHANGED)
except Exception as e: except Exception as e:
imread_failed.append((line, e.message)) imread_failed.append((line, str(e)))
continue continue
is_gray = is_label_gray(grt) is_gray = is_label_gray(grt)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册