moved image resize script

scripts/resize_images.py → cvdatasets/resize.py

@@ -6,51 +6,39 @@ import multiprocessing as mp
 import os
 import shutil
 
-from PIL import Image
+from cvargparse import Arg
 from cvargparse import BaseParser
-from tqdm.auto import tqdm
-from cvargparse import cvdataclass
 from functools import partial
 from pathlib import Path
+from tqdm.auto import tqdm
 
-@cvdataclass
-class Args:
-	source:       str  = "source"
-	destination:  str  = "dest"
-	n_jobs:       int  = -1
-	extensions:   tuple = (".jpg", ".jpeg", ".png")
-
-	size:         int  = 1000
-	fit_short:    bool = False
-
-	remove_existing: bool = False
-
-
+from cvdatasets.utils.image import read_image
 
 def resize(name, *, source: Path, dest: Path, size: int, fit_short: bool = False):
 	src = source / name
 	dst = dest / name
 
+	im  = read_image(src)
+	w, h = im.size
 
-	with Image.open(src) as im:
-		w, h = im.size
-
-		if fit_short:
-			if w > h:
-				W, H = (int(size * w / h), size)
-			else:
-				W, H = (size, int(size * h / w))
+	if fit_short:
+		if w > h:
+			W, H = (int(size * w / h), size)
+		else:
+			W, H = (size, int(size * h / w))
 
+	else:
+		if w > h:
+			W, H = (size, int(size * h / w))
 		else:
-			if w > h:
-				W, H = (size, int(size * h / w))
-			else:
-				W, H = (int(size * w / h), size)
+			W, H = (int(size * w / h), size)
+
+	dst.parent.mkdir(parents=True, exist_ok=True)
+	im.resize((W, H)).save(dst)
 
-		dst.parent.mkdir(parents=True, exist_ok=True)
-		im.resize((W, H)).save(dst)
 
-def main(args: Args):
+def main(args):
+
 	source = Path(args.source)
 	destination = Path(args.destination).resolve()
 	assert source.exists(), \
@@ -93,5 +81,26 @@ def main(args: Args):
 			work(imname)
 
 
-args = BaseParser(Args).parse_args()
-main(args)
+
+parser = BaseParser([
+	Arg("source"),
+
+	Arg("destination"),
+
+	Arg.int("--n_jobs", "-j", default=-1,
+		help="number of concurrent resize processes"),
+
+	Arg("--extensions", "-ext", nargs="+", default=[".jpg", ".jpeg", ".png"],
+		help="file extensions to processs"),
+
+	Arg.int("--size", "-s", default=1000,
+		help="size in pixels"),
+
+	Arg.flag("--fit_short",
+		help="resize to the given size the short size or the long size (default)."),
+
+	Arg.flag("--remove_existing",
+		help="remove existing images."),
+])
+
+main(parser.parse_args())