def main():
try:
parser = ArgumentParser(description="Deep dreamer")
parser.add_argument(
"--zoom", choices=["true", "false"], default="true",
help="zoom dreams (default: true)")
parser.add_argument(
"--scale", type=float, default=0.05,
help="scale coefficient for zoom (default: 0.05)")
parser.add_argument(
"--dreams", type=int, default=100,
help="number of images (default: 100)")
parser.add_argument(
"--itern", type=int, default=10,
help="dream iterations (default: 10)")
parser.add_argument(
"--octaves", type=int, default=4,
help="dream octaves (default: 4)")
parser.add_argument(
"--octave-scale", type=float, default=1.4,
help="dream octave scale (default: 1.4)")
parser.add_argument(
"--layers", type=str, default="inception_4c/output",
help="dream layers (default: inception_4c/output)")
parser.add_argument(
"--clip", choices=["true", "false"], default="true",
help="clip dreams (default: true)")
group = parser.add_mutually_exclusive_group(required=True)
group.add_argument("image", nargs="?")
group.add_argument(
"--list-layers", action="store_true", help="list layers")
args = parser.parse_args()
if args.list_layers:
list_layers()
else:
zoom = True
if args.zoom == "false":
zoom = False
clip = True
if args.clip == "false":
clip = False
deepdream(
args.image, zoom=zoom, scale_coefficient=args.scale,
irange=args.dreams, iter_n=args.itern, octave_n=args.octaves,
octave_scale=args.octave_scale, end=args.layers, clip=clip)
except Exception as e:
print("Error: {}".format(e))
sys.exit(2)
def main():
try:
parser = ArgumentParser(description="Deep dreamer")
parser.add_argument(
"--zoom", choices=["true", "false"], default="true",
help="zoom dreams (default: true)")
parser.add_argument(
"--scale", type=float, default=0.05,
help="scale coefficient for zoom (default: 0.05)")
parser.add_argument(
"--dreams", type=int, default=100,
help="number of images (default: 100)")
parser.add_argument(
"--itern", type=int, default=10,
help="dream iterations (default: 10)")
parser.add_argument(
"--octaves", type=int, default=4,
help="dream octaves (default: 4)")
parser.add_argument(
"--octave-scale", type=float, default=1.4,
help="dream octave scale (default: 1.4)")
parser.add_argument(
"--layers", type=str, default="inception_4c/output",
help="dream layers (default: inception_4c/output)")
parser.add_argument(
"--clip", choices=["true", "false"], default="true",
help="clip dreams (default: true)")
parser.add_argument(
"--network", choices=['bvlc_googlenet', 'googlenet_place205'],
default='bvlc_googlenet',
help="choose the network to use (default: bvlc_googlenet)")
parser.add_argument(
"--gif", choices=["true", "false"], default="false",
help="make a gif (default: false)")
parser.add_argument(
"--reverse", choices=["true", "false"], default="false",
help="make a reverse gif (default: false)")
parser.add_argument(
"--duration", type=float, default=0.1,
help="gif frame duration in seconds (default: 0.1)")
parser.add_argument(
"--loop", choices=["true", "false"], default="false",
help="enable gif loop (default: false)")
group = parser.add_mutually_exclusive_group(required=True)
group.add_argument("image", nargs="?")
group.add_argument(
"--list-layers", action="store_true", help="list layers")
args = parser.parse_args()
if args.list_layers:
list_layers(network=args.network)
else:
zoom = True
if args.zoom == "false":
zoom = False
clip = True
if args.clip == "false":
clip = False
gif = False
if args.gif == "true":
gif = True
reverse = False
if args.reverse == "true":
reverse = True
loop = False
if args.loop == "true":
loop = True
deepdream(
args.image, zoom=zoom, scale_coefficient=args.scale,
irange=args.dreams, iter_n=args.itern, octave_n=args.octaves,
octave_scale=args.octave_scale, end=args.layers, clip=clip,
network=args.network, gif=gif, reverse=reverse,
duration=args.duration, loop=loop)
except Exception as e:
print("Error: {}".format(e))
sys.exit(2)
def main():
try:
parser = ArgumentParser(description="Deep dreamer")
parser.add_argument("--zoom",
choices=["true", "false"],
default="true",
help="zoom dreams (default: true)")
parser.add_argument("--scale",
type=float,
default=0.05,
help="scale coefficient for zoom (default: 0.05)")
parser.add_argument("--dreams",
type=int,
default=100,
help="number of images (default: 100)")
parser.add_argument("--itern",
type=int,
default=10,
help="dream iterations (default: 10)")
parser.add_argument("--octaves",
type=int,
default=4,
help="dream octaves (default: 4)")
parser.add_argument("--octave-scale",
type=float,
default=1.4,
help="dream octave scale (default: 1.4)")
parser.add_argument("--layers",
type=str,
default="inception_4c/output",
help="dream layers (default: inception_4c/output)")
parser.add_argument("--clip",
choices=["true", "false"],
default="true",
help="clip dreams (default: true)")
parser.add_argument(
"--network",
choices=['bvlc_googlenet', 'googlenet_place205'],
default='bvlc_googlenet',
help="choose the network to use (default: bvlc_googlenet)")
parser.add_argument("--gif",
choices=["true", "false"],
default="false",
help="make a gif (default: false)")
parser.add_argument("--reverse",
choices=["true", "false"],
default="false",
help="make a reverse gif (default: false)")
parser.add_argument(
"--duration",
type=float,
default=0.1,
help="gif frame duration in seconds (default: 0.1)")
parser.add_argument("--loop",
choices=["true", "false"],
default="false",
help="enable gif loop (default: false)")
parser.add_argument("--framerate",
type=int,
default=24,
help="framerate for video (default: 24)")
parser.add_argument(
"--gpuid",
type=int,
default=-1,
help="enable GPU with id GPUID (default: disabled)")
group = parser.add_mutually_exclusive_group(required=True)
group.add_argument("image", nargs="?")
group.add_argument("--list-layers",
action="store_true",
help="list layers")
group.add_argument("--video", type=str, help="video file")
args = parser.parse_args()
if args.list_layers:
list_layers(network=args.network)
elif args.video:
clip = True
if args.clip == "false":
clip = False
deepdream_video(args.video,
iter_n=args.itern,
octave_n=args.octaves,
octave_scale=args.octave_scale,
end=args.layers,
clip=clip,
network=args.network,
frame_rate=args.framerate)
else:
zoom = True
if args.zoom == "false":
zoom = False
clip = True
if args.clip == "false":
clip = False
gif = False
if args.gif == "true":
gif = True
reverse = False
if args.reverse == "true":
reverse = True
loop = False
if args.loop == "true":
loop = True
gpu = False
if args.gpuid >= 0:
gpu = True
deepdream(args.image,
zoom=zoom,
scale_coefficient=args.scale,
irange=args.dreams,
iter_n=args.itern,
octave_n=args.octaves,
octave_scale=args.octave_scale,
end=args.layers,
clip=clip,
network=args.network,
gif=gif,
reverse=reverse,
duration=args.duration,
loop=loop,
gpu=gpu,
gpuid=args.gpuid)
except KeyboardInterrupt:
print("\nWaking up...")
sys.exit(2)
except Exception as e:
print("Error: {}".format(e))
sys.exit(2)