Merge pull request #32 from Jianx-Gao/patch-1

Update my_utils_cls.py
OpenXLab-Edu · Mar 24, 2022 · df53570 · df53570
2 parents 94d7e21 + 584f301
commit df53570
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/utils/my_utils_cls.py b/utils/my_utils_cls.py
@@ -143,17 +143,17 @@ def load_dataset(self, path):
             to_rgb=True
         )
 
-        self.cfg.data.train.data_prefix = os.path.join(self.dataset_path, '/training_set')
-        self.cfg.data.train.classes = os.path.join(self.dataset_path, '/classes.txt')
+        self.cfg.data.train.data_prefix = os.path.join(self.dataset_path, 'training_set')
+        self.cfg.data.train.classes = os.path.join(self.dataset_path, 'classes.txt')
         # self.cfg.data.train.ann_file = path + '/train.txt'
 
-        self.cfg.data.val.data_prefix = os.path.join(self.dataset_path, '/val_set')
-        self.cfg.data.val.ann_file = os.path.join(self.dataset_path, '/val.txt')
-        self.cfg.data.val.classes = os.path.join(self.dataset_path, '/classes.txt')
+        self.cfg.data.val.data_prefix = os.path.join(self.dataset_path, 'val_set')
+        self.cfg.data.val.ann_file = os.path.join(self.dataset_path, 'val.txt')
+        self.cfg.data.val.classes = os.path.join(self.dataset_path, 'classes.txt')
 
-        self.cfg.data.test.data_prefix = os.path.join(self.dataset_path, '/test_set')
-        self.cfg.data.test.ann_file = os.path.join(self.dataset_path, '/test.txt')
-        self.cfg.data.test.classes = os.path.join(self.dataset_path, '/classes.txt')
+        self.cfg.data.test.data_prefix = os.path.join(self.dataset_path, 'test_set')
+        self.cfg.data.test.ann_file = os.path.join(self.dataset_path, 'test.txt')
+        self.cfg.data.test.classes = os.path.join(self.dataset_path, 'classes.txt')
 
     # def print_configs(self):
     #     if self.backbone is None: