pytorch-deepdream:PyTorch的深梦

时间:2024-05-12 21:22:44
【文件属性】:

文件名称:pytorch-deepdream:PyTorch的深梦

文件大小:1.98MB

文件格式:ZIP

更新时间:2024-05-12 21:22:44

JupyterNotebook

PyTorch深梦 这是PyTorch中Deep Dream的实现。 使用例 import timm import torch from deepdreamer import DeepDreamer from utils import open_image dream = DeepDreamer ( model_name = "inception_v3" , layers_names = [], "cuda:0" ) img = open_image ( "example_img.jpg" ) dream . set_layers ([ "Mixed_5b" ]) dream_img1 = dream . run_deep_dream_with_octaves ( img ) dream . set_layers ([ "Mixed_5c" ]) dream_img2 = dream .


【文件预览】:
pytorch-deepdream-master
----images()
--------dream_img2.png(412KB)
--------example_img.jpg(46KB)
--------dream_img1.png(452KB)
----utils.py(1KB)
----example.ipynb(1.45MB)
----.gitignore(62B)
----README.md(973B)
----deepdreamer.py(4KB)

网友评论