:param model: :return:
(model)
| 71 | return True |
| 72 | |
| 73 | def generagedImage(model): |
| 74 | """ |
| 75 | :param model: |
| 76 | :return: |
| 77 | """ |
| 78 | random_noise = torch.randn(size=(16,config.Z_DIM,1,1), device='cpu') |
| 79 | alpha = 1e-5 # start with very low alpha |
| 80 | steps = 5 |
| 81 | genImg = model(random_noise,alpha,steps).detach().cpu().numpy() |
| 82 | genImg = np.squeeze(genImg) |
| 83 | fig = plt.figure(figsize=(4, 4)) |
| 84 | for i in range(16): |
| 85 | plt.subplot(4, 4, i + 1) |
| 86 | # 由于生成器输出的结果为[-1,1]之间，所以需要转换为[0,1]之间 |
| 87 | plt.imshow(np.transpose((genImg[i] + 1) / 2)) |
| 88 | plt.axis('off') |
| 89 | plt.savefig('cur.png') |
| 90 | |
| 91 | def loadModel(model,root): |
| 92 | """ |