本文整理汇总了Python中tflib.cifar10方法的典型用法代码示例。如果您正苦于以下问题：Python tflib.cifar10方法的具体用法？Python tflib.cifar10怎么用？Python tflib.cifar10使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类tflib的用法示例。
在下文中一共展示了tflib.cifar10方法的2个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: generate_image
# 需要导入模块: import tflib [as 别名]
# 或者: from tflib import cifar10 [as 别名]
def generate_image(frame, netG):
fixed_noise_128 = torch.randn(128, 128)
if use_cuda:
fixed_noise_128 = fixed_noise_128.cuda(gpu)
noisev = autograd.Variable(fixed_noise_128, volatile=True)
samples = netG(noisev)
samples = samples.view(-1, 3, 32, 32)
samples = samples.mul(0.5).add(0.5)
samples = samples.cpu().data.numpy()
lib.save_images.save_images(samples, './tmp/cifar10/samples_{}.jpg'.format(frame))
# For calculating inception score 示例2: get_data_loader
# 需要导入模块: import tflib [as 别名]
# 或者: from tflib import cifar10 [as 别名]
def get_data_loader(self):
if self.cfg.DATA_LOADER == 'pickle':
return lib.batchloader
if self.cfg.DATA_LOADER == 'cifar-10':
return lib.cifar10
if self.cfg.DATA_LOADER == 'lld-logo':
return lib.twitter_images
if self.cfg.DATA_LOADER == 'hdf5':
return lib.hdf5_images
if self.cfg.DATA_LOADER == 'imagenet-small':
return lib.small_imagenet
# restores a GAN model from checkpoint