File size: 758 Bytes
29a5ed9 | 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 | p = sd.getp()
c = sd.cond('bunny')
uc = sd.negcond('banana')
latent = sd.sample(p, c, uc)
-- Generate original images
tmp = torch.clamp(torch.mul(torch.add(latent, 1.0), 0.5), 0.0, 1.0)
ui.gallery.add(sd.toimage(tmp))
ui.gallery.add(sd.toimage(sd.vae(latent)))
-- The latent space we got from the sampler has
-- shape [1][4][64][64], lets put random numbers
-- in the upper left corner.
-- (Yes, this is Lua, but the tensors start at 0.)
for z = 0, 3, 1 do
for x = 0, 32, 1 do
for y = 0, 32, 1 do
-- Poke the tensor
latent[0][z][x][y] = math.random()
end
end
end
-- Generate new images
tmp = torch.clamp(torch.mul(torch.add(latent, 1.0), 0.5), 0.0, 1.0)
ui.gallery.add(sd.toimage(tmp))
ui.gallery.add(sd.toimage(sd.vae(latent)))
|