Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Neural-video-delivery/CaFM-Pytorch-ICCV2021
/ functions
Functions
187 in github.com/Neural-video-delivery/CaFM-Pytorch-ICCV2021
⨍
Functions
187
◇
Types & classes
49
Method
__init__
(self, datasets)
src/data/__init__.py:8
Method
__init__
(self, args, name='Video', train=False, benchmark=False)
src/data/video.py:13
Method
__init__
(self, args, name='', train=True, benchmark=False)
src/data/div2kjpeg.py:6
Method
__init__
(self, args, name='Demo', train=False, benchmark=False)
src/data/demo.py:12
Method
__init__
(self, args, name='ITW2K', train=True, benchmark=False)
src/data/div2k.py:5
Method
__init__
(self, args)
src/loss/discriminator.py:9
Method
__init__
(self, args, ckp)
src/loss/__init__.py:16
Method
__init__
(self, conv_index, rgb_range=1)
src/loss/vgg.py:9
Method
__init__
(self, args, gan_type)
src/loss/adversarial.py:13
Method
__iter__
(self)
src/dataloader.py:156
Method
__len__
(self)
src/data/srdata.py:105
Method
__len__
(self)
src/data/video.py:39
Method
__len__
(self)
src/data/demo.py:34
Method
_block
(_in_channels, _out_channels, stride=1)
src/loss/discriminator.py:16
Method
_forward
(x)
src/loss/vgg.py:25
Function
_ms_loop
(dataset, index_queue, data_queue, done_event, collate_fn, scale, seed, init_fn, worker_id)
src/dataloader.py:22
Method
_prepare
(tensor)
src/trainer_cafm.py:164
Method
_prepare
(tensor)
src/videotester.py:67
Method
_scan
(self)
src/data/div2k.py:21
Method
_set_filesystem
(self, dir_data)
src/data/benchmark.py:17
Method
_set_filesystem
(self, dir_data)
src/data/div2kjpeg.py:12
Method
_set_filesystem
(self, dir_data)
src/data/div2k.py:28
Method
_transform
(v, op)
src/model/__init__.py:188
Function
augment
(*args, hflip=True, rot=True)
src/data/common.py:65
Method
basic_block
(args, in_channels, out_channels, act)
src/model/vdsrr.py:27
Method
bg_target
(queue)
src/utility.py:134
Function
calc_psnr
(sr, hr, scale, rgb_range, dataset=None)
src/utility.py:170
Function
default_conv
(in_channels, out_channels, kernel_size, bias=True)
src/model/common.py:8
Method
display_loss
(self, batch)
src/loss/__init__.py:149
Method
forward
(self, x)
src/model/rcan.py:25
Method
forward
(self, x)
src/model/rcan.py:42
Method
forward
(self, x,num)
src/model/rcan.py:69
Method
forward
(self, x,num)
src/model/rcan.py:94
Method
forward
(self, x, num)
src/model/rcan.py:147
Method
forward
(self, x)
src/model/mdconv.py:21
Method
forward
(self, x)
src/model/mdconv.py:40
Method
forward
(self, x)
src/model/mdconv.py:60
Method
forward
(self, x)
src/model/espcn.py:25
Method
forward
(self, x, num)
src/model/espcn.py:56
Method
forward
(self, x, idx_scale, num)
src/model/__init__.py:42
Method
forward
(self, x, num)
src/model/vdsrr.py:51
Method
forward
(self, input, num)
src/model/common.py:56
Method
forward
(self, x)
src/model/common.py:72
Method
forward
(self, input, num)
src/model/common.py:103
Method
forward
(self, input, num)
src/model/common.py:146
Method
forward
(self, x)
src/model/common.py:177
Method
forward
(self, x, num)
src/model/edsr.py:64
Method
forward
(self, x)
src/model/mdsr.py:51
Method
forward
(self, x)
src/model/srcnn.py:27
Method
forward
(self, x, num)
src/model/srcnn.py:50
Method
forward
(self, x, krl=None)
src/model/srcnn.py:75
Method
forward
(self, x)
src/model/rdn.py:23
Method
forward
(self, x)
src/model/rdn.py:42
Method
forward
(self, x)
src/model/rdn.py:93
Method
forward
(self, x)
src/model/ddbpn.py:55
Method
forward
(self, x)
src/model/ddbpn.py:112
Method
forward
(self, x)
src/loss/discriminator.py:50
Method
forward
(self, sr, hr, hr_r, sr_r, sr_r_g)
src/loss/__init__.py:73
Method
forward
(self, sr, hr)
src/loss/vgg.py:24
Method
forward
(self, fake, real)
src/loss/adversarial.py:35
Method
forward_chop
(self, *args, shave=10, min_size=160000)
src/model/__init__.py:128
Function
get_patch
(*args, patch_size=96, scale=2, multi=False, input_large=False, psnr_index=None, data_partion=0)
src/data/common.py:8
Method
load_state_dict
(self, state_dict, strict=True)
src/model/espcn.py:83
Method
load_state_dict
(self, state_dict, strict=True)
src/model/edsr.py:84
Function
make_model
(args, parent=False)
src/model/rcan.py:8
Function
make_model
(args, parent=False)
src/model/espcn.py:7
Function
make_model
(args, parent=False)
src/model/vdsrr.py:10
Function
make_model
(args, parent=False)
src/model/edsr.py:19
Function
make_model
(args, parent=False)
src/model/mdsr.py:10
Function
make_model
(args, parent=False)
src/model/srcnn.py:9
Function
make_model
(args, parent=False)
src/model/rdn.py:10
Function
make_model
(args, parent=False)
src/model/ddbpn.py:10
Function
make_optimizer
make optimizer and scheduler together
src/utility.py:188
Function
make_testChunk
(length, filename)
src/utility.py:260
Function
make_trainChunk
(num, length, segnum)
src/utility.py:250
Function
np2Tensor
(*args, rgb_range=255)
src/data/common.py:55
Function
quantize
(img, rgb_range)
src/utility.py:166
Method
reset
(self)
src/utility.py:40
Method
save
(self, save_dir)
src/utility.py:226
Method
save
(self, apath)
src/loss/__init__.py:177
Method
save_everyepoch
(self, trainer, epoch, is_best=False)
src/utility.py:93
Function
set_channel
(*args, n_channels=3)
src/data/common.py:40
Method
set_scale
(self, idx_scale)
src/data/srdata.py:151
Method
set_scale
(self, idx_scale)
src/data/__init__.py:13
Method
set_scale
(self, idx_scale)
src/data/video.py:42
Function
set_template
(args)
src/template.py:1
Method
test
(self)
src/videotester.py:22
← previous
101–187 of 187, ranked by callers