Functions — github.com/ZHKKKe/MODNet

CodeHub MCPcopy Create free account

hub / github.com/ZHKKKe/MODNet / functions

Functions73 in github.com/ZHKKKe/MODNet

⨍ Functions 73 ◇ Types & classes 26

↓ 6 callersMethod__init__

(self, in_channels)

torchscript/modnet_torchscript.py:24

↓ 6 callersMethod__init__

(self, in_channels)

src/models/modnet.py:16

↓ 6 callersMethod__init__

(self, in_channels)

onnx/modnet_onnx.py:24

↓ 3 callersFunction_make_divisible

(v, divisor, min_value=None)

src/models/backbones/mobilenetv2.py:15

↓ 3 callersMethodload_pretrained_ckpt

src/models/backbones/wrapper.py:24

↓ 1 callersMethod__init__

(self, in_channels)

src/models/backbones/wrapper.py:14

↓ 1 callersMethod__init__

(self, in_channels, alpha=1.0, expansion=6, num_classes=1000)

src/models/backbones/mobilenetv2.py:91

↓ 1 callersMethod_init_conv

torchscript/modnet_torchscript.py:249

↓ 1 callersMethod_init_conv

src/models/modnet.py:246

↓ 1 callersMethod_init_conv

onnx/modnet_onnx.py:243

↓ 1 callersMethod_init_kernel

src/trainer.py:65

↓ 1 callersMethod_init_norm

torchscript/modnet_torchscript.py:255

↓ 1 callersMethod_init_norm

src/models/modnet.py:252

↓ 1 callersMethod_init_norm

onnx/modnet_onnx.py:249

↓ 1 callersMethod_init_weights

src/models/backbones/mobilenetv2.py:186

↓ 1 callersFunctionconv_1x1_bn

src/models/backbones/mobilenetv2.py:33

↓ 1 callersFunctionconv_bn

(inp, oup, stride)

src/models/backbones/mobilenetv2.py:25

↓ 1 callersMethodforward

torchscript/modnet_torchscript.py:33

↓ 1 callersMethodforward

src/models/modnet.py:25

↓ 1 callersMethodforward

onnx/modnet_onnx.py:33

↓ 1 callersMethodfreeze_norm

src/models/modnet.py:238

↓ 1 callersFunctionget_scale_factor

(im_h, im_w, ref_size)

onnx/inference_onnx.py:43

↓ 1 callersFunctionmatting

(video, result, alpha_matte=False, fps=30)

demo/video_matting/custom/run.py:23

(self, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, b

torchscript/modnet_torchscript.py:44

(self, in_channels, out_channels, reduction=1)

torchscript/modnet_torchscript.py:70

(self, backbone)

torchscript/modnet_torchscript.py:96

(self, hr_channels, enc_channels)

torchscript/modnet_torchscript.py:124

(self, hr_channels, enc_channels)

torchscript/modnet_torchscript.py:174

(self, in_channels=3, hr_channels=32, backbone_arch='mobilenetv2', backbone_pretrained=True)

torchscript/modnet_torchscript.py:205

Arguments: channels (int): Channel for input tensor kernel_size (int): Size of the kernel used in blurring

src/trainer.py:27

(self, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bi

src/models/modnet.py:36

(self, in_channels, out_channels, reduction=1)

src/models/modnet.py:62

(self, backbone)

src/models/modnet.py:88

(self, hr_channels, enc_channels)

src/models/modnet.py:121

(self, hr_channels, enc_channels)

src/models/modnet.py:177

(self, in_channels=3, hr_channels=32, backbone_arch='mobilenetv2', backbone_pretrained=True)

src/models/modnet.py:208

(self, in_channels)

src/models/backbones/wrapper.py:32

(self, inp, oup, stride, expansion, dilation=1)

src/models/backbones/mobilenetv2.py:46

(self, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bia

onnx/modnet_onnx.py:44

(self, in_channels, out_channels, reduction=1)

onnx/modnet_onnx.py:70

(self, backbone)

onnx/modnet_onnx.py:96

(self, hr_channels, enc_channels)

onnx/modnet_onnx.py:124

(self, hr_channels, enc_channels)

onnx/modnet_onnx.py:174

(self, in_channels=3, hr_channels=32, backbone_arch='mobilenetv2', backbone_pretrained=True)

onnx/modnet_onnx.py:205

Method_load_pretrained_model

(self, pretrained_file)

src/models/backbones/mobilenetv2.py:173

torchscript/modnet_torchscript.py:62

torchscript/modnet_torchscript.py:80

torchscript/modnet_torchscript.py:107

(self, img, enc2x, enc4x, lr8x)

torchscript/modnet_torchscript.py:151

(self, img, lr8x, hr2x)

torchscript/modnet_torchscript.py:184

torchscript/modnet_torchscript.py:228

Arguments: x (torch.Tensor): input 4D tensor Returns: torch.Tensor: Blurred version of the input

src/trainer.py:47

src/models/modnet.py:54

src/models/modnet.py:72

(self, img, inference)

src/models/modnet.py:99

(self, img, enc2x, enc4x, lr8x, inference)

src/models/modnet.py:148

(self, img, lr8x, hr2x)

src/models/modnet.py:187

(self, img, inference)

src/models/modnet.py:231

src/models/backbones/wrapper.py:21

src/models/backbones/wrapper.py:38

src/models/backbones/mobilenetv2.py:79

src/models/backbones/mobilenetv2.py:139

onnx/modnet_onnx.py:62

onnx/modnet_onnx.py:80

onnx/modnet_onnx.py:107

(self, img, enc2x, enc4x, lr8x)

onnx/modnet_onnx.py:151

(self, img, lr8x, hr2x)

onnx/modnet_onnx.py:184

onnx/modnet_onnx.py:228

Methodfreeze_norm

torchscript/modnet_torchscript.py:241

Methodfreeze_norm

onnx/modnet_onnx.py:235

Methodload_pretrained_ckpt

src/models/backbones/wrapper.py:74

Functionsoc_adaptation_iter

Self-Supervised sub-objective consistency (SOC) adaptation iteration of MODNet This function fine-tunes MODNet for one iteration in an unlabeled

src/trainer.py:177

Functionsupervised_training_iter

Supervised training iteration of MODNet This function trains MODNet for one iteration in a labeled dataset. Arguments: modnet (torch

src/trainer.py:86