#-------------------------------------# # 运行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)
#-------------------------------------# # 对单张图片进行预测 #-------------------------------------# 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()