Update scripts/run_cropping.py

scripts/run_cropping.py

@@ -3,8 +3,6 @@ import os
 import sys
 import importlib
 
-print("easy stuff") 
-
 import mediapy
 from PIL import Image
 import tyro
@@ -12,14 +10,11 @@ import tyro
 import torchvision.transforms as transforms
 
 
-print("boring stuff") 
-
 from pixel3dmm import env_paths
-sys.path.append(f'{env_paths.CODE_BASE}/src/pixel3dmm/preprocessing/PIPNet/FaceBoxesV2/')
+sys.path.append(f'./src/pixel3dmm/preprocessing/PIPNet/FaceBoxesV2/')
 from pixel3dmm.preprocessing.pipnet_utils import demo_image
 from pixel3dmm import env_paths
 
-print("all we got") 
 
 
 def run(exp_path, image_dir, start_frame = 0,
@@ -29,8 +24,6 @@ def run(exp_path, image_dir, start_frame = 0,
         disable_cropping : bool = False,
         ):
 
-    print("are we even here")
-    
     experiment_name = exp_path.split('/')[-1][:-3]
     data_name = exp_path.split('/')[-2]
     config_path = '.experiments.{}.{}'.format(data_name, experiment_name)
@@ -41,17 +34,14 @@ def run(exp_path, image_dir, start_frame = 0,
     cfg.experiment_name = experiment_name
     cfg.data_name = data_name
 
-    save_dir = os.path.join(f'{env_paths.CODE_BASE}/src/pixel3dmm/preprocessing/PIPNet/snapshots', cfg.data_name, cfg.experiment_name)
+    save_dir = os.path.join(f'./src/pixel3dmm/preprocessing/PIPNet/snapshots', cfg.data_name, cfg.experiment_name)
 
-    print(f'save dir {save_dir}')
 
     normalize = transforms.Normalize(mean=[0.485, 0.456, 0.406],
                                      std=[0.229, 0.224, 0.225])
     preprocess = transforms.Compose(
         [transforms.Resize((cfg.input_size, cfg.input_size)), transforms.ToTensor(), normalize])
 
-    print(f'we made it so far')
-
     #for pid in pids:
     pid = "FaMoS_180424_03335_TA_selfie_IMG_0092.jpg"
     pid = "FaMoS_180426_03336_TA_selfie_IMG_0152.jpg"