MCPcopy Create free account

hub / github.com/danier97/ST-MFNet / functions

Functions184 in github.com/danier97/ST-MFNet

⨍ Functions 184◇ Types & classes 68
↓ 12 callersFunctiontensor2rgb
Convert GPU Tensor to RGB image (numpy array)
utility.py:213
↓ 11 callersMethodeval
(self, model, metrics=['PSNR', 'SSIM'], output_dir=None, output_name=None)
data/testsets.py:20
↓ 10 callersMethod__init__
(self, in_ch, out_ch, kernel_size, stride=1, padding=0, bias=True)
models/misc/resnet_3D.py:401
↓ 6 callersMethod__init__
(self, in_chs, out_chs, grid_chs=(32, 64, 96))
models/misc/gridnet.py:194
↓ 5 callersMethod__init__
( self, in_channels, out_channels, kernel_size, stride=1, bias=False, bn=True, act=nn.
losses/discriminator.py:6
↓ 5 callersFunctioncupy_kernel
(strFunction, objVariables)
models/misc/correlation/correlation.py:235
↓ 5 callersFunctioncupy_launch
(strFunction, strKernel)
models/misc/correlation/correlation.py:274
↓ 5 callersMethodexpand
(self, x)
losses/laplacianpyramid.py:49
↓ 5 callersFunctionquantize
(imTensor)
utility.py:209
↓ 4 callersMethod__init__
(self, input_dim, reduction=16)
models/feature.py:14
↓ 4 callersMethod_make_layer
(self, block, conv_builder, planes, blocks, stride=1, temporal_stride=None)
models/misc/resnet_3D.py:251
↓ 4 callersFunctioncupy_kernel
(strFunction, intFilterSize, intDilation, objectVariables)
cupy_module/adacof.py:261
↓ 4 callersFunctioncupy_launch
(strFunction, strKernel)
cupy_module/adacof.py:305
↓ 4 callersFunctionread_frame_yuv2rgb
(stream, width, height, iFrame, bit_depth, pix_fmt='420')
utility.py:155
↓ 3 callersMethod__init__
(self, db_dir)
data/testsets.py:14
↓ 3 callersMethod__init__
(self, args)
models/stmfnet.py:198
↓ 3 callersMethod__init__
(self)
models/misc/pwcnet.py:72
↓ 3 callersFunction_video_resnet
(arch, pretrained=False, progress=True, **kwargs)
models/misc/resnet_3D.py:287
↓ 3 callersFunctioncreate_window
(window_size, channel)
metrics/pytorch_ssim/__init__.py:11
↓ 3 callersFunctioncupy_kernel
(strFunction, objVariables)
cupy_module/softsplat.py:179
↓ 3 callersFunctioncupy_launch
(strFunction, strKernel)
cupy_module/softsplat.py:235
↓ 2 callersMethod__init__
(self)
losses/laplacianpyramid.py:7
↓ 2 callersFunction_ssim
(img1, img2, window, window_size, channel, size_average = True)
metrics/pytorch_ssim/__init__.py:26
↓ 2 callersMethodbackward
(ctx, gradOutput)
cupy_module/adacof.py:365
↓ 2 callersFunctionbackwarp
(tenInput, tenFlow)
models/misc/pwcnet.py:45
↓ 2 callersFunctiongaussian
(window_size, sigma)
metrics/pytorch_ssim/__init__.py:7
↓ 2 callersFunctionmoduleNormalize
(frame)
utility.py:76
↓ 1 callersFunctionFunctionSoftsplat
(tenInput, tenFlow, tenMetric, strType)
cupy_module/softsplat.py:334
↓ 1 callersMethod__init__
(self, *args)
models/misc/__init__.py:11
↓ 1 callersMethod__len__
(self)
data/datasets.py:116
↓ 1 callersMethod_initialize_weights
(self)
models/misc/resnet_3D.py:272
↓ 1 callersFunctioncreate_window_3d
(window_size, channel=1)
metrics/pytorch_ssim/__init__.py:18
↓ 1 callersFunctiondownload_weights
(url, dest)
predict.py:24
↓ 1 callersMethodextract_pyramid
(self, tenFirst, tenSecond)
models/misc/pwcnet.py:280
↓ 1 callersMethodfold_to_patches
args: frames -- list of (1,3,H,W) tensors returns: list of (B,3,h,w) image patches
utility.py:99
↓ 1 callersFunctiongaussian_kernel
(sz, sigma)
utility.py:68
↓ 1 callersMethodget_downsample_stride
(stride)
models/misc/resnet_3D.py:56
↓ 1 callersMethodget_quintuplet
lst -- list of paths of a triplet
data/testsets.py:451
↓ 1 callersFunctionmain
()
train.py:49
↓ 1 callersFunctionmain
()
interpolate_yuv.py:33
↓ 1 callersFunctionmain
()
evaluate.py:26
↓ 1 callersFunctionr3d_18
Construct 18 layer Resnet3D model as in https://arxiv.org/abs/1711.11248 Args: pretrained (bool): If True, returns a model pre-traine
models/misc/resnet_3D.py:297
↓ 1 callersMethodreduce
(self, x)
losses/laplacianpyramid.py:46
↓ 1 callersMethodsave_checkpoint
(self)
trainers/quintuplet.py:94
↓ 1 callersMethodterminate
(self)
trainers/quintuplet.py:98
↓ 1 callersMethodtrain
(self)
trainers/quintuplet.py:31
↓ 1 callersMethodunfold_to_frame
args: patches -- tensor of shape (B,3,h,w) returns: frame -- tensor of shape (1,3,H,W)
utility.py:125
↓ 1 callersMethodvalidate
(self)
trainers/quintuplet.py:65
FunctionFunctionCorrelation
(tenFirst, tenSecond)
models/misc/correlation/correlation.py:385
MethodSubnet_offset
(ks)
models/stmfnet.py:78
MethodSubnet_offset_ds
(ks)
models/stmfnet.py:103
MethodSubnet_offset_us
(ks)
models/stmfnet.py:122
MethodSubnet_weight
(ks)
models/stmfnet.py:90
MethodSubnet_weight_ds
(ks)
models/stmfnet.py:112
MethodSubnet_weight_us
(ks)
models/stmfnet.py:134
Method__getitem__
(self, index)
data/datasets.py:30
Method__getitem__
(self, index)
data/datasets.py:63
Method__getitem__
(self, index)
data/datasets.py:103
Method__init__
(self, height, width, patch_size, overlap)
utility.py:86
Method__init__
(self, window_size = 11, size_average = True)
metrics/pytorch_ssim/__init__.py:106
Method__init__
(self, db_dir, train=True, crop_sz=(256,256), augment_s=True, augment_t=True)
data/datasets.py:13
Method__init__
(self, db_dir, res=None, crop_sz=(256,256), augment_s=True, augment_t=True)
data/datasets.py:55
Method__init__
(self, datasets, p_datasets=None, iter=False, samples_per_epoch=1000)
data/datasets.py:88
Method__init__
(self, db_dir)
data/testsets.py:84
Method__init__
(self, db_dir)
data/testsets.py:160
Method__init__
(self, db_dir)
data/testsets.py:226
Method__init__
(self, db_dir)
data/testsets.py:296
Method__init__
(self, db_dir)
data/testsets.py:344
Method__init__
(self, db_dir, mode='extreme')
data/testsets.py:396
Method__init__
(self, db_dir)
data/testsets.py:472
Method__init__
(self, db_dir)
data/testsets.py:477
Method__init__
(self, db_dir)
data/testsets.py:482
Method__init__
(self, args, train_loader, valid_loader, model, loss, start_epoch=0)
trainers/quintuplet.py:8
Method__init__
(self, *args)
models/feature.py:7
Method__init__
(self, down, cin, cout, ks, stride=1, groups=32, base_width=4, norm_layer=None)
models/feature.py:33
Method__init__
(self, down, cin, cout, ks_s, ks_l, stride, norm_layer)
models/feature.py:92
Method__init__
(self, channels=[64,128,256,512], norm_layer='batch', inplanes=6, **kwargs)
models/feature.py:108
Method__init__
(self, channels=[32,64,96,128], bn=True)
models/stmfnet.py:16
Method__init__
(self, kernel_size)
models/stmfnet.py:74
Method__init__
(self)
models/stmfnet.py:202
Method__init__
(self)
models/misc/pwcnet.py:76
Method__init__
(self, intLevel)
models/misc/pwcnet.py:147
Method__init__
(self)
models/misc/pwcnet.py:225
Method__init__
(self, in_chs, out_chs, grid_chs=(32, 64, 96), n_row=3, n_col=6)
models/misc/gridnet.py:6
Method__init__
(self, in_chs, out_chs, grid_chs=(32, 64, 96), n_row=3, n_col=6, outrow=(0,1,2))
models/misc/gridnet.py:96
Method__init__
(self, in_chs, out_chs, grid_chs=(32, 64, 96), n_row=3, n_col=6)
models/misc/gridnet.py:149
Method__init__
(self, ch_in, ch_out)
models/misc/gridnet.py:246
Method__init__
(self, ch_in, ch_out)
models/misc/gridnet.py:265
Method__init__
(self, ch_in, ch_out)
models/misc/gridnet.py:279
Method__init__
(self, in_planes, out_planes, midplanes=None,
models/misc/resnet_3D.py:14
Method__init__
(self, in_planes, out_planes, midplanes, s
models/misc/resnet_3D.py:39
Method__init__
(self, in_planes, out_planes, midplanes=None,
models/misc/resnet_3D.py:62
Method__init__
(self , inplanes , reduction=16)
models/misc/resnet_3D.py:84
Method__init__
(self, inplanes, planes, conv_builder, stride=1, downsample=None)
models/misc/resnet_3D.py:105
Method__init__
(self, inplanes, planes, conv_builder, stride=1, downsample=None)
models/misc/resnet_3D.py:140
Method__init__
(self, outplanes=32)
models/misc/resnet_3D.py:186
Method__init__
(self)
models/misc/resnet_3D.py:197
Method__init__
Generic resnet video generator. Args: block (nn.Module): resnet building block conv_makers (list(functions)): generat
models/misc/resnet_3D.py:213
Method__init__
(self, in_ch, out_ch, kernel_size, stride, padding, upmode="transpose")
models/misc/resnet_3D.py:377
Method__init__
(self)
models/misc/__init__.py:20
next →1–100 of 184, ranked by callers