Skip to content

coltonweaver/PyTorch-DeepDream

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 

Repository files navigation

PyTorch-DeepDream

A simplified implementation of the DeepDream algorithm using PyTorch. Based on https://hackernoon.com/dl06-deepdream-with-code-5f735052e21f

Dependencies

  • torch
  • torchvision
  • numpy
  • matplotlib
  • PIL (Pillow)

Usage

deep_dream.py [-f FILENAME]

Optional Arguments:

-h, --help show this help message and exit

-l LAYER Desired network layer to push image through to. (Default = 16)

-i ITER Desired iterations to apply to image. (Default = 11)

-lr LR Desired learning rate to use in algorithm. (Default = 0.0012)

About

A simplified implementation of the DeepDream algorithm using PyTorch. Based on https://hackernoon.com/dl06-deepdream-with-code-5f735052e21f

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages