Example #1
0
#-------------------------------------#
#   运行predict.py可以生成图片
#   生成1x1的图片和5x5的图片
#-------------------------------------#
from dcgan import DCGAN

if __name__ == "__main__":
    save_path_5x5 = "results/predict_out/predict_5x5_results.png"
    save_path_1x1 = "results/predict_out/predict_1x1_results.png"

    dcgan = DCGAN()
    while True:
        img = input('Just Click Enter~')
        dcgan.generate_5x5_image(save_path_5x5)
        dcgan.generate_1x1_image(save_path_1x1)
Example #2
0
#-------------------------------------#
#       对单张图片进行预测
#-------------------------------------#
from dcgan import DCGAN
from PIL import Image

dcgan = DCGAN()
while True:
    img = input('Just Click Enter~')
    dcgan.generate_5x5_image()
    dcgan.generate_1x1_image()