Spaces:
Running
on
Zero
Running
on
Zero
Update scripts/run_cropping.py
Browse files- 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"
|