def dream(self, image, zoom=True, scale=0.05, dreams=10, itern=10, octaves=4, octave_scale=1.4, layers='inception_4c/output', clip=True, # choices=['bvlc_googlenet', 'googlenet_place205'] network='bvlc_googlenet', gif=True, reverse=False, # of one gif frame duration=1, loop=False, # for video framerate=24, gpuid=0, gpu=True): try: deepdream( image, zoom=zoom, scale_coefficient=scale, irange=dreams, iter_n=itern, octave_n=octaves, octave_scale=octave_scale, end=layers, clip=clip, network=network, gif=gif, reverse=reverse, duration=duration, loop=loop, gpu=gpu, gpuid=gpuid) except Exception as e: print("Error: {}".format(e)) sys.exit(2) finally: self.purge(self.upload_path(), self.FILE_NAME + '_*')
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)