本文整理匯總了Python中config.random_seed方法的典型用法代碼示例。如果您正苦於以下問題:Python config.random_seed方法的具體用法?Python config.random_seed怎麽用?Python config.random_seed使用的例子?那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在類config
的用法示例。
在下文中一共展示了config.random_seed方法的1個代碼示例,這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚,您的評價將有助於係統推薦出更棒的Python代碼示例。
示例1: predict_gan
# 需要導入模塊: import config [as 別名]
# 或者: from config import random_seed [as 別名]
def predict_gan():
separate_funcs = False
drange_net = [-1,1]
drange_viz = [-1,1]
image_grid_size = None
image_grid_type = 'default'
resume_network = 'pre-trained_weight'
np.random.seed(config.random_seed)
if resume_network:
print("Resuming weight from:"+resume_network)
G = Generator(num_channels=3, resolution=128, label_size=0, **config.G)
G = load_G_weights(G,resume_network,True)
print(G.summary())
# Misc init.
if image_grid_type == 'default':
if image_grid_size is None:
w, h = G.output_shape[1], G.output_shape[2]
print("w:%d,h:%d"%(w,h))
image_grid_size = np.clip(int(1920 // w), 3, 16).astype('int'), np.clip(1080 / h, 2, 16).astype('int')
print("image_grid_size:",image_grid_size)
else:
raise ValueError('Invalid image_grid_type', image_grid_type)
result_subdir = misc.create_result_subdir('pre-trained_result', config.run_desc)
for i in range(1,6):
snapshot_fake_latents = random_latents(np.prod(image_grid_size), G.input_shape)
snapshot_fake_images = G.predict_on_batch(snapshot_fake_latents)
misc.save_image_grid(snapshot_fake_images, os.path.join(result_subdir, 'pre-trained_%03d.png'%i), drange=drange_viz, grid_size=image_grid_size)