debug pipeline

2024-10-21 07:33:32 +00:00
parent d0fbb0f198
commit 9ca0851bf7
6 changed files with 55 additions and 47 deletions
--- a/utils/data_load.py
+++ b/utils/data_load.py
@@ -34,6 +34,8 @@ class DataLoadUtil:
    @staticmethod
    def get_label_num(root, scene_name):
        label_dir = os.path.join(root, scene_name, "label")
+        if not os.path.exists(label_dir):
+            return 0
        return len(os.listdir(label_dir))

    @staticmethod