本文整理匯總了Python中util.util.tensor2label方法的典型用法代碼示例。如果您正苦於以下問題：Python util.tensor2label方法的具體用法？Python util.tensor2label怎麽用？Python util.tensor2label使用的例子？那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在類util.util的用法示例。
在下文中一共展示了util.tensor2label方法的3個代碼示例，這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚，您的評價將有助於係統推薦出更棒的Python代碼示例。
示例1: get_current_visuals
# 需要導入模塊: from util import util [as 別名]
# 或者: from util.util import tensor2label [as 別名]
def get_current_visuals(self, getLabel=False):
mask = self.mask
if self.mask is not None:
mask = np.transpose(self.mask[0].cpu().float().numpy(), (1,2,0)).astype(np.uint8)
dict_list = [('fake_image', self.fake_image), ('mask', mask)]
if getLabel: # only output label map if needed to save bandwidth
label = util.tensor2label(self.net_input.data[0], self.opt.label_nc)
dict_list += [('label', label)]
return OrderedDict(dict_list) 示例2: get_current_visuals
# 需要導入模塊: from util import util [as 別名]
# 或者: from util.util import tensor2label [as 別名]
def get_current_visuals(self, getLabel=False):
mask = self.mask
if self.mask is not None:
mask = np.transpose(self.mask[0].cpu().float().numpy(), (1,2,0)).astype(np.uint8)
dict_list = [('fake_image', self.fake_image), ('mask', mask)]
if getLabel: # only output label map if needed to save bandwidth
label = util.tensor2label(self.net_input.data[0], self.opt.label_nc)
dict_list += [('label', label)]
return OrderedDict(dict_list) 示例3: get_current_visuals
# 需要導入模塊: from util import util [as 別名]
# 或者: from util.util import tensor2label [as 別名]
def get_current_visuals(self):
return OrderedDict([
('input_label', util.tensor2label(self.input_label, self.opt.label_nc)),
('input_image', util.tensor2im(self.input_image)),
('real_image', util.tensor2im(self.real_image)),
('synthesized_image', util.tensor2im(self.fake_image))
]) 開發者ID:xcyan，項目名稱:neurips18_hierchical_image_manipulation，代碼行數:9，代碼來源:pix2pixHD_condImgColor_model.py