search

ProGamerGov / neural-dream

PyTorch implementation of DeepDream algorithm
MIT License
132 stars 19 forks source link

How to use neural_dream.py with own DeepDream model trained on a custom image datasets. #16

Closed Bird-NZ closed 2 years ago

Bird-NZ commented 2 years ago

Hi, I'm wanting to:

  1. use a DeepDream model created with your dream-creater process as seen here: https://github.com/ProGamerGov/dream-creator
  2. and then mount/load this model and use is with the process in neural-dream as seen here: https://github.com/ProGamerGov/neural-dream

As a guess do I need to add the model I created with the dream creator process to the CaffeeLoader.py https://github.com/ProGamerGov/neural-dream/blob/master/neural_dream/CaffeLoader.py

What I'm wanting to do is build models from my own image sets and then manipulate photos with the models I've created. Your neural_dream.py is full of great features that I'm wanting to use but with my own DeepDream model trained on my custom image datasets.

Thx!