alexnasa commited on
Commit
3475ab4
·
verified ·
1 Parent(s): 3e37320

Update scripts/run_cropping.py

Browse files
Files changed (1) hide show
  1. scripts/run_cropping.py +8 -1
scripts/run_cropping.py CHANGED
@@ -3,6 +3,8 @@ import os
3
  import sys
4
  import importlib
5
 
 
 
6
  import mediapy
7
  from PIL import Image
8
  import tyro
@@ -10,12 +12,14 @@ import tyro
10
  import torchvision.transforms as transforms
11
 
12
 
 
 
13
  from pixel3dmm import env_paths
14
  sys.path.append(f'{env_paths.CODE_BASE}/src/pixel3dmm/preprocessing/PIPNet/FaceBoxesV2/')
15
  from pixel3dmm.preprocessing.pipnet_utils import demo_image
16
  from pixel3dmm import env_paths
17
 
18
-
19
 
20
 
21
  def run(exp_path, image_dir, start_frame = 0,
@@ -24,6 +28,9 @@ def run(exp_path, image_dir, start_frame = 0,
24
  max_bbox : bool = False,
25
  disable_cropping : bool = False,
26
  ):
 
 
 
27
  experiment_name = exp_path.split('/')[-1][:-3]
28
  data_name = exp_path.split('/')[-2]
29
  config_path = '.experiments.{}.{}'.format(data_name, experiment_name)
 
3
  import sys
4
  import importlib
5
 
6
+ print("easy stuff")
7
+
8
  import mediapy
9
  from PIL import Image
10
  import tyro
 
12
  import torchvision.transforms as transforms
13
 
14
 
15
+ print("boring stuff")
16
+
17
  from pixel3dmm import env_paths
18
  sys.path.append(f'{env_paths.CODE_BASE}/src/pixel3dmm/preprocessing/PIPNet/FaceBoxesV2/')
19
  from pixel3dmm.preprocessing.pipnet_utils import demo_image
20
  from pixel3dmm import env_paths
21
 
22
+ print("all we got")
23
 
24
 
25
  def run(exp_path, image_dir, start_frame = 0,
 
28
  max_bbox : bool = False,
29
  disable_cropping : bool = False,
30
  ):
31
+
32
+ print("are we even here")
33
+
34
  experiment_name = exp_path.split('/')[-1][:-3]
35
  data_name = exp_path.split('/')[-2]
36
  config_path = '.experiments.{}.{}'.format(data_name, experiment_name)