importcv2importnumpyfrom utils importget_image_paths, load_images, stack_imagesfrom training_data importget_training_datafrom model importautoencoder_Afrom model importautoencoder_Bfrom model importencoder, decoder_A, decoder_Btry:
encoder .load_weights("models/encoder.h5")
decoder_A.load_weights("models/decoder_A.h5")
decoder_B.load_weights("models/decoder_B.h5")except:pass
defsave_model_weights():
encoder .save_weights("models/encoder.h5")
decoder_A.save_weights("models/decoder_A.h5")
decoder_B.save_weights("models/decoder_B.h5")print( "save model weights")
images_A= get_image_paths( "data/trump")
images_B= get_image_paths( "data/cage")
images_A= load_images( images_A ) / 255.0images_B= load_images( images_B ) / 255.0images_A+= images_B.mean( axis=(0,1,2) ) - images_A.mean( axis=(0,1,2) )print( "