mirror of
https://github.com/brycedrennan/imaginAIry
synced 2024-10-31 03:20:40 +00:00
40 lines
973 B
Python
40 lines
973 B
Python
|
import click
|
||
|
|
|
||
|
|
|
||
|
|
@click.command("edit-demo")
|
||
|
|
@click.argument("image_paths", metavar="PATH|URL", required=True, nargs=-1)
|
||
|
|
@click.option(
|
||
|
|
"--outdir",
|
||
|
|
default="./outputs",
|
||
|
|
show_default=True,
|
||
|
|
type=click.Path(),
|
||
|
|
help="Where to write results to.",
|
||
|
|
)
|
||
|
|
@click.option(
|
||
|
|
"-h",
|
||
|
|
"--height",
|
||
|
|
default=512,
|
||
|
|
show_default=True,
|
||
|
|
type=int,
|
||
|
|
help="Image height. Should be multiple of 8.",
|
||
|
|
)
|
||
|
|
@click.option(
|
||
|
|
"-w",
|
||
|
|
"--width",
|
||
|
|
default=512,
|
||
|
|
show_default=True,
|
||
|
|
type=int,
|
||
|
|
help="Image width. Should be multiple of 8.",
|
||
|
|
)
|
||
|
|
def edit_demo_cmd(image_paths, outdir, height, width):
|
||
|
|
"""Make some fun pre-set edits to input photos."""
|
||
|
|
|
||
|
|
from imaginairy.log_utils import configure_logging
|
||
|
|
from imaginairy.surprise_me import create_surprise_me_images
|
||
|
|
|
||
|
|
configure_logging()
|
||
|
|
for image_path in image_paths:
|
||
|
|
create_surprise_me_images(
|
||
|
|
image_path, outdir=outdir, make_gif=True, width=width, height=height
|
||
|
|
)
|