MCPcopy Create free account

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

Functions184 in github.com/danier97/ST-MFNet

⨍ Functions 184◇ Types & classes 68
Method__init__
(self)
models/misc/correlation/correlation.py:390
Method__init__
(self, strType)
cupy_module/softsplat.py:363
Method__init__
(self, args, gan_type='GAN')
losses/discriminator.py:20
Method__init__
(self, args)
losses/discriminator.py:61
Method__init__
(self, args)
losses/discriminator.py:114
Method__init__
(self, args)
losses/discriminator.py:156
Method__init__
(self, args)
losses/discriminator.py:209
Method__init__
(self, args)
losses/__init__.py:6
Method__init__
(self, max_level=5)
losses/laplacianpyramid.py:26
Method__init__
(self)
losses/laplacianpyramid.py:62
Method__init__
(self, epsilon=0.001)
losses/charbonnier.py:5
Method__init__
(self, args, gan_type)
losses/adversarial.py:11
Method__len__
(self)
data/datasets.py:50
Method__len__
(self)
data/datasets.py:83
Methodbackward
(self, gradOutput)
models/misc/correlation/correlation.py:333
Methodbackward
(self, gradOutput)
cupy_module/softsplat.py:278
Functioncalc_psnr
args: gt, out -- (B,3,H,W) cuda Tensors
utility.py:223
Functioncalc_ssim
(gt, out, size_average=False)
utility.py:232
Functionestimate
(tenFirst, tenSecond)
models/misc/pwcnet.py:291
Methodeval
(self, model, metrics=['PSNR', 'SSIM'], output_dir=None, output_name=None)
data/testsets.py:90
Methodeval
(self, model, metrics=['PSNR', 'SSIM'], output_dir=None, output_name=None)
data/testsets.py:166
Methodeval
(self, model, metrics=['PSNR', 'SSIM'], output_dir=None, output_name=None)
data/testsets.py:232
Methodeval
(self, model, metrics=['PSNR', 'SSIM'], output_dir=None, output_name='output.png')
data/testsets.py:310
Methodeval
(self, model, metrics=['PSNR', 'SSIM'], output_dir=None, output_name='output.png')
data/testsets.py:362
Methodeval
(self, model, metrics=['PSNR', 'SSIM'], output_dir=None, output_name='output.png')
data/testsets.py:408
Methodextract_pyramid_single
(self, tenFirst)
models/misc/pwcnet.py:283
Methodforward
(self, img1, img2)
metrics/pytorch_ssim/__init__.py:113
Methodforward
(self, x)
models/feature.py:9
Methodforward
(self, x)
models/feature.py:25
Methodforward
(self, x)
models/feature.py:68
Methodforward
(self, tensorCombine)
models/feature.py:99
Methodforward
(self, im0, im2)
models/feature.py:121
Methodforward
(self, im1, im3, im5, im7, im4_tilde)
models/stmfnet.py:45
Methodforward
(self, tensorCombine)
models/stmfnet.py:169
Methodforward
(self, tenFirst, tenSecond, tenFlow)
models/stmfnet.py:205
Methodforward
(self, I0, I1, I2, I3, *args)
models/stmfnet.py:240
Methodforward
(self, tenInput)
models/misc/pwcnet.py:134
Methodforward
(self, tenFirst, tenSecond, objPrevious)
models/misc/pwcnet.py:187
Methodforward
(self, tenInput)
models/misc/pwcnet.py:245
Methodforward
(self, tenFirst, tenSecond, *args)
models/misc/pwcnet.py:263
Methodforward
(self, *args)
models/misc/gridnet.py:30
Methodforward
(self, *args)
models/misc/gridnet.py:124
Methodforward
(self, x)
models/misc/gridnet.py:173
Methodforward
(self, x)
models/misc/gridnet.py:218
Methodforward
(self, x)
models/misc/gridnet.py:257
Methodforward
(self, x)
models/misc/gridnet.py:274
Methodforward
(self, x)
models/misc/gridnet.py:289
Methodforward
(self , x)
models/misc/resnet_3D.py:94
Methodforward
(self, x)
models/misc/resnet_3D.py:123
Methodforward
(self, x)
models/misc/resnet_3D.py:167
Methodforward
(self, x)
models/misc/resnet_3D.py:243
Methodforward
(self, x)
models/misc/resnet_3D.py:397
Methodforward
(self, x)
models/misc/resnet_3D.py:409
Methodforward
(self, x)
models/misc/__init__.py:13
Methodforward
(self, im)
models/misc/__init__.py:25
Methodforward
(self, first, second)
models/misc/correlation/correlation.py:280
Methodforward
(self, tenFirst, tenSecond)
models/misc/correlation/correlation.py:394
Methodforward
(self, input, flow)
cupy_module/softsplat.py:241
Methodforward
(self, tenInput, tenFlow, tenMetric)
cupy_module/softsplat.py:369
Methodforward
(ctx, input, weight, offset_i, offset_j, dilation)
cupy_module/adacof.py:314
Methodforward
(self, x)
losses/discriminator.py:53
Methodforward
(self, f0, f1, f2)
losses/discriminator.py:99
Methodforward
(self, f0, f1)
losses/discriminator.py:147
Methodforward
(self, f0, f1, f2)
losses/discriminator.py:194
Methodforward
(self, f0, f1, f2)
losses/discriminator.py:253
Methodforward
(self, output, gt, input_frames)
losses/__init__.py:43
Methodforward
(self, x, factor=1)
losses/laplacianpyramid.py:16
Methodforward
(self, X)
losses/laplacianpyramid.py:31
Methodforward
(self, x, y)
losses/laplacianpyramid.py:68
Methodforward
(self, output, gt)
losses/charbonnier.py:9
Methodforward
(self, fake, real, input_frames=None)
losses/adversarial.py:36
Methodget_downsample_stride
(stride , temporal_stride)
models/misc/resnet_3D.py:30
Methodget_downsample_stride
(stride)
models/misc/resnet_3D.py:78
Functionmake_optimizer
(args, my_model)
utility.py:10
Functionmake_scheduler
(args, my_optimizer)
utility.py:38
Functionmc3_18
Constructor for 18 layer Mixed Convolution network as in https://arxiv.org/abs/1711.11248 Args: pretrained (bool): If True, returns a
models/misc/resnet_3D.py:323
Methodpredict
Enhance a video by increasing its frame rate using frame interpolation. Parameters: - mp4 (Path): Path to the video file.
predict.py:80
Functionr2plus1d_18
Constructor for the 18 layer deep R(2+1)D network as in https://arxiv.org/abs/1711.11248 Args: pretrained (bool): If True, returns a
models/misc/resnet_3D.py:349
Functionrand_crop
(*args, sz)
data/vfitransforms.py:6
Functionrand_flip
(*args, p)
data/vfitransforms.py:14
Functionrand_reverse
(*args, p)
data/vfitransforms.py:25
Methodsetup
Set up the prediction environment. This method initializes the model, its parameters, and the GPU device for computation. It
predict.py:37
Functionssim
(img1, img2, window_size = 11, size_average = True)
metrics/pytorch_ssim/__init__.py:131
Functionssim_matlab
(img1, img2, window_size=11, window=None, size_average=True, full=False, val_range=255)
metrics/pytorch_ssim/__init__.py:49
← previous101–184 of 184, ranked by callers