MCPcopy Create free account

hub / github.com/XingangPan/DragGAN / functions

Functions1,199 in github.com/XingangPan/DragGAN

↓ 1 callersFunctionmain
( ctx: click.Context, raw_img_dir: str, raw_seg_dir: str, outdir: str)
stylegan_human/bg_white.py:40
↓ 1 callersFunctionmain
( ctx: click.Context, face_network: str, body_network: str, face_seed: int,
stylegan_human/insetgan.py:347
↓ 1 callersFunctionmain
( ctx: click.Context, ckpt_path: str, attr_name: str, truncation: float, gen_video: bool,
stylegan_human/edit.py:55
↓ 1 callersFunctionmain
Train a GAN using the techniques described in the paper "Training Generative Adversarial Networks with Limited Data". Examples: \b #
stylegan_human/training_scripts/sg2/train.py:458
↓ 1 callersFunctionmain
Train a GAN using the techniques described in the paper "Alias-Free Generative Adversarial Networks". Examples: \b # Train StyleGAN3
stylegan_human/training_scripts/sg3/train.py:170
↓ 1 callersFunctionmain
Main function to download files from URLs in a config file.
scripts/download_model.py:57
↓ 1 callersFunctionmake_cache_dir_path
(*paths: str)
stylegan_human/dnnlib/util.py:125
↓ 1 callersFunctionmake_cache_dir_path
(*paths: str)
dnnlib/util.py:124
↓ 1 callersFunctionmake_dataset
(dir)
stylegan_human/utils/data_utils.py:28
↓ 1 callersFunctionmake_latent_interp_animation
(G, code1, code2, img1, img2, num_interps, noise_mode, save_mid_image, truncation_psi,device, outdir,fps)
stylegan_human/interpolation.py:65
↓ 1 callersFunctionmake_transform
(translate: Tuple[float,float], angle: float)
gen_images.py:57
↓ 1 callersMethodmaximize
(self)
gui_utils/glfw_window.py:109
↓ 1 callersMethodmodel
(self)
stylegan_human/PP_HumanSeg/deploy/infer.py:50
↓ 1 callersFunctionmodulated_conv2d
( x, # Input tensor: [batch_size, in_channels, in_height, in_width] w,
stylegan_human/training/networks_stylegan3.py:26
↓ 1 callersFunctionmodulated_conv2d
( x, # Input tensor: [batch_size, in_channels, in_height, in_width] w,
stylegan_human/training_scripts/sg3/training/networks_stylegan3.py:27
↓ 1 callersFunctionmodulated_conv2d
( x, # Input tensor: [batch_size, in_channels, in_height, in_width] w,
training/networks_stylegan3.py:26
↓ 1 callersFunctionnan_to_num
(input, nan=0.0, posinf=None, neginf=None, *, out=None)
torch_utils/misc.py:49
↓ 1 callersMethodnum
r"""Returns the number of scalars that were accumulated for the given statistic between the last two calls to `update()`, or zero if n
torch_utils/training_stats.py:180
↓ 1 callersMethodnum
r"""Returns the number of scalars that were accumulated for the given statistic between the last two calls to `update()`, or zero if n
stylegan_human/torch_utils/training_stats.py:182
↓ 1 callersFunctionnumpy_to_tensor
(x)
stylegan_human/utils/util.py:73
↓ 1 callersFunctionpersistent_class
r"""Class decorator that extends a given class to save its source code when pickled. Example: from torch_utils import persistence
torch_utils/persistence.py:35
↓ 1 callersFunctionpersistent_class
r"""Class decorator that extends a given class to save its source code when pickled. Example: from torch_utils import persistence
stylegan_human/torch_utils/persistence.py:37
↓ 1 callersMethodpop_drag_and_drop_paths
(self)
gui_utils/glfw_window.py:147
↓ 1 callersMethodpreprocess
(self, img)
stylegan_human/PP_HumanSeg/deploy/infer.py:89
↓ 1 callersMethodprint_error
(self, error)
visualizer_drag.py:78
↓ 1 callersFunctionrecurse
(obj)
torch_utils/persistence.py:236
↓ 1 callersFunctionreport
r"""Broadcasts the given set of scalars to all interested instances of `Collector`, across device and process boundaries. This function is ex
torch_utils/training_stats.py:56
↓ 1 callersFunctionreport
r"""Broadcasts the given set of scalars to all interested instances of `Collector`, across device and process boundaries. This function is ex
stylegan_human/torch_utils/training_stats.py:58
↓ 1 callersMethodreset_optimizer_state
Reset internal state of the underlying optimizer.
stylegan_human/dnnlib/tflib/optimizer.py:268
↓ 1 callersFunctionrotate2d
(theta, **kwargs)
stylegan_human/training/augment.py:88
↓ 1 callersFunctionrotate2d
(theta, **kwargs)
training/augment.py:88
↓ 1 callersFunctionrotate3d
(v, theta, **kwargs)
stylegan_human/training/augment.py:95
↓ 1 callersFunctionrotate3d
(v, theta, **kwargs)
training/augment.py:95
↓ 1 callersFunctionrun
(args)
stylegan_human/alignment.py:101
↓ 1 callersFunctionrun_PTI
(run_name='', use_wandb=False, use_multi_id_training=False)
stylegan_human/run_pti.py:16
↓ 1 callersFunctionscale3d
(sx, sy, sz, **kwargs)
stylegan_human/training/augment.py:80
↓ 1 callersFunctionscale3d
(sx, sy, sz, **kwargs)
training/augment.py:80
↓ 1 callersMethodset_args
(self, **args)
visualizer_drag.py:281
↓ 1 callersMethodset_encoder
(self)
stylegan_human/pti/pti_models/e4e/psp.py:29
↓ 1 callersMethodset_font_size
(self, target)
gui_utils/imgui_window.py:68
↓ 1 callersMethodset_points
(self, points)
viz/drag_widget.py:70
↓ 1 callersFunctionset_vars
Set the values of given tf.Variables. Equivalent to the following, but more efficient and does not bloat the tf graph: tflib.run([tf.assign(v
stylegan_human/dnnlib/tflib/tfutil.py:196
↓ 1 callersMethodset_vsync
(self, vsync)
gui_utils/glfw_window.py:118
↓ 1 callersFunctionsetup_snapshot_image_grid
(training_set, random_seed=0)
stylegan_human/training/training_loop.py:31
↓ 1 callersFunctionsetup_snapshot_image_grid
(training_set, random_seed=0)
training/training_loop.py:31
↓ 1 callersFunctionsetup_training_loop_kwargs
( # General options (not included in desc). gpus = None, # Number of GPUs: <int>, default = 1 gp
stylegan_human/training_scripts/sg2/train.py:35
↓ 1 callersMethodshould_close
(self)
gui_utils/glfw_window.py:127
↓ 1 callersMethodskip_frames
(self, num)
gui_utils/glfw_window.py:133
↓ 1 callersMethodspace_regulizer_loss
(self, new_G, w_batch, use_wandb)
stylegan_human/pti/training/coaches/localitly_regulizer.py:61
↓ 1 callersMethodstd
r"""Returns the standard deviation of the scalars that were accumulated for the given statistic between the last two calls to `update(
torch_utils/training_stats.py:198
↓ 1 callersFunctionstyle_mixing_video
(network_pkl: str, src_seed: List[int], # Seed of the source image style
stylegan_human/stylemixing_video.py:49
↓ 1 callersFunctionsubprocess_fn
(rank, args, temp_dir)
stylegan_human/training_scripts/sg2/train.py:378
↓ 1 callersFunctionsubprocess_fn
(rank, c, temp_dir)
stylegan_human/training_scripts/sg3/train.py:29
↓ 1 callersFunctiontensor_to_numpy
(x)
stylegan_human/utils/util.py:66
↓ 1 callersMethodto_device
(self, buf)
viz/renderer.py:187
↓ 1 callersMethodtrain
(self)
stylegan_human/pti/training/coaches/single_id_coach.py:16
↓ 1 callersFunctiontranslate3d
(tx, ty, tz, **kwargs)
stylegan_human/training/augment.py:65
↓ 1 callersFunctiontranslate3d
(tx, ty, tz, **kwargs)
training/augment.py:65
↓ 1 callersMethodundo_loss_scaling
Undo the effect of dynamic loss scaling for the given expression.
stylegan_human/dnnlib/tflib/optimizer.py:284
↓ 1 callersMethodupdate_lr
(self, lr)
viz/renderer.py:265
↓ 1 callersFunctionupfirdn2d_native
( input, kernel, up_x, up_y, down_x, down_y, pad_x0, pad_x1, pad_y0, pad_y1 )
stylegan_human/torch_utils/op_edit/upfirdn2d.py:161
↓ 1 callersFunctionupfirdn2d_native
( input, kernel, up_x, up_y, down_x, down_y, pad_x0, pad_x1, pad_y0, pad_y1 )
stylegan_human/pti/pti_models/e4e/stylegan2/op/upfirdn2d.py:19
↓ 1 callersFunctionupfirdn_2d
r"""Pad, upsample, FIR filter, and downsample a batch of 2D images. Accepts a batch of 2D images of the shape `[majorDim, inH, inW, minorDim]`
stylegan_human/dnnlib/tflib/ops/upfirdn_2d.py:21
↓ 1 callersMethodvariables
(self)
stylegan_human/dnnlib/tflib/optimizer.py:303
FunctionIR_101
Constructs a ir-101 model.
stylegan_human/pti/pti_models/e4e/encoders/model_irse.py:57
FunctionIR_152
Constructs a ir-152 model.
stylegan_human/pti/pti_models/e4e/encoders/model_irse.py:63
FunctionIR_50
Constructs a ir-50 model.
stylegan_human/pti/pti_models/e4e/encoders/model_irse.py:51
FunctionIR_SE_101
Constructs a ir_se-101 model.
stylegan_human/pti/pti_models/e4e/encoders/model_irse.py:75
FunctionIR_SE_152
Constructs a ir_se-152 model.
stylegan_human/pti/pti_models/e4e/encoders/model_irse.py:81
FunctionIR_SE_50
Constructs a ir_se-50 model.
stylegan_human/pti/pti_models/e4e/encoders/model_irse.py:69
Method__call__
(self, oriImg)
stylegan_human/openpose/src/body.py:23
Method__call__
(self, show=True)
viz/latent_widget.py:31
Method__call__
(self, show=True)
viz/drag_widget.py:91
Method__call__
(self, show=True)
viz/pickle_widget.py:72
Method__call__
(self, show=True)
viz/capture_widget.py:53
Method__del__
(self)
stylegan_human/training/dataset.py:78
Method__del__
(self)
stylegan_human/training_scripts/sg2/training/dataset.py:84
Method__del__
(self)
stylegan_human/training_scripts/sg3/training/dataset.py:87
Method__del__
(self)
training/dataset.py:78
Method__del__
(self)
gui_utils/glfw_window.py:53
Method__del__
(self)
gui_utils/gl_utils.py:167
Method__del__
(self)
gui_utils/gl_utils.py:284
Method__delattr__
(self, name: str)
stylegan_human/dnnlib/util.py:53
Method__delattr__
(self, name: str)
dnnlib/util.py:52
Method__enter__
(self)
stylegan_human/torch_utils/misc.py:72
Method__enter__
(self)
dnnlib/util.py:72
Method__exit__
(self, exc_type: Any, exc_value: Any, traceback: Any)
stylegan_human/dnnlib/util.py:76
Method__exit__
(self, exc_type: Any, exc_value: Any, traceback: Any)
dnnlib/util.py:75
Method__getattr__
(self, name: str)
stylegan_human/dnnlib/util.py:44
Method__getattr__
(self, name: str)
dnnlib/util.py:43
Method__getitem__
r"""Convenience getter. `collector[name]` is a synonym for `collector.mean(name)`.
torch_utils/training_stats.py:226
Method__getitem__
(self, index)
stylegan_human/utils/ImagesDataset.py:20
Method__getitem__
r"""Convenience getter. `collector[name]` is a synonym for `collector.mean(name)`.
stylegan_human/torch_utils/training_stats.py:228
Method__getitem__
(self, idx)
stylegan_human/training/dataset.py:87
Method__getitem__
(self, idx)
stylegan_human/training_scripts/sg2/training/dataset.py:93
Method__getitem__
(self, idx)
stylegan_human/training_scripts/sg3/training/dataset.py:96
Method__getitem__
(self, idx)
training/dataset.py:87
Method__getstate__
(self)
stylegan_human/training/dataset.py:209
Method__getstate__
Pickle export.
stylegan_human/dnnlib/tflib/network.py:257
Method__getstate__
(self)
stylegan_human/training_scripts/sg2/training/dataset.py:227
← previousnext →401–500 of 1,199, ranked by callers