joel-simon / ganbreeder

Breed and share images using biggan
https://ganbreeder.app
GNU General Public License v3.0
1.28k stars 233 forks source link

other models to be used in ganbreeder #25

Open kbuggenhout opened 3 years ago

kbuggenhout commented 3 years ago

Hi,

is there a way to use my own trained model in ganbreeder? do I just change the reference in the ganbreeder ganserver.py.

it would be very interesting to be able to utilize my own trained model. which is being created for a master thesis in graphical design.

thnx in advance Kris

johndpope commented 3 years ago

try changing this line here https://github.com/joel-simon/ganbreeder/blob/3c53c52cce0e66dfd92e11d4103635d657d9414f/gan_server/server.py#L15

you may switch in compatible models https://tfhub.dev/s?q=biggan

if you google fine tuning big gan - you may get other related models and drop them in here https://stackoverflow.com/questions/60578801/how-to-load-tf-hub-model-from-local-system

supposedly - you can just drop in a local file and point tensorflow hub there