From e06af50389f849be0bfe4fa97d39f4519ef2c711 Mon Sep 17 00:00:00 2001 From: adamhrv Date: Fri, 18 Jan 2019 11:00:18 +0100 Subject: change to cli_proc --- megapixels/commands/cv/mirror.py | 57 ---------------------------------------- 1 file changed, 57 deletions(-) delete mode 100644 megapixels/commands/cv/mirror.py (limited to 'megapixels/commands/cv/mirror.py') diff --git a/megapixels/commands/cv/mirror.py b/megapixels/commands/cv/mirror.py deleted file mode 100644 index 9ca1cac7..00000000 --- a/megapixels/commands/cv/mirror.py +++ /dev/null @@ -1,57 +0,0 @@ -""" -Crop images to prepare for training -""" - -import click -import cv2 as cv -from PIL import Image, ImageOps, ImageFilter - -from app.settings import types -from app.utils import click_utils -from app.settings import app_cfg as cfg - - -@click.command() -@click.option('-i', '--input', 'opt_dir_in', required=True, - help='Input directory') -@click.option('-o', '--output', 'opt_dir_out', required=True, - help='Output directory') -@click.option('--slice', 'opt_slice', type=(int, int), default=(None, None), - help='Slice the input list') -@click.pass_context -def cli(ctx, opt_dir_in, opt_dir_out, opt_slice): - """Mirror augment image directory""" - - import os - from os.path import join - from pathlib import Path - from glob import glob - from tqdm import tqdm - - from app.utils import logger_utils, file_utils, im_utils - - # ------------------------------------------------- - # init - - log = logger_utils.Logger.getLogger() - - # ------------------------------------------------- - # process here - - # get list of files to process - fp_ims = glob(join(opt_dir_in, '*.jpg')) - fp_ims += glob(join(opt_dir_in, '*.png')) - - if opt_slice: - fp_ims = fp_ims[opt_slice[0]:opt_slice[1]] - log.info('processing {:,} files'.format(len(fp_ims))) - - # ensure output dir exists - file_utils.mkdirs(opt_dir_out) - - # resize and save images - for fp_im in tqdm(fp_ims): - im = Image.open(fp_im) - fpp_im = Path(fp_im) - fp_out = join(opt_dir_out, '{}_mirror{}'.format(fpp_im.stem, fpp_im.suffix)) - im.save(fp_out) \ No newline at end of file -- cgit v1.2.3-70-g09d2