cc

scripts/build_cyclegan_dataset.py

@@ -45,7 +45,7 @@ def build_cyclegan_dataset(swim_dir: str, output_dir: str, type: str, no_night:
 
     if type != "night":
         for label in tqdm(train_labels, desc="train"):
-            if no_night and label["timeofdata"] == "night":
+            if no_night and label["timeofday"] == "night":
                 continue
 
             if label["weather"] == type:
@@ -58,7 +58,7 @@ def build_cyclegan_dataset(swim_dir: str, output_dir: str, type: str, no_night:
                 )
 
         for label in tqdm(val_labels, desc="val"):
-            if no_night and label["timeofdata"] == "night":
+            if no_night and label["timeofday"] == "night":
                 continue
 
             if label["weather"] == type:
@@ -71,21 +71,21 @@ def build_cyclegan_dataset(swim_dir: str, output_dir: str, type: str, no_night:
                 )
     else:
         for label in tqdm(train_labels, desc="train"):
-            if label["timeofdata"] == "night":
+            if label["timeofday"] == "night":
                 os.system(
                     f"cp {os.path.join(swim_dir, 'train', 'images', label['name'])} {os.path.join(output_dir, 'trainB', label['name'])}"
                 )
-            elif label["timeofdata"] == "daytime":
+            elif label["timeofday"] == "daytime":
                 os.system(
                     f"cp {os.path.join(swim_dir, 'train', 'images', label['name'])} {os.path.join(output_dir, 'trainA', label['name'])}"
                 )
 
         for label in tqdm(val_labels, desc="val"):
-            if label["timeofdata"] == "night":
+            if label["timeofday"] == "night":
                 os.system(
                     f"cp {os.path.join(swim_dir, 'val', 'images', label['name'])} {os.path.join(output_dir, 'testB', label['name'])}"
                 )
-            elif label["timeofdata"] == "daytime":
+            elif label["timeofday"] == "daytime":
                 os.system(
                     f"cp {os.path.join(swim_dir, 'val', 'images', label['name'])} {os.path.join(output_dir, 'testA', label['name'])}"
                 )