MCPcopy Create free account

hub / github.com/GuoLanqing/ShadowFormer / functions

Functions140 in github.com/GuoLanqing/ShadowFormer

Methodforward
(self, x, xm, mask=None, img_size = (128, 128))
model.py:792
Methodforward
(self, x, xm, mask=None, img_size=(128,128))
model.py:938
Methodforward
(self, x, xm, mask=None)
model.py:1117
Methodforward
(self, inp)
utils/antialias.py:52
Methodforward
(self, inp)
utils/antialias.py:106
Functionfreeze
(model)
utils/model_utils.py:6
Functionget_arch
(opt)
utils/model_utils.py:56
Functionget_last_path
(path, session)
utils/dir_utils.py:16
Functionimsave
(img, img_path)
utils/image_utils.py:119
Functionis_frozen
(model)
utils/model_utils.py:14
Functionis_image_file
(filename)
utils/image_utils.py:10
Functionis_numpy_file
(filename)
utils/image_utils.py:7
Functionis_pkl_file
(filename)
utils/image_utils.py:16
Functionload_checkpoint
(model, weights)
utils/model_utils.py:23
Functionload_checkpoint_multigpu
(model, weights)
utils/model_utils.py:36
Functionload_npy
(filepath)
utils/image_utils.py:28
Functionload_optim
(optimizer, weights)
utils/model_utils.py:50
Functionload_pkl
(filename_)
utils/image_utils.py:19
Functionload_start_epoch
(weights)
utils/model_utils.py:45
Functionload_val_img
(filepath)
utils/image_utils.py:39
Functionload_val_mask
(filepath)
utils/image_utils.py:60
Functionmkdirs
(paths)
utils/dir_utils.py:5
Methodno_weight_decay
(self)
model.py:154
Methodno_weight_decay
(self)
model.py:1107
Methodno_weight_decay_keywords
(self)
model.py:1111
Functionsave_checkpoint
(model_dir, state, session)
utils/model_utils.py:18
Functionsave_dict
(dict_, filename_)
utils/image_utils.py:24
Functionsave_img
(img, filepath)
utils/image_utils.py:68
Functiontensor2im
Converts a Tensor array into a numpy image array. Parameters: input_image (tensor) -- the input image tensor array imtype (typ
utils/image_utils.py:84
Functiontensor2uint
(img)
utils/image_utils.py:113
Methodtransform0
(self, torch_tensor)
utils/dataset_utils.py:8
Methodtransform1
(self, torch_tensor)
utils/dataset_utils.py:10
Methodtransform2
(self, torch_tensor)
utils/dataset_utils.py:13
Methodtransform3
(self, torch_tensor)
utils/dataset_utils.py:16
Methodtransform4
(self, torch_tensor)
utils/dataset_utils.py:19
Methodtransform5
(self, torch_tensor)
utils/dataset_utils.py:22
Methodtransform6
(self, torch_tensor)
utils/dataset_utils.py:25
Methodtransform7
(self, torch_tensor)
utils/dataset_utils.py:28
Functiontv_loss
Calculates TV loss for an image `x`. Args: x: image, torch.Variable of torch.Tensor beta: See https://arxiv.org/abs/1412.
losses.py:6
Functionunfreeze
(model)
utils/model_utils.py:10
← previous101–140 of 140, ranked by callers