Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Index your code
hub
/
github.com/Time-Travel-Rephotography/Time-Travel-Rephotography.github.io
/ functions
Functions
204 in github.com/Time-Travel-Rephotography/Time-Travel-Rephotography.github.io
⨍
Functions
204
◇
Types & classes
60
Method
__init__
(self)
tools/initialize.py:33
Method
__init__
(self, args: Namespace)
tools/initialize.py:106
Method
__init__
(self)
utils/projector_arguments.py:18
Method
__init__
( self, generator: torch.nn, noise_ramp: float = 0.75, noise_strength: float = 0.05,
utils/optimize.py:58
Method
__init__
:param predictor_model_path: path to shape_predictor_68_face_landmarks.dat file
utils/ffhq_dataset/landmarks_detector.py:6
Method
__init__
:param DNN: "TF" or "CAFFE" predictor_model_path: path to shape_predictor_68_face_landmarks.dat file
utils/ffhq_dataset/landmarks_detector.py:30
Method
__init__
(self, params, lr=1e-3, betas=(0.9, 0.999), eps=1e-8, weight_decay=0, degenerated_to_sgd=True)
optim/radam.py:103
Method
__init__
(self, params, lr=1e-3, betas=(0.9, 0.999), eps=1e-8, weight_decay=0, warmup=0)
optim/radam.py:181
Method
__init__
(self, channel, negative_slope=0.2, scale=2 ** 0.5)
op/fused_act.py:74
Method
__init__
(self, channels, kernel_size, sigma, dim=2)
models/gaussian_smoothing.py:21
Method
__init__
(self)
models/degrade.py:32
Method
__init__
(self, spectral_sensitivity: str = 'b')
models/degrade.py:50
Method
__init__
(self, input_size: int, output_size: int, channels: int)
models/degrade.py:73
Method
__init__
( self, in_channels: int, size: int, latent_size: int = 512, activation: str = 'leaky_
models/encoder.py:25
Method
__init__
(self)
models/vggface.py:8
Method
__init__
(self, in_channels: int, out_channels: int, activation: str = 'relu')
models/resnet.py:33
Method
__init__
( self, in_channels: int, out_channels: int, expansion: int = 1, downsampling: int = 1
models/resnet.py:55
Method
__init__
( self, in_channels: int, out_channels: int, bias: bool = False, *args, **kwargs )
models/resnet.py:87
Method
__init__
(self, sibling: torch.Tensor, args: Namespace, size: int = 256)
losses/joint_loss.py:62
Method
__init__
(self, sibling: torch.Tensor, args: Namespace, size: int = 256)
losses/joint_loss.py:79
Method
__init__
( self, init_rgbs, scale_rgb: bool = False )
losses/color_transfer_loss.py:37
Method
__init__
target: target image
losses/reconstruction.py:87
Method
__init__
(self, resize=False)
losses/perceptual_loss.py:14
Method
__init__
(self, weight_path: str = "checkpoint/vgg_face_dag.pt", resize: bool = False)
losses/perceptual_loss.py:54
Method
__repr__
(self)
model.py:126
Method
__repr__
(self)
model.py:164
Method
__repr__
(self)
model.py:231
Method
__setstate__
(self, state)
optim/radam.py:118
Method
__setstate__
(self, state)
optim/radam.py:195
Function
add_arguments
(parser: ArgumentParser)
models/encoder.py:9
Method
add_arguments
(cls, parser: ArgumentParser)
tools/initialize.py:50
Method
add_arguments
(parser: ArgumentParser)
utils/optimize.py:33
Method
add_arguments
(parser: ArgumentParser)
models/degrade.py:17
Method
add_arguments
(parser: ArgumentParser)
losses/joint_loss.py:33
Method
add_arguments
(parser: ArgumentParser)
losses/reconstruction.py:18
Method
backward
(ctx, gradgrad_input, gradgrad_bias)
op/fused_act.py:42
Method
backward
(ctx, grad_output)
op/fused_act.py:63
Method
backward
(ctx, gradgrad_input)
op/upfirdn2d.py:62
Function
cos_loss
(fi, ft)
losses/perceptual_loss.py:9
Function
create_model
(args)
models/encoder.py:14
Method
create_parameters
(latent_coarse)
utils/optimize.py:106
Function
device
(gpu_id=0)
utils/torch_helpers.py:5
Function
encode_color
(imgs: torch.Tensor, args: Namespace)
tools/initialize.py:89
Method
expanded_channels
(self)
models/resnet.py:69
Method
forward
(self, input)
model.py:19
Method
forward
(self, input)
model.py:49
Method
forward
(self, input)
model.py:70
Method
forward
(self, input)
model.py:89
Method
forward
(self, input)
model.py:115
Method
forward
(self, input)
model.py:152
Method
forward
(self, input)
model.py:176
Method
forward
(self, input, style)
model.py:237
Method
forward
(self, image, noise=None)
model.py:287
Method
forward
(self, input)
model.py:301
Method
forward
(self, input, style, noise=None)
model.py:336
Method
forward
(self, input, style, skip=None)
model.py:355
Method
forward
( self, styles, return_latents=False, inject_index=None, truncation=1,
model.py:491
Method
forward
(self, input)
model.py:628
Method
forward
(self, input)
model.py:678
Method
forward
(self, imgs: torch.Tensor)
tools/initialize.py:149
Method
forward
(ctx, grad_output, out, negative_slope, scale)
op/fused_act.py:21
Method
forward
(ctx, input, bias, negative_slope, scale)
op/fused_act.py:53
Method
forward
(self, input)
op/fused_act.py:81
Method
forward
( ctx, grad_output, kernel, grad_kernel, up, down, pad, g_pad, in_size, out_size )
op/upfirdn2d.py:20
Method
forward
(ctx, input, kernel, up, down, pad)
op/upfirdn2d.py:89
Method
forward
Apply gaussian filter to input. Arguments: input (torch.Tensor): Input to apply gaussian filter on. stride fo
models/gaussian_smoothing.py:63
Method
forward
(self, x: torch.Tensor)
models/degrade.py:39
Method
forward
(self, rgb: torch.Tensor)
models/degrade.py:60
Method
forward
(self, im: torch.Tensor)
models/degrade.py:91
Method
forward
(self, img: torch.Tensor, downsample: nn.Module = None)
models/degrade.py:111
Method
forward
(self, input_)
models/encoder.py:20
Method
forward
(self, input_)
models/encoder.py:61
Method
forward
(self, x0)
models/vggface.py:52
Method
forward
(self, x0)
models/vggface.py:110
Method
forward
(self, x)
models/resnet.py:40
Method
forward
(self, img: torch.Tensor)
losses/joint_loss.py:70
Method
forward
(self, img: torch.Tensor)
losses/joint_loss.py:84
Method
forward
(self, noises: Iterable[torch.Tensor])
losses/regularize_noise.py:8
Method
forward
(self, rgbs: List[torch.Tensor], level: int = None)
losses/color_transfer_loss.py:49
Method
forward
img: it should be the degraded version of the generated image
losses/reconstruction.py:69
Method
forward
img: it should be the degraded version of the generated image
losses/reconstruction.py:112
Method
forward
(self, input, target, max_layer=4, cos_dist: bool = False)
losses/perceptual_loss.py:30
Method
forward
(self, input, target, max_layer: int = 4, cos_dist: bool = False)
losses/perceptual_loss.py:65
Method
forward
(self, input, target, eps=1e-8, use_vggface: bool = True, use_vgg=True, max_vgg_layer=4)
losses/perceptual_loss.py:104
Function
fused_bias_act
op/fused_bias_act.cpp:11
Method
get_landmarks
(self, image)
utils/ffhq_dataset/landmarks_detector.py:63
Method
get_latent
(self, input)
model.py:488
Function
get_optimizer_class
(optimizer_name)
optim/__init__.py:13
Function
load_matching_state_dict
(model: nn.Module, state_dict)
utils/torch_helpers.py:11
Method
mean_latent
(self, n_latent)
model.py:480
Function
parent_dir
(path: str)
utils/misc.py:9
Method
should_apply_shortcut
(self)
models/resnet.py:50
Method
should_apply_shortcut
(self)
models/resnet.py:73
Method
step
(self, closure=None)
optim/radam.py:121
Method
step
(self, closure=None)
optim/radam.py:198
Method
to_string
(args: Namespace)
tools/initialize.py:59
Method
to_string
(args: Namespace)
utils/projector_arguments.py:67
Method
to_string
(args: Namespace)
utils/optimize.py:49
Method
to_string
(args: Namespace)
losses/joint_loss.py:46
Method
to_string
(args: Namespace)
losses/reconstruction.py:24
← previous
next →
101–200 of 204, ranked by callers