MCPcopy Create free account

hub / github.com/creotiv/hdrnet-pytorch / functions

Functions48 in github.com/creotiv/hdrnet-pytorch

⨍ Functions 48◇ Types & classes 11
↓ 6 callersMethod__init__
(self, inc , outc, activation=nn.ReLU, batch_norm=False)
model.py:36
↓ 4 callersFunctionlerp_weight
Linear interpolation weight from a sample at x to xs. Returns the linear interpolation weight of a "query point" at coordinate `x` with respect to
slice.py:3
↓ 3 callersFunctionweight_z
ops/cuda/bilateral_slice.cu.cc:32
↓ 2 callersFunctiondiff_abs
ops/cuda/bilateral_slice.cu.cc:22
↓ 2 callersFunctionload_image
(filename, size=None, use_crop=False)
utils.py:26
↓ 2 callersFunctionload_params
(state)
utils.py:47
↓ 2 callersFunctionresize
(img, size=512, strict=False)
utils.py:6
↓ 2 callersFunctionsmoothed_lerp_weight
Smoothed version of `LerpWeight` with gradients more suitable for backprop. Let f(x, xs) = LerpWeight(x, xs) = max(1 - |x - xs|, 0)
slice.py:27
↓ 2 callersFunctiontest
(ckpt, args={})
test.py:13
↓ 1 callersFunction_bilateral_slice
Slices a bilateral grid using the a guide image. Args: grid: The bilateral grid with shape (gh, gw, gd, gc). guide: A guide image with
slice.py:52
↓ 1 callersMethodbackward
(ctx, grad_output)
ops/ops.py:25
↓ 1 callersFunctionbatch_bilateral_slice
(grid, guide)
slice.py:134
↓ 1 callersFunctioncrop
(img, size=512)
utils.py:17
↓ 1 callersFunctiond_diff_abs
ops/cuda/bilateral_slice.cu.cc:27
↓ 1 callersFunctiond_weight_z
ops/cuda/bilateral_slice.cu.cc:37
↓ 1 callersFunctionget_latest_ckpt
(path)
utils.py:35
↓ 1 callersMethodlist_files
(self, in_path)
dataset.py:47
↓ 1 callersFunctionpsnr
(target, prediction)
metrics.py:5
↓ 1 callersFunctionsave_params
(state, params)
utils.py:43
↓ 1 callersFunctionsmoothed_abs
A smoothed version of |x| with improved numerical stability.
slice.py:22
↓ 1 callersFunctiontrain
(params=None)
train.py:24
FunctionBilateralSliceApplyGradKernelLauncher
ops/cuda/bilateral_slice.cu.cc:357
FunctionBilateralSliceApplyGridGradKernel
ops/cuda/bilateral_slice.cu.cc:115
FunctionBilateralSliceApplyGuideGradKernel
ops/cuda/bilateral_slice.cu.cc:201
FunctionBilateralSliceApplyInputGradKernel
ops/cuda/bilateral_slice.cu.cc:273
FunctionBilateralSliceApplyKernel
ops/cuda/bilateral_slice.cu.cc:47
FunctionBilateralSliceApplyKernelLauncher
// -- KERNEL LAUNCHERS ---------------------------------------------------------
ops/cuda/bilateral_slice.cu.cc:337
Method__getitem__
(self, idx)
dataset.py:35
Method__init__
(self, image_path, params=None, suffix='', aug=False)
dataset.py:9
Method__init__
(self, inc , outc, kernel_size=3, padding=1, stride=1, use_bias=True, activation=nn.ReLU, batch_norm=False)
model.py:16
Method__init__
(self)
model.py:57
Method__init__
(self)
model.py:86
Method__init__
(self, params=None)
model.py:125
Method__init__
(self, nin=4, nout=3, params=None)
model.py:136
Method__init__
(self, params)
model.py:222
Method__init__
(self)
ops/ops.py:42
Method__len__
(self)
dataset.py:32
Methodforward
(self, x,y)
model.py:12
Methodforward
(self, x)
model.py:27
Methodforward
(self, x)
model.py:48
Methodforward
(self, bilateral_grid, guidemap)
model.py:60
Methodforward
Affine: r = a11*r + a12*g + a13*b + a14 g = a21*r + a22*g + a23*b + a24 ...
model.py:89
Methodforward
(self, x)
model.py:131
Methodforward
(self, lowres_input)
model.py:182
Methodforward
(self, lowres, fullres)
model.py:230
Methodforward
(ctx, grid, guide, image)
ops/ops.py:13
Methodforward
(self, grid, guide, image)
ops/ops.py:45
Functiontrace_bilateral_slice
(grid, guide)
slice.py:140