-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathevaluate.py
183 lines (183 loc) · 11 KB
/
evaluate.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
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
concept_prompts = {"round_bird": ["Watercolor painting of {} on a branch",
"A house in style of {}",
"Grainy photo of {} in angry birds",
"{} made of chocolate",
"Ukiyo-e painting of {}"],
"cat_statue": ["Painting of two {} fishing on a boat",
"A {} backpack",
"Banksy art of {}",
"A {} themed lunchbox",
"Painting of a {} riding a dragon"],
"red_teapot": ["A carpet with {} embroidery",
"A {} stamp",
"{} fedora",
"Felt {}"],
"clock": ["A photo of {} on the beach",
"A photo of {} on the moon",
"Elmo holding a {}"],
"physics_mug": ["A photo of {} on the beach",
"A photo of {} on the moon",
"Elmo holding a {}"],
"elephant": ["Anime painting of a Kawaii {}",
"A mossy {}",
"{} skeleton in the National Science Museum",
"Negative space painting of {}"],
"thin_bird": ["Painting of {} in the style of Monet",
"A banana {}",
"Advertisement brochure for {}",
"{} cookies"],
"colorful_teapot": ["Manga drawing of a steaming {}",
"A {} watering can",
"{} Death Star",
"A poster for the movie 'The Teapot' starring {}"],
'mug_skulls': ["A {} in the Grand Canyon",
"A {} floating in the sea",
"A {} wearing a Superman outfit"],
"dog": ["A {} in a swimming pool",
"A {} wearing sunglasses",
"A {} oil painting, Ghibli inspired"],
"tortoise_plushy": ["A watercolor painting of {} plushy on a mountain",
"A {} backpack",
"A {} wearing a Superman outfit"],
"teddybear": ["{} in Times Square",
"{} wearing a fedora cap",
"{} wearing a beret cap"],
"table": ["{} and an orange sofa",
"Photo of a {} and a chair in the garden",
"A mossy {}"],
"flower": ["A {} with violet color petals",
"{} in a wooden pot on a table",
"A {} in the Grand Canyon"],
"cat": ["{} wearing sunglasses",
"A chair with the {} sitting on it near a beach",
"{} is sitting inside a wooden pot and looking up"],
"chair": ["{} with a cat sitting on it near a beach",
"Photo of a table and a {} in the garden",
"Ukiyo-e painting of {}"],
"wooden_pot": ["A cat is sitting inside a {} and looking up",
"A flower in the {} on a table",
"Elmo holding a {}"],
"barn": ["{} made of chocolate",
"Photo of a table and a {} in the garden",
"A photo of {} on the moon"],
"backpack": ["a {} on top of pink fabric",
"a {} floating in an ocean of milk",
"a {} on the beach",
"a {} with a city in the background"],
"backpack_dog": ["a {} with a city in the background",
"a {} with the Eiffel Tower in the background",
"a {} on top of a dirt road",
"a cube shaped {}"],
"bear_plushie": ["a {} floating in an ocean of milk",
"a {} with a mountain in the background",
"a {} with a blue house in the background",
"a {} on top of a wooden floor"],
"berry_bowl": ["a {} floating on top of water",
"a wet {}",
"a {} with a mountain in the background",
"a {} on top of a white rug"],
"can": ["a purple {}",
"a {} with a blue house in the background",
"a cube shaped {}",
"a {} on top of a white rug"],
"candle": ["a {} on the beach",
"a {} floating in an ocean of milk",
"a red {}",
"a {} with a city in the background"],
"cat1": ["a {} on top of a dirt road",
"a {} with a blue house in the background",
"a {} on top of a mirror",
"a {} with the Eiffel Tower in the background"],
"cat2": ["a {} in the jungle",
"a {} on top of a mirror",
"a {} with a city in the background",
"a {} on top of a dirt road"],
"clock1": ["a {} in the jungle",
"a {} on top of the sidewalk in a crowded street",
"a {} with a tree and autumn leaves in the background",
"a {} on top of green grass with sunflowers around it"],
"colorful_sneaker": ["a {} with a blue house in the background",
"a purple {}",
"a shiny {}",
"a {} in the jungle"],
"dog1": ["a {} on top of a mirror",
"a {} with a blue house in the background",
"a {} on the beach",
"a {} floating in an ocean of milk"],
"dog2": ["a {} on top of the sidewalk in a crowded street",
"a {} floating in an ocean of milk",
"a {} on top of the sidewalk in a crowded street",
"a {} on top of the sidewalk in a crowded street"],
"dog3": ["a {} on top of the sidewalk in a crowded street",
"a {} on top of a mirror",
"a {} with the Eiffel Tower in the background",
"a red {}"],
"dog5": ["a {} in the jungle",
"a {} with the Eiffel Tower in the background",
"a {} with a tree and autumn leaves in the background",
"a {} on top of the sidewalk in a crowded street"],
"dog6": ["a {} with a mountain in the background",
"a {} on the beach",
"a {} with a blue house in the background",
"a {} on top of green grass with sunflowers around it"],
"dog7": ["a {} floating in an ocean of milk",
"a {} on top of pink fabric",
"a {} floating in an ocean of milk",
"a {} with a mountain in the background"],
"dog8": ["a {} with a city in the background",
"a {} with a blue house in the background",
"a {} on the beach",
"a {} on top of a wooden floor"],
"duck_toy": ["a {} on top of a mirror",
"a {} on top of a mirror",
"a {} on top of a purple rug in a forest",
"a {} in the jungle"],
"fancy_boot": ["a {} on the beach",
"a {} in the jungle",
"a {} with a blue house in the background",
"a {} with a wheat field in the background"],
"grey_sloth_plushie": ["a {} on top of a white rug",
"a {} with a tree and autumn leaves in the background",
"a {} on top of green grass with sunflowers around it",
"a wet {}"],
"monster_toy": ["a {} floating on top of water",
"a {} in the snow",
"a shiny {}",
"a {} on a cobblestone street"],
"pink_sunglasses": ["a {} floating in an ocean of milk",
"a {} with a mountain in the background",
"a {} with a tree and autumn leaves in the background",
"a {} floating in an ocean of milk"],
"poop_emoji": ["a {} on top of a mirror",
"a {} on top of pink fabric",
"a shiny {}",
"a {} on top of a white rug"],
"rc_car": ["a {} with the Eiffel Tower in the background",
"a cube shaped {}",
"a {} on top of green grass with sunflowers around it",
"a {} on top of a purple rug in a forest"],
"red_cartoon": ["a shiny {}",
"a {} in the snow",
"a {} with a mountain in the background",
"a {} on top of a dirt road"],
"robot_toy": ["a cube shaped {}",
"a wet {}",
"a wet {}",
"a red {}"],
"shiny_sneaker": ["a {} on top of pink fabric",
"a {} on top of a dirt road",
"a {} on top of pink fabric",
"a {} on a cobblestone street"],
"teapot": ["a {} with the Eiffel Tower in the background",
"a shiny {}",
"a {} on the beach",
"a cube shaped {}"],
"vase": ["a red {}",
"a {} with a tree and autumn leaves in the background",
"a {} on a cobblestone street",
"a {} on top of a white rug"],
"wolf_plushie": ["a shiny {}",
"a purple {}",
"a {} in the jungle",
"a {} on the beach"],
}