anishathalye / neural-style

Neural style in TensorFlow! 🎨
https://anishathalye.com/an-ai-that-can-mimic-any-artist/
GNU General Public License v3.0
5.54k stars 1.52k forks source link

i want to use another pre trained model #157

Closed Kim-kyung-jin closed 4 years ago

Kim-kyung-jin commented 4 years ago

I want to test using the latest cnn model such as inception, not vgg. May I ask how?

anishathalye commented 4 years ago

This involves writing some code, e.g. replacing vgg.py with an inception.py to load Inception, and updating stylize.py to use the appropriate CONTENT_LAYERS and STYLE_LAYERS.

Kim-kyung-jin commented 4 years ago

Thank you !!!!!!!!!!