tensorflow-for-deep-learnin.../Deep_Art/bat-country-test/bat_dream2.py
2024-11-17 16:18:55 -08:00

10 lines
No EOL
328 B
Python

#!/usr/bin/env python
""" Create dream 2 """
bc = BatCountry(args.base_model)
features = bc.prepare_guide(Image.open(args.guide_image), end=args.layer)
image = bc.dream(np.float32(Image.open(args.image)), end=args.layer,
iter_n=20, objective_fn=BatCountry.guided_objective,
objective_features=features,)
bc.cleanup()