diff options
| -rwxr-xr-x | augment.py | 2 | ||||
| -rwxr-xr-x | cleanup.sh | 3 | ||||
| -rw-r--r-- | crop-equirectangular.py | 7 |
3 files changed, 7 insertions, 5 deletions
@@ -102,7 +102,7 @@ print("Got sequence {}, {} images".format(data_opt.sequence_name, len(sequence)) if data_opt.no_symlinks: _len = len(sequence) - data_opt.augment_make - 2 else: - _len = len(sequence) + _len = len(sequence) - data_opt.augment_make - 2 if _len <= 0: print("Got empty sequence...") @@ -15,6 +15,9 @@ rm checkpoints/*/5_*.pth rm checkpoints/*/6_*.pth rm checkpoints/*/7_*.pth rm checkpoints/*/8_*.pth +rm checkpoints/*/9_*.pth +rm checkpoints/*/10_*.pth +rm checkpoints/*/11_*.pth echo "Cleaning up recursive frames..." diff --git a/crop-equirectangular.py b/crop-equirectangular.py index 8f02112..38e93ae 100644 --- a/crop-equirectangular.py +++ b/crop-equirectangular.py @@ -3,7 +3,7 @@ import glob import argparse from shutil import rmtree from PIL import Image -from multiprocessing import Pool +from multiprocessing import Pool, cpu_count from dotenv import load_dotenv, find_dotenv import subprocess load_dotenv(find_dotenv()) @@ -14,7 +14,7 @@ load_dotenv(find_dotenv()) # Setting a higher overlap means you can have taller vertical FOV. parser = argparse.ArgumentParser() -parser.add_argument('--folder', default="./pix2pixhd/sequences/venice_360/equi") +parser.add_argument('--folder', default="./sequences/venice_360/equi") parser.add_argument('--label', required=True) parser.add_argument('--vertical_offset', type=int, default=983) parser.add_argument('--folder_id', type=int, required=True) @@ -90,9 +90,8 @@ def build_thumbnail(i, fn): canvas.crop(p1).resize((1024, 512,), Image.ANTIALIAS).save(os.path.join(path_1, out_fn)) canvas.crop(p2).resize((1024, 512,), Image.ANTIALIAS).save(os.path.join(path_2, out_fn)) -agents = multiprocessing.cpu_count() chunksize = 3 -with Pool(processes=agents) as pool: +with Pool(processes=cpu_count()) as pool: pool.starmap(build_thumbnail, dataset, chunksize) if opt.folder_id > 0: |