Fix GDrive rate limit

phygitalism · Jul 1, 2021 · f8f5d2d · f8f5d2d
1 parent c5b4b74
commit f8f5d2d
Show file tree

Hide file tree

Showing 3 changed files with 891 additions and 884 deletions.
diff --git a/notebooks/inference_playground.ipynb b/notebooks/inference_playground.ipynb
diff --git a/training/coaches/single_id_coach.py b/training/coaches/single_id_coach.py
@@ -20,7 +20,6 @@ def train(self):
         use_ball_holder = True
 
         for fname, image in tqdm(self.data_loader):
-            torch.cuda.empty_cache()
             image_name = fname[0]
 
             self.restart_training()

diff --git a/utils/align_data.py b/utils/align_data.py
@@ -9,7 +9,6 @@
 def pre_process_images(raw_images_path):
     current_directory = os.getcwd()
 
-
     IMAGE_SIZE = 1024
     predictor = dlib.shape_predictor(paths_config.dlib)
     os.chdir(raw_images_path)
@@ -24,8 +23,13 @@ def pre_process_images(raw_images_path):
         except Exception as e:
             print(e)
 
+    os.makedirs(paths_config.input_data_path, exist_ok=True)
     for image, name in zip(aligned_images, images_names):
         real_name = name.split('.')[0]
         image.save(f'{paths_config.input_data_path}/{real_name}.jpeg')
 
     os.chdir(current_directory)
+
+
+if __name__ == "__main__":
+    pre_process_images('')