Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/MADF-inpainting/Pytorch-MADF
/ functions
Functions
40 in github.com/MADF-inpainting/Pytorch-MADF
⨍
Functions
40
◇
Types & classes
12
↓ 6 callers
Function
unnormalize
(x)
util/image.py:5
↓ 5 callers
Method
__init__
(self, query_dim, key_dim, value_dim, activation='relu')
net.py:100
↓ 4 callers
Function
gram_matrix
(feat)
loss.py:6
↓ 2 callers
Function
load_ckpt
(ckpt_name, models, optimizers=None)
util/io.py:23
↓ 1 callers
Function
DataParallel_withLoss
(model, extractor, args, **kwargs)
data_parallel.py:47
↓ 1 callers
Method
__init__
(self, list_file, img_transform, mask_transform, return_name=False)
dataset.py:58
↓ 1 callers
Function
evaluate
(model, dataset, device, path)
test.py:24
↓ 1 callers
Function
evaluate
(model, dataset, device, filename)
evaluation.py:8
↓ 1 callers
Function
get_state_dict_on_cpu
(obj)
util/io.py:5
↓ 1 callers
Function
is_image_file
(filename)
dataset.py:17
↓ 1 callers
Function
is_mask_file
(filename)
dataset.py:20
↓ 1 callers
Method
loop
(self)
train.py:30
↓ 1 callers
Function
save_ckpt
(ckpt_name, models, optimizers, n_iter)
util/io.py:13
↓ 1 callers
Function
total_variation_loss
(image)
loss.py:14
↓ 1 callers
Method
train
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
Method
forward
(self, input, mask, outputs, gt, comp_feats, feats, feat_gt)
loss.py:27
Method
forward
(self, image)
net.py:24
Method
forward
(self, input)
net.py:46
Method
forward
(self, features, filters)
net.py:76
Method
forward
(self, query, key)
net.py:112
Method
forward
(self, input, edge_map=None)
net.py:132
Method
forward
(self, input, input_mask)
net.py:220
Method
forward
(self, x, pre_feat)
normalization.py:21
Method
forward
(self, image, mask, gt)
data_parallel.py:11