AttnGAN复现问题

从预训练的模型提取epoch

下载好模型，命名与程序epoch提取过程不一致

NET_G: ''  # '../models/bird_AttnGAN2.pth'

 istart = cfg.TRAIN.NET_G.rfind('_') + 1
 iend = cfg.TRAIN.NET_G.rfind('.')
 epoch = cfg.TRAIN.NET_G[istart:iend]
 epoch = int(epoch) + 1

0dim tenser.data[0]

errD.item()

图片存储地址

s_tmp = '%s/single/%s' % (save_dir, keys[j])
folder = s_tmp[:s_tmp.rfind('/')]
if not os.path.isdir(folder):
    print('Make a new folder: ', folder)
    mkdir_p(folder)

skimage