Fix SPair-71k process

Thinklab-SJTU · Oct 15, 2023 · e7d134e · e7d134e
1 parent 69bffbe
commit e7d134e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pygmtools/dataset.py b/pygmtools/dataset.py
@@ -839,7 +839,7 @@ def process(self):
         train_file = os.path.join(self.dataset_dir, 'train.json')
         test_file = os.path.join(self.dataset_dir, 'test.json')
         img_file = os.path.join(self.dataset_dir, 'data-' + str(self.obj_resize) + '-' + self.suffix + '.json')
-        if (not os.path.exists(train_file)) or (not os.path.exists(test_file)):
+        if (not os.path.exists(train_file)) or (not os.path.exists(test_file)) or (not os.path.exists(img_file)):
             train_list = []
             test_list = []
             if self.sets == 'trn':