• torch_09_DCGAN_注意的细节


    DCGAN github链接:https://github.com/darr/DCGAN

    DCGAN:
    1.在一次epoch中,如果第i批的i能够整除every_print,则打印到output文件中(打印出来)
    2.训练过程:计算损失,梯度下降
    3.iters:所有epochs累加到一块,迭代总次数
        如果iters能够整除save_print,或者进行到了最后一个epoch and i是最后一个epoch的最后一组,则保存该生成模型
    具体操作:
         1.将生成的64张图片放入G网络中,生成一组假图片测试G网络的效果,这些假图片放入img_lst中
         2.将train_model的字典的state_dict(),使用torch.save(model_dict, .tar)保存到tar文件中

    4.在show image中: 

           1.显示loss值的图片:生成G_D_losses.jpg
         2.将生成的假图片保存成动画      

     1 def _save_img_list(img_list, save_path, config):
     2     #_show_img_list(img_list)
     3     metadata = dict(title='generator images', artist='Matplotlib', comment='Movie support!')
     4     writer = ImageMagickWriter(fps=1,metadata=metadata)
     5     ims = [np.transpose(i, (1, 2, 0)) for i in img_list]
     6     fig, ax = plt.subplots()
     7     with writer.saving(fig, "%s/img_list.gif" % save_path,500):
     8         for i in range(len(ims)):
     9             ax.imshow(ims[i])
    10             ax.set_title("step {}".format(i * config["save_every"]))
    11             writer.grab_frame()

        3.将生成的假图片保存成图片    

     1 def _save_img_list(img_list, save_path): # 假图片的列表,保存路径,
     2     
     3     ims = [np.transpose(i, (1, 2, 0)) for i in img_list]
     4     name_img = 0
     5     for i in range(len(ims)):
     6         plt.figure(figsize=(8, 8))
     7         plt.subplot(1, 2, 1)
     8         plt.axis("off")
     9         str_name = "fake Images"+str(name_img)
    10         plt.title("fake Images"+str(name_img))
    11         name_img += 500
    12         plt.imshow(ims[i])
    13         name = str_name
    14         full_path_name = "%s/%s" % (save_path, name)
    15         plt.savefig(full_path_name) 
  • 相关阅读:
    Omi框架学习之旅
    Omi框架学习之旅
    Omi框架学习之旅
    加密解密
    RSA加密解密
    CMDB后台管理(AutoServer)
    CMDB Autoclient思路分析
    CMDB开发(需求分析)
    Django之model操作(续)
    Django之Model操作
  • 原文地址:https://www.cnblogs.com/shuangcao/p/11796642.html
Copyright © 2020-2023  润新知