python util_Python util.util方法代码示例

本文整理汇总了Python中util.util.util方法的典型用法代码示例。如果您正苦于以下问题：Python util.util方法的具体用法？Python util.util怎么用？Python util.util使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在模块util.util的用法示例。

在下文中一共展示了util.util方法的25个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于我们的系统推荐出更棒的Python代码示例。

示例1: get_current_visuals

​点赞 6

​

# 需要导入模块: from util import util [as 别名]

# 或者: from util.util import util [as 别名]

def get_current_visuals(self):

fake_B_audio = self.audio_gen_fakes.view(-1, self.opt.sequence_length, self.opt.image_channel_size, self.opt.image_size, self.opt.image_size)

fake_B_image = self.image_gen_fakes.view(-1, self.opt.sequence_length, self.opt.image_channel_size, self.opt.image_size, self.opt.image_size)

real_A = util.tensor2im(self.real_A.data)

oderdict = OrderedDict([('real_A', real_A)])

fake_audio_B = {}

fake_image_B = {}

real_B = {}

for i in range(self.opt.sequence_length):

fake_audio_B[i] = util.tensor2im(fake_B_audio[:, i, :, :, :].data)

fake_image_B[i] = util.tensor2im(fake_B_image[:, i, :, :, :].data)

real_B[i] = util.tensor2im(self.real_videos[:, i, :, :, :].data)

oderdict['real_B_' + str(i)] = real_B[i]

oderdict['fake_audio_B_' + str(i)] = fake_audio_B[i]

oderdict['fake_image_B_' + str(i)] = fake_image_B[i]

return oderdict

开发者ID:Hangz-nju-cuhk，项目名称:Talking-Face-Generation-DAVS，代码行数:19，

示例2: get_latent_space_visualization

​点赞 6

​

# 需要导入模块: from util import util [as 别名]

# 或者: from util.util import util [as 别名]

def get_latent_space_visualization(self,num_interpolate=20,label_1=-1,label_2=-1):

rand_perm = np.random.permutation( self.opt.n_classes )

if label_1 == -1:

label_1 = self.label[0] #rand_perm[0]

if label_2 == -1:

label_2 = self.opt.target_label #rand_perm[1]

alpha_blends = np.linspace(0,1,num_interpolate)

self.label[0] = label_1

output_gate_1 = self.netG.forward_gate(self.label)

self.label[0] = label_2

output_gate_2 = self.netG.forward_gate(self.label)

results={}

results['latent_real_A']=util.tensor2im(self.real_A.data)

results['latent_real_B']=util.tensor2im(self.real_B.data)

for i in range(num_interpolate):

alpha_blend = alpha_blends[i]

output_gate = output_gate_1*alpha_blend + output_gate_2*(1-alpha_blend)

self.fake_B = self.netG.forward_main( self.real_A,output_gate)

results['%d_L_fake_B_inter'%(i)]=util.tensor2im(self.fake_B.data)

return OrderedDict(results)

开发者ID:arnabgho，项目名称:iSketchNFill，代码行数:25，

示例3: add_objects

​点赞 6

​

# 需要导入模块: from util import util [as 别名]

# 或者: from util.util import util [as 别名]

def add_objects(self, click_src, label_tgt, mask, style_id=0):

y, x = click_src[0], click_src[1]

mask = np.transpose(mask, (2, 0, 1))[np.newaxis,...]

idx_src = torch.from_numpy(mask).cuda().nonzero()

idx_src[:,2] += y

idx_src[:,3] += x

# backup current maps

self.backup_current_state()

# update label map

self.label_map[idx_src[:,0], idx_src[:,1], idx_src[:,2], idx_src[:,3]] = label_tgt

for k in range(self.opt.label_nc):

self.net_input[idx_src[:,0], idx_src[:,1] + k, idx_src[:,2], idx_src[:,3]] = 0

self.net_input[idx_src[:,0], idx_src[:,1] + label_tgt, idx_src[:,2], idx_src[:,3]] = 1

# update instance map

self.inst_map[idx_src[:,0], idx_src[:,1], idx_src[:,2], idx_src[:,3]] = label_tgt

self.net_input[:,-1,:,:] = self.get_edges(self.inst_map)

# update feature map

self.set_features(idx_src, self.feat, style_id)

self.fake_image = util.tensor2im(self.single_forward(self.net_input, self.feat_map))

开发者ID:Lotayou，项目名称:everybody_dance_now_pytorch，代码行数:26，

示例4: recurrent_test

​点赞 6

​

# 需要导入模块: from util import util [as 别名]

# 或者: from util.util import util [as 别名]

def recurrent_test(self, step=5):

input_size = self.input_A.cpu().shape

width,height = input_size[3], input_size[2]

results = []

self.real_A = Variable(self.input_A, volatile=True)

self.fake_B = self.netG.forward(self.real_A)

real_A = util.tensor2im(self.real_A.data)

fake_B = util.tensor2im(self.fake_B.data)

results.append(('real_{}_A'.format(0), real_A))

results.append(('fake_{}_B'.format(0), fake_B))

for i in range(1, step):

# rw = random.randint(0, width)

# rh = random.randint(0, height)

rw = int(width/2)

rh = int(height/2)

self.real_A = Variable(self.fake_B.data[:, :, rh:rh + height, rw:rw + width], volatile=True)

self.fake_B = self.netG.forward(self.real_A)

real_A = util.tensor2im(self.real_A.data)

fake_B = util.tensor2im(self.fake_B.data)

results.append(('real_{}_A'.format(i), real_A))

results.append(('fake_{}_B'.format(i), fake_B))

return OrderedDict(results)

开发者ID:jessemelpolio，项目名称:non-stationary_texture_syn，代码行数:24，

示例5: recurrent_test_l2_searching

​点赞 6

​

# 需要导入模块: from util import util [as 别名]

# 或者: from util.util import util [as 别名]

def recurrent_test_l2_searching(self, step=5):

input_size = self.input_A.cpu().shape

width,height = input_size[3], input_size[2]

results = []

self.real_A = Variable(self.input_A, volatile=True)

self.fake_B = self.netG.forward(self.real_A)

real_A = util.tensor2im(self.real_A.data)

fake_B = util.tensor2im(self.fake_B.data)

results.append(('l2_search_real_{}_A'.format(0), real_A))

res