Update scripts/run_cropping.py

scripts/run_cropping.py

@@ -11,7 +11,7 @@ import torchvision.transforms as transforms
 
 
 from pixel3dmm import env_paths
-sys.path.append(f'./src/pixel3dmm/preprocessing/PIPNet/FaceBoxesV2/')
+sys.path.append(f'{env_paths.CODE_BASE}/src/pixel3dmm/preprocessing/PIPNet/FaceBoxesV2/')
 from pixel3dmm.preprocessing.pipnet_utils import demo_image
 from pixel3dmm import env_paths
 
@@ -34,7 +34,7 @@ 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'./src/pixel3dmm/preprocessing/PIPNet/snapshots', cfg.data_name, cfg.experiment_name)
+    save_dir = os.path.join(f'{env_paths.CODE_BASE}/src/pixel3dmm/preprocessing/PIPNet/snapshots', cfg.data_name, cfg.experiment_name)
 
 
     normalize = transforms.Normalize(mean=[0.485, 0.456, 0.406],