MCPcopy Create free account

hub / github.com/Picsart-AI-Research/MI-GAN / functions

Functions526 in github.com/Picsart-AI-Research/MI-GAN

↓ 1 callersFunction_init
()
torch_utils/ops/bias_act.py:41
↓ 1 callersFunction_init
()
torch_utils/ops/upfirdn2d.py:26
↓ 1 callersFunction_module_to_src
r"""Query the source code of a given Python module.
torch_utils/persistence.py:206
↓ 1 callersFunction_should_use_custom_op
()
torch_utils/ops/grid_sample_gradfix.py:34
↓ 1 callersFunction_src_to_module
r"""Get or create a Python module for the given source code.
torch_utils/persistence.py:216
↓ 1 callersFunction_sync
r"""Synchronize the global cumulative counters across devices and processes. Called internally by `Collector.update()`.
torch_utils/training_stats.py:234
↓ 1 callersFunction_upfirdn2d_ref
Slow reference implementation of `upfirdn2d()` using standard PyTorch ops.
torch_utils/ops/upfirdn2d.py:169
↓ 1 callersMethodappend_cuda
(self, x)
lib/evaluator/stylegan_metrics/metric_utils.py:96
↓ 1 callersFunctioncall_func_by_name
Finds the python object with the given name and calls it as a function.
dnnlib/util.py:279
↓ 1 callersFunctioncfg_initiates
Step1.1: find the GPU device and count Step1.2: set batchsize and batchsize per gpu Step1.3: set dataloader worker and worker per gpu
lib/cfg_helper.py:386
↓ 1 callersFunctioncommon_init
(m)
lib/model_zoo/common/utils.py:144
↓ 1 callersFunctioncompute_ssim
(img1, img2, window_size = 11, size_average = True)
lib/evaluator/eva_ssim.py:68
↓ 1 callersFunctioncopy_weights
(source, dest, resolution=256)
scripts/export_inference_model.py:17
↓ 1 callersFunctionedict_2_dict
(x)
lib/cfg_helper.py:591
↓ 1 callersFunctiongaussian
(window_size, sigma)
lib/evaluator/eva_ssim.py:11
↓ 1 callersMethodget_all
(self)
lib/evaluator/stylegan_metrics/metric_utils.py:125
↓ 1 callersFunctionget_args
()
scripts/generate_masks.py:96
↓ 1 callersFunctionget_args
()
scripts/create_onnx_pipeline.py:266
↓ 1 callersFunctionget_args
()
scripts/export_inference_model.py:88
↓ 1 callersFunctionget_args
()
scripts/demo.py:69
↓ 1 callersFunctionget_command_line_args
()
lib/cfg_helper.py:272
↓ 1 callersMethodget_ddp_modules
(self, RANK, G, D, **kwargs)
lib/experiments/comodgan_default.py:164
↓ 1 callersMethodget_ddp_modules
(self, RANK, G, D, **kwargs)
lib/experiments/comodgan_default.py:784
↓ 1 callersMethodget_ddp_modules
(self, RANK, G, D, **kwargs)
lib/experiments/migan_default.py:157
↓ 1 callersFunctionget_experiment_id
()
lib/cfg_helper.py:236
↓ 1 callersFunctionget_feature_detector_name
(url)
lib/evaluator/stylegan_metrics/metric_utils.py:37
↓ 1 callersMethodget_masked_bbox
(self, mask)
scripts/create_onnx_pipeline.py:132
↓ 1 callersFunctionget_obj_by_name
Finds the python object with the given name.
dnnlib/util.py:273
↓ 1 callersMethodget_sync_order
(self)
lib/data_factory/common/ds_sampler.py:77
↓ 1 callersFunctionget_total_param
(net)
lib/model_zoo/common/utils.py:176
↓ 1 callersFunctionget_total_param_sum
(net)
lib/model_zoo/common/utils.py:180
↓ 1 callersMethodget_version
(self, name)
lib/model_zoo/common/get_model.py:95
↓ 1 callersMethodget_yaml_path
(self, name)
lib/cfg_helper.py:146
↓ 1 callersMethodget_yaml_path
(self, name)
lib/cfg_helper.py:194
↓ 1 callersMethodget_yaml_path
(self, name)
lib/cfg_helper.py:227
↓ 1 callersMethodinit_load_info
(self, cfg)
lib/data_factory/common/ds_base.py:31
↓ 1 callersMethodis_better
(self, stat)
lib/experiments/comodgan_default.py:146
↓ 1 callersMethodis_better
(self, stat)
lib/experiments/migan_default.py:139
↓ 1 callersFunctionis_top_level_function
Determine whether the given object is a top-level function, i.e., defined at module scope using 'def'.
dnnlib/util.py:298
↓ 1 callersFunctionis_url
Determine whether the given object is a valid URL string.
dnnlib/util.py:364
↓ 1 callersFunctionlegacy_load_D
(D, D_tfmodel)
lib/experiments/comodgan_default.py:678
↓ 1 callersMethodload_network_pkl
(self, f)
lib/experiments/comodgan_default.py:730
↓ 1 callersFunctionload_state_dict
(net, model_path)
lib/model_zoo/common/get_model.py:9
↓ 1 callersMethodload_teacher1_model
(self, path, device, resolution=256, load_D=False)
lib/experiments/loss.py:55
↓ 1 callersFunctionmain
()
scripts/generate_masks.py:104
↓ 1 callersFunctionmain
()
scripts/create_onnx_pipeline.py:278
↓ 1 callersFunctionmain
()
scripts/export_inference_model.py:99
↓ 1 callersFunctionmain
()
scripts/evaluate_fid_lpips.py:177
↓ 1 callersFunctionmain
()
scripts/calculate_flops.py:112
↓ 1 callersFunctionmain
()
scripts/demo.py:81
↓ 1 callersMethodmain
(self, batch, net, RANK, **kwargs)
lib/experiments/comodgan_default.py:189
↓ 1 callersMethodmain
(self, batch, net, RANK, **kwargs)
lib/experiments/migan_default.py:177
↓ 1 callersFunctionmake_cache_dir_path
(*paths: str)
dnnlib/util.py:124
↓ 1 callersFunctionmapping_D
(k)
lib/experiments/comodgan_default.py:681
↓ 1 callersFunctionmapping_G
(k)
lib/experiments/comodgan_default.py:625
↓ 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 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 callersMethodpostprocess
(self, image, mask, model_output)
scripts/create_onnx_pipeline.py:241
↓ 1 callersFunctionpreprocess
(img: Image, mask: Image, resolution: int)
scripts/demo.py:56
↓ 1 callersMethodpreprocess
(self, image, mask)
scripts/create_onnx_pipeline.py:233
↓ 1 callersFunctionpreprocess_model_args
(args)
lib/model_zoo/common/get_model.py:41
↓ 1 callersFunctionread_mask
(mask_path, invert=False)
scripts/create_onnx_pipeline.py:27
↓ 1 callersFunctionread_mask
(mask_path, invert=False)
scripts/demo.py:26
↓ 1 callersMethodregister
(self, evaf, name)
lib/evaluator/eva_base.py:26
↓ 1 callersMethodregister
(self, model, name, version='x')
lib/model_zoo/common/get_model.py:62
↓ 1 callersMethodregister
(self, ds)
lib/data_factory/common/ds_base.py:66
↓ 1 callersMethodregister
(self, formatf)
lib/data_factory/common/ds_formatter.py:16
↓ 1 callersMethodregister
(self, loadf)
lib/data_factory/common/ds_loader.py:22
↓ 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 callersMethodresume
(self, resume_pkl, models)
lib/experiments/comodgan_default.py:155
↓ 1 callersMethodresume
(self, resume_pkl, models)
lib/experiments/comodgan_default.py:767
↓ 1 callersMethodresume
(self, resume_pkl, models)
lib/experiments/migan_default.py:148
↓ 1 callersFunctionrun_generator
(x)
lib/evaluator/stylegan_metrics/metric_utils.py:261
↓ 1 callersMethodrun_teacher1_G
(self, x, sync)
lib/experiments/loss.py:131
↓ 1 callersMethodsave
(self, pkl_file)
lib/evaluator/stylegan_metrics/metric_utils.py:139
↓ 1 callersMethodsave_cfg
(self, cfg)
lib/cfg_holder.py:25
↓ 1 callersFunctionsearch_experiment_folder
(root, exid)
lib/cfg_helper.py:606
↓ 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 callersMethodsync_
(self, data, rank)
lib/evaluator/eva_base.py:112
FunctionPYBIND11_MODULE
torch_utils/ops/bias_act.cpp:94
FunctionPYBIND11_MODULE
torch_utils/ops/upfirdn2d.cpp:98
Method__call__
(self, name)
lib/cfg_helper.py:109
Method__call__
(self, name)
lib/cfg_helper.py:162
Method__call__
(self, name)
lib/cfg_helper.py:213
Method__call__
Args: RANK: int, the rank of the stage process. If not multi process, please set 0.
lib/utils.py:29
Method__call__
(self, **kwargs)
lib/experiments/comodgan_default.py:51
Method__call__
(self, RANK, **paras)
lib/experiments/comodgan_default.py:253
Method__call__
(self, RANK, **paras)
lib/experiments/comodgan_default.py:805
Method__call__
(self, **kwargs)
lib/experiments/migan_default.py:71
Method__call__
(self, RANK, **paras)
lib/experiments/migan_default.py:227
Method__call__
(self, pipeline_cfg=None)
lib/evaluator/eva_base.py:29
Method__call__
(self)
lib/evaluator/stylegan_metrics/frechet_inception_distance.py:34
Method__call__
(self, x, gain=1)
lib/model_zoo/migan_inference.py:20
Method__call__
Construct model based on the config.
lib/model_zoo/common/get_model.py:66
Method__call__
(self, name)
lib/model_zoo/common/utils.py:62
Method__call__
(self, x, gain=1)
lib/model_zoo/common/utils.py:114
Method__call__
(self, path, element)
lib/data_factory/ds_ffhq.py:76
Method__call__
(self, path, element)
lib/data_factory/ds_ffhq.py:91
Method__call__
(self, element)
lib/data_factory/ds_ffhq.py:111
Method__call__
(self, element)
lib/data_factory/ds_ffhq.py:134
← previousnext →101–200 of 526, ranked by callers