本文整理匯總了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