From b3995b2eaae294afa27cc521e0aea688331a8255 Mon Sep 17 00:00:00 2001
From: "jules@lens" <julescarbon@gmail.com>
Date: Sun, 6 Jan 2019 20:28:15 +0100
Subject: cpu count etc

---
 augment.py              | 2 +-
 cleanup.sh              | 3 +++
 crop-equirectangular.py | 7 +++----
 3 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/augment.py b/augment.py
index 43ef6a3..cb66f8d 100755
--- a/augment.py
+++ b/augment.py
@@ -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...")
diff --git a/cleanup.sh b/cleanup.sh
index 9d16216..43f2750 100755
--- a/cleanup.sh
+++ b/cleanup.sh
@@ -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:
-- 
cgit v1.2.3-70-g09d2