neural-style-pt icon indicating copy to clipboard operation
neural-style-pt copied to clipboard

Published 20 hours ago • verified publisher icon ProGamerGov

Support for additional pre trained models

Open Vigilence opened this issue 3 years ago • 1 comments

Hello!

I would like to experiment with different models. Can you please share what steps I would need to take to add support for other models such as these?

Vigilence avatar Jul 24 '22 21:07 Vigilence

@Vigilence The models you suggest are using ResNet architecture, which can require extra work to use for things like style transfer. neural-style-pt uses VGG models by default, but you can add support for them in the modelSelector function in Caffeloader.

ProGamerGov avatar Jul 26 '22 14:07 ProGamerGov