Update check.py (#39)

2d04181b · LutaoChu · wuzewu · 164ade2c · 2d04181b
隐藏空白更改
内联并排

Showing with 3 addition and 2 deletion

pdseg/check.py pdseg/check.py +3 -2

未找到文件。
--- a/pdseg/check.py
+++ b/pdseg/check.py
@@ -417,7 +417,8 @@ def check_val_dataset():
                img = cv2_imread(img_path, cv2.IMREAD_UNCHANGED)
                grt = cv2_imread(grt_path, cv2.IMREAD_UNCHANGED)
            except Exception as e:
-                imread_failed.append((line, e.message))
+                imread_failed.append((line, str(e)))
+                continue

            is_gray = is_label_gray(grt)
            if not is_gray:
@@ -470,7 +471,7 @@ def check_test_dataset():
                    img = cv2_imread(img_path, cv2.IMREAD_UNCHANGED)
                    grt = cv2_imread(grt_path, cv2.IMREAD_UNCHANGED)
                except Exception as e:
-                    imread_failed.append((line, e.message))
+                    imread_failed.append((line, str(e)))
                    continue

                is_gray = is_label_gray(grt)