forked from kesara/deepdreamer
/
deepdreamer.py
112 lines (106 loc) · 4.41 KB
/
deepdreamer.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
###############################################################################
# Deep Dreamer
# Author: Kesara Rathnayake ( kesara [at] kesara [dot] lk )
###############################################################################
from argparse import ArgumentParser
import sys
from deepdreamer.deepdreamer import deepdream, deepdream_video, list_layers
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)
if __name__ == "__main__":
main()