MCPcopy Create free account

hub / github.com/Kibeom-Hong/Domain-Aware-Style-Transfer / functions

Functions166 in github.com/Kibeom-Hong/Domain-Aware-Style-Transfer

↓ 33 callersFunctiongram_matrix
(y)
utils.py:417
↓ 26 callersMethodinterpolate
(self, args)
baseline.py:494
↓ 13 callersFunctionfeature_wct_simple
(content_feat, style_feat, alpha=1)
utils.py:265
↓ 12 callersFunctionget_LL_HH
(x)
style_indicator.py:8
↓ 12 callersFunctionimsave
(tensor, path, nrow=4, npadding=0)
data/dataset_util.py:235
↓ 9 callersMethodtrain
(self)
baseline.py:110
↓ 8 callersMethod__init__
(self)
style_indicator.py:17
↓ 8 callersFunction_transformer
(imsize=None, cropsize=None, cencrop=False)
data/dataset_util.py:219
↓ 8 callersFunctionget_LL_HH
(x)
baseline_models.py:25
↓ 8 callersFunctionget_LL_HH
(x)
baseline_models_seg.py:25
↓ 8 callersMethodget_features
(self, x, level)
baseline_models.py:216
↓ 5 callersMethod__init__
(self, use_noise, sigma=0.2)
baseline_models.py:71
↓ 5 callersMethod__init__
(self, use_noise, sigma=0.2)
baseline_models_seg.py:74
↓ 5 callersMethod__init__
(self, root_path, imsize=None, cropsize=None, cencrop=False)
data/dataset_util.py:22
↓ 5 callersMethodget_alphas
(self, imgs)
baseline.py:558
↓ 4 callersFunction_normalizer
(denormalize=False)
data/dataset_util.py:208
↓ 4 callersFunctionones_like
(x)
utils.py:445
↓ 4 callersFunctionsvd
(feat, iden=False, device='cuda:0')
utils.py:14
↓ 3 callersMethod__len__
(self)
data/dataset_util.py:148
↓ 3 callersFunctioncompute_cx
(dist_tilde, band_width)
contextual_utils.py:114
↓ 3 callersFunctioncompute_l2_distance
(x, y)
contextual_utils.py:166
↓ 3 callersFunctioncompute_relative_distance
(dist_raw)
contextual_utils.py:120
↓ 3 callersFunctiongaussian_blur
(x, alpha)
baseline_models.py:32
↓ 3 callersFunctiongaussian_blur
(x, alpha)
baseline_models_seg.py:33
↓ 3 callersFunctionget_rank
(singular_values, dim, eps=0.00001)
utils.py:37
↓ 3 callersFunctionget_squeeze_feat
(feat)
utils.py:31
↓ 2 callersFunctionBw_wct_core
(content_feat, style_feat, weight=1, registers=None, device='cpu')
utils.py:189
↓ 2 callersFunctionSwitchWhiten2d
(x)
utils.py:118
↓ 2 callersFunctioncalc_mean_std
(feat, eps=1e-5)
AdaIN.py:92
↓ 2 callersFunctioncompute_cosine_distance
(x, y)
contextual_utils.py:126
↓ 2 callersFunctioncompute_l1_distance
(x: torch.Tensor, y: torch.Tensor)
contextual_utils.py:152
↓ 2 callersFunctioncompute_label_info
(content_segment, style_segment)
utils.py:391
↓ 2 callersFunctioncontextual_loss_v2
Computes contextual loss between x and y. Args: x: features of shape (N, C, H, W). y: features of shape (N, C, H, W).
contextual_utils.py:194
↓ 2 callersMethodconv2d_with_style_kernels
(self, features, kernels, patch_size, padding='zero', deconv_flag=False)
style_decorator.py:48
↓ 2 callersFunctioncovsqrt_mean
(feature, inverse=False, tolerance=1e-14)
utils.py:46
↓ 2 callersMethoddecode
(self, x, content_skips, style_skips, level, alphas=[], is_recon=True)
baseline_models.py:279
↓ 2 callersMethoddecode
(self, x, content_skips, style_skips, level, content_segment, style_segment, label_set, label_indicator, alpha
baseline_models_seg.py:288
↓ 2 callersMethodencode
(self, x, skips)
baseline_models.py:165
↓ 2 callersMethodencode
(self, x, skips)
baseline_models.py:380
↓ 2 callersMethodencode
(self, x, skips)
baseline_models_seg.py:168
↓ 2 callersMethodencode
(self, x, skips)
baseline_models_seg.py:424
↓ 2 callersFunctionget_HH_LL
(x)
baseline.py:33
↓ 2 callersFunctionget_index
(feat, label)
utils.py:214
↓ 2 callersFunctioninit_weights
(net)
utils.py:286
↓ 2 callersFunctionload_segment
(image_path, image_size=None)
utils.py:376
↓ 2 callersFunctionopen_image
(image_path, image_size=None)
utils.py:326
↓ 2 callersFunctionresize
(feat, target)
utils.py:207
↓ 2 callersFunctionsize_arrange
(x)
baseline_models.py:10
↓ 2 callersFunctionsize_arrange
(x)
baseline_models_seg.py:10
↓ 2 callersFunctionwhitening
(feature)
utils.py:79
↓ 2 callersFunctionzeros_like
(x)
utils.py:442
↓ 1 callersFunctionTVloss
Compute total variation loss. Inputs: - img: PyTorch Variable of shape (1, 3, H, W) holding an input image. - tv_weight: Scalar giving the weight
utils.py:424
↓ 1 callersFunction_normalizer
(denormalize=False)
utils.py:315
↓ 1 callersFunctionadaptive_instance_normalization
(content_feat, style_feat)
AdaIN.py:103
↓ 1 callersMethodbinarization_patch_score
(self, features)
style_decorator.py:77
↓ 1 callersFunctionchange_seg
(seg)
utils.py:341
↓ 1 callersFunctioncoloring
(feature, target)
utils.py:109
↓ 1 callersFunctioncompute_meshgrid
(shape)
contextual_utils.py:181
↓ 1 callersMethodcreate_window
(self, w_size, channel=1)
evaluation.py:42
↓ 1 callersMethoddecode
(self, x, content_skips, style_skips, level, alphas, is_recon=True)
baseline_models.py:383
↓ 1 callersMethoddecode
(self, x, content_skips, style_skips, level, content_segment, style_segment, label_set, label_indicator, alpha
baseline_models_seg.py:427
↓ 1 callersFunctiondenorm
(x)
utils.py:449
↓ 1 callersFunctionextract_patches
(feature, patch_size, stride, padding='zero')
style_decorator.py:6
↓ 1 callersFunctionfeature_wct
(content_feat, style_feat, content_segment=None, style_segment=None, label_set=None, label_indicator=None,
utils.py:252
↓ 1 callersMethodgaussian
(self, w_size, sigma)
evaluation.py:38
↓ 1 callersFunctionget_domainess
(cur_iter, total_iter, batch)
baseline.py:40
↓ 1 callersFunctionget_loader
(image_dir, crop_size=0, image_size=0, batch_size=16, normalize=True, noise=True, split='train', num_workers=2
data/data_loader.py:10
↓ 1 callersMethodkernel_normalize
(self,kernel, k=3, eps=1e-5)
style_decorator.py:35
↓ 1 callersMethodnorm_deconvolution
(self, h, w, patch_size)
style_decorator.py:119
↓ 1 callersMethodreassemble_feature
(self, normalized_content_feature, normalized_style_feature, patch_size, patch_stride)
style_decorator.py:135
↓ 1 callersFunctionstyle_transfer
(vgg, decoder, content, style, alpha=1.0)
AdaIN.py:142
↓ 1 callersMethodtransfer
(self, args)
baseline.py:348
↓ 1 callersMethodtransfer_user_guided
(self, args)
baseline.py:396
↓ 1 callersMethodtrim
(self, video)
data/dataset_util.py:175
↓ 1 callersMethodvideo_transform
(self, video)
data/dataset_util.py:184
↓ 1 callersFunctionwct_core
(cont_feat, styl_feat, weight=1, registers=None, device='cuda:0')
utils.py:149
↓ 1 callersFunctionwct_core_segment
(content_feat, style_feat, content_segment, style_segment, label_set, label_indicator, weight=1, registe
utils.py:204
Method__call__
args: y_true : 4-d ndarray in [batch_size, channels, img_rows, img_cols] y_pred : 4-d ndarray in [batch_size, channel
evaluation.py:14
Method__call__
args: y_true : 4-d ndarray in [batch_size, channels, img_rows, img_cols] y_pred : 4-d ndarray in [batch_size, channel
evaluation.py:48
Method__enter__
(self)
utils.py:308
Method__exit__
(self, exc_type, exc_value, exc_tb)
utils.py:311
Method__getitem__
(self, index)
data/dataset_util.py:34
Method__getitem__
(self, index)
data/dataset_util.py:63
Method__getitem__
(self, index)
data/dataset_util.py:94
Method__getitem__
(self, index)
data/dataset_util.py:123
Method__getitem__
(self, index)
data/dataset_util.py:151
Method__getitem__
(self, index)
data/dataset_util.py:192
Method__init__
(self)
style_indicator.py:86
Method__init__
(self)
style_indicator.py:153
Method__init__
(self)
style_indicator.py:247
Method__init__
(self)
style_indicator.py:341
Method__init__
(self)
style_indicator.py:443
Method__init__
(self, size=256)
style_indicator.py:558
Method__init__
(self, size=256)
style_indicator.py:648
Method__init__
(self, size=256)
style_indicator.py:738
Method__init__
(self, args)
baseline.py:47
Method__init__
(self)
baseline_models.py:84
Method__init__
(self, vgg)
baseline_models.py:102
Method__init__
(self)
baseline_models.py:255
Method__init__
(self, pretrained_vgg=None)
baseline_models.py:371
next →1–100 of 166, ranked by callers