Generate deep-dreams in PyTorch
Project description
torch-dreams
deep dreams in PyTorch
Less lines of code, more deep-dreams
from torch_dreams.simple import vgg19_dreamer
import cv2 ## for saving images
simple_dreamer = vgg19_dreamer()
dreamed_image = simple_dreamer.dream(
image_path = "your_image.png",
layer_index= 27,
iterations= 2,
size = (256,256)
)
cv2.imwrite("dream.jpg", dreamed_image)
deep-dreams on a video
from torch_dreams.simple import vgg19_dreamer
simple_dreamer = vgg19_dreamer()
simple_dreamer.deep_dream_on_video(
video_path = "sample_videos/tiger_mini.mp4",
save_name = "dream.mp4",
layer = simple_dreamer.layers[13],
octave_scale= 1.3,
num_octaves = 2,
iterations= 2,
lr = 0.09,
size = None,
framerate= 30.0
)
Generating deep dreams with your own PyTorch model
- importing
torch_dreams
from torch_dreams import utils
from torch_dreams import dreamer
import matplotlib.pyplot as plt ## for viewing the deep-dreams
- choosing a model (could be some other PyTorch model as well)
model= models.vgg19(pretrained=True)
model.eval()
- selecting one of the model's layers for the deep-dream
layers = list(model.features.children())
layer = layers[13]
- Defining the torch transforms to be applied before the forward pass (could be any set of torch transforms). Or if you're using the VGG19 like me, you could use
utils.preprocess_func_vgg
andutils.deprocess_func_vgg
preprocess = utils.preprocess_func_vgg
deprocess = utils.deprocess_func_vgg
- Calling an instance of the
dreamer
class and generating a deep-dream
dreamer = dreamer(model, preprocess, deprocess)
dreamed = dreamer.deep_dream(
image_np =image_sample,
layer = layer,
octave_scale = 1.5,
num_octaves = 2,
iterations = 2,
lr = 0.09,
)
plt.imshow(dreamed)
plt.show()
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
torch-dreams-0.0.1.tar.gz
(4.5 kB
view hashes)
Built Distributions
torch_dreams-0.0.1-py3.7.egg
(11.0 kB
view hashes)
Close
Hashes for torch_dreams-0.0.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | e5cf11436f23dcfce0391b05bb9f2c2edb363e362f0620c55762d2aa30f7d188 |
|
MD5 | fd4a15c94beb0b0878f436a9c0e9c345 |
|
BLAKE2b-256 | fb234caa0555444da8b54fc8ae467601a0aa1b56ee74ac45a087a309d1219ba5 |