MCPcopy Create free account

hub / github.com/MADF-inpainting/Pytorch-MADF / functions

Functions40 in github.com/MADF-inpainting/Pytorch-MADF

↓ 6 callersFunctionunnormalize
(x)
util/image.py:5
↓ 5 callersMethod__init__
(self, query_dim, key_dim, value_dim, activation='relu')
net.py:100
↓ 4 callersFunctiongram_matrix
(feat)
loss.py:6
↓ 2 callersFunctionload_ckpt
(ckpt_name, models, optimizers=None)
util/io.py:23
↓ 1 callersFunctionDataParallel_withLoss
(model, extractor, args, **kwargs)
data_parallel.py:47
↓ 1 callersMethod__init__
(self, list_file, img_transform, mask_transform, return_name=False)
dataset.py:58
↓ 1 callersFunctionevaluate
(model, dataset, device, path)
test.py:24
↓ 1 callersFunctionevaluate
(model, dataset, device, filename)
evaluation.py:8
↓ 1 callersFunctionget_state_dict_on_cpu
(obj)
util/io.py:5
↓ 1 callersFunctionis_image_file
(filename)
dataset.py:17
↓ 1 callersFunctionis_mask_file
(filename)
dataset.py:20
↓ 1 callersMethodloop
(self)
train.py:30
↓ 1 callersFunctionsave_ckpt
(ckpt_name, models, optimizers, n_iter)
util/io.py:13
↓ 1 callersFunctiontotal_variation_loss
(image)
loss.py:14
↓ 1 callersMethodtrain
Override the default train() to freeze the BN parameters
net.py:265
Method__getitem__
(self, index)
dataset.py:45
Method__getitem__
(self, index)
dataset.py:70
Method__init__
(self, extractor, args)
loss.py:21
Method__init__
(self, num_samples)
train.py:21
Method__init__
(self)
net.py:12
Method__init__
(self, in_ch, conv_in_ch, conv_out_ch, kernel_size, stride, padding)
net.py:33
Method__init__
(self, out_ch, kernel_size, stride=0, padding=0, dilation=1, label_ch=0, bn='batchnorm', activ='relu')
net.py:57
Method__init__
(self, in_ch, out_ch, label_ch=0, bn=False, activ='relu')
net.py:119
Method__init__
(self, args, layer_size=7, input_channels=3, upsampling_mode=upsampling)
net.py:141
Method__init__
(self, img_root, mask_root, img_transform, mask_transform)
dataset.py:25
Method__init__
(self, norm_nc, label_nc, upsampling)
normalization.py:6
Method__init__
(self, model, extractor, args)
data_parallel.py:5
Method__iter__
(self)
train.py:24
Method__len__
(self)
train.py:27
Method__len__
(self)
dataset.py:53
Method__len__
(self)
dataset.py:84
Methodforward
(self, input, mask, outputs, gt, comp_feats, feats, feat_gt)
loss.py:27
Methodforward
(self, image)
net.py:24
Methodforward
(self, input)
net.py:46
Methodforward
(self, features, filters)
net.py:76
Methodforward
(self, query, key)
net.py:112
Methodforward
(self, input, edge_map=None)
net.py:132
Methodforward
(self, input, input_mask)
net.py:220
Methodforward
(self, x, pre_feat)
normalization.py:21
Methodforward
(self, image, mask, gt)
data_parallel.py:11