Spaces:
Paused
Paused
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"
|