alexnasa commited on
Commit
3e37320
·
verified ·
1 Parent(s): 9e2567f

Update scripts/run_cropping.py

Browse files
Files changed (1) hide show
  1. scripts/run_cropping.py +2 -1
scripts/run_cropping.py CHANGED
@@ -36,13 +36,14 @@ def run(exp_path, image_dir, start_frame = 0,
36
 
37
  save_dir = os.path.join(f'{env_paths.CODE_BASE}/src/pixel3dmm/preprocessing/PIPNet/snapshots', cfg.data_name, cfg.experiment_name)
38
 
39
-
40
 
41
  normalize = transforms.Normalize(mean=[0.485, 0.456, 0.406],
42
  std=[0.229, 0.224, 0.225])
43
  preprocess = transforms.Compose(
44
  [transforms.Resize((cfg.input_size, cfg.input_size)), transforms.ToTensor(), normalize])
45
 
 
46
 
47
  #for pid in pids:
48
  pid = "FaMoS_180424_03335_TA_selfie_IMG_0092.jpg"
 
36
 
37
  save_dir = os.path.join(f'{env_paths.CODE_BASE}/src/pixel3dmm/preprocessing/PIPNet/snapshots', cfg.data_name, cfg.experiment_name)
38
 
39
+ print(f'save dir {save_dir}')
40
 
41
  normalize = transforms.Normalize(mean=[0.485, 0.456, 0.406],
42
  std=[0.229, 0.224, 0.225])
43
  preprocess = transforms.Compose(
44
  [transforms.Resize((cfg.input_size, cfg.input_size)), transforms.ToTensor(), normalize])
45
 
46
+ print(f'we made it so far')
47
 
48
  #for pid in pids:
49
  pid = "FaMoS_180424_03335_TA_selfie_IMG_0092.jpg"