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