Spaces:
Build error
Build error
neverix
commited on
Commit
•
17ea825
1
Parent(s):
fbf122b
Fix big bug
Browse files- pulsar_clip.py +6 -6
pulsar_clip.py
CHANGED
@@ -12,11 +12,11 @@ CONFIG_SPEC = [
|
|
12 |
("text", "A cloud at dawn", str),
|
13 |
("iterations", 5000, (0, 7500)),
|
14 |
("seed", 12, int),
|
|
|
15 |
]),
|
16 |
("Rendering", [
|
17 |
("w", 224, [224, 252]),
|
18 |
("h", 224, [224, 252]),
|
19 |
-
# ("show_every", 50, int),
|
20 |
("showoff", 5000, (0, 10000)),
|
21 |
("turns", 4, int),
|
22 |
("focal_length", 0.1, float),
|
@@ -217,11 +217,11 @@ class PulsarCLIP(object):
|
|
217 |
self.scheduler.step()
|
218 |
except AttributeError:
|
219 |
pass
|
220 |
-
|
221 |
-
|
222 |
-
|
223 |
-
|
224 |
-
|
225 |
except KeyboardInterrupt:
|
226 |
pass
|
227 |
|
|
|
12 |
("text", "A cloud at dawn", str),
|
13 |
("iterations", 5000, (0, 7500)),
|
14 |
("seed", 12, int),
|
15 |
+
("show_every", 10, int),
|
16 |
]),
|
17 |
("Rendering", [
|
18 |
("w", 224, [224, 252]),
|
19 |
("h", 224, [224, 252]),
|
|
|
20 |
("showoff", 5000, (0, 10000)),
|
21 |
("turns", 4, int),
|
22 |
("focal_length", 0.1, float),
|
|
|
217 |
self.scheduler.step()
|
218 |
except AttributeError:
|
219 |
pass
|
220 |
+
if i % self.args.show_every == 0:
|
221 |
+
cam_params = self.camera(i / self.args.iterations * np.pi * 2 * self.args.turns, use_random=False)
|
222 |
+
img_show, _ = self.render(cam_params)
|
223 |
+
img = Image.fromarray((img_show.cpu().detach().numpy() * 255).astype(np.uint8))
|
224 |
+
yield img
|
225 |
except KeyboardInterrupt:
|
226 |
pass
|
227 |
|