Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Picsart-AI-Research/MI-GAN
/ functions
Functions
526 in github.com/Picsart-AI-Research/MI-GAN
⨍
Functions
526
◇
Types & classes
121
↓ 1 callers
Function
_init
()
torch_utils/ops/bias_act.py:41
↓ 1 callers
Function
_init
()
torch_utils/ops/upfirdn2d.py:26
↓ 1 callers
Function
_module_to_src
r"""Query the source code of a given Python module.
torch_utils/persistence.py:206
↓ 1 callers
Function
_should_use_custom_op
()
torch_utils/ops/grid_sample_gradfix.py:34
↓ 1 callers
Function
_src_to_module
r"""Get or create a Python module for the given source code.
torch_utils/persistence.py:216
↓ 1 callers
Function
_sync
r"""Synchronize the global cumulative counters across devices and processes. Called internally by `Collector.update()`.
torch_utils/training_stats.py:234
↓ 1 callers
Function
_upfirdn2d_ref
Slow reference implementation of `upfirdn2d()` using standard PyTorch ops.
torch_utils/ops/upfirdn2d.py:169
↓ 1 callers
Method
append_cuda
(self, x)
lib/evaluator/stylegan_metrics/metric_utils.py:96
↓ 1 callers
Function
call_func_by_name
Finds the python object with the given name and calls it as a function.
dnnlib/util.py:279
↓ 1 callers
Function
cfg_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 callers
Function
common_init
(m)
lib/model_zoo/common/utils.py:144
↓ 1 callers
Function
compute_ssim
(img1, img2, window_size = 11, size_average = True)
lib/evaluator/eva_ssim.py:68
↓ 1 callers
Function
copy_weights
(source, dest, resolution=256)
scripts/export_inference_model.py:17
↓ 1 callers
Function
edict_2_dict
(x)
lib/cfg_helper.py:591
↓ 1 callers
Function
gaussian
(window_size, sigma)
lib/evaluator/eva_ssim.py:11
↓ 1 callers
Method
get_all
(self)
lib/evaluator/stylegan_metrics/metric_utils.py:125
↓ 1 callers
Function
get_args
()
scripts/generate_masks.py:96
↓ 1 callers
Function
get_args
()
scripts/create_onnx_pipeline.py:266
↓ 1 callers
Function
get_args
()
scripts/export_inference_model.py:88
↓ 1 callers
Function
get_args
()
scripts/demo.py:69
↓ 1 callers
Function
get_command_line_args
()
lib/cfg_helper.py:272
↓ 1 callers
Method
get_ddp_modules
(self, RANK, G, D, **kwargs)
lib/experiments/comodgan_default.py:164
↓ 1 callers
Method
get_ddp_modules
(self, RANK, G, D, **kwargs)
lib/experiments/comodgan_default.py:784
↓ 1 callers
Method
get_ddp_modules
(self, RANK, G, D, **kwargs)
lib/experiments/migan_default.py:157
↓ 1 callers
Function
get_experiment_id
()
lib/cfg_helper.py:236
↓ 1 callers
Function
get_feature_detector_name
(url)
lib/evaluator/stylegan_metrics/metric_utils.py:37
↓ 1 callers
Method
get_masked_bbox
(self, mask)
scripts/create_onnx_pipeline.py:132
↓ 1 callers
Function
get_obj_by_name
Finds the python object with the given name.
dnnlib/util.py:273
↓ 1 callers
Method
get_sync_order
(self)
lib/data_factory/common/ds_sampler.py:77
↓ 1 callers
Function
get_total_param
(net)
lib/model_zoo/common/utils.py:176
↓ 1 callers
Function
get_total_param_sum
(net)
lib/model_zoo/common/utils.py:180
↓ 1 callers
Method
get_version
(self, name)
lib/model_zoo/common/get_model.py:95
↓ 1 callers
Method
get_yaml_path
(self, name)
lib/cfg_helper.py:146
↓ 1 callers
Method
get_yaml_path
(self, name)
lib/cfg_helper.py:194
↓ 1 callers
Method
get_yaml_path
(self, name)
lib/cfg_helper.py:227
↓ 1 callers
Method
init_load_info
(self, cfg)
lib/data_factory/common/ds_base.py:31
↓ 1 callers
Method
is_better
(self, stat)
lib/experiments/comodgan_default.py:146
↓ 1 callers
Method
is_better
(self, stat)
lib/experiments/migan_default.py:139
↓ 1 callers
Function
is_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 callers
Function
is_url
Determine whether the given object is a valid URL string.
dnnlib/util.py:364
↓ 1 callers
Function
legacy_load_D
(D, D_tfmodel)
lib/experiments/comodgan_default.py:678
↓ 1 callers
Method
load_network_pkl
(self, f)
lib/experiments/comodgan_default.py:730
↓ 1 callers
Function
load_state_dict
(net, model_path)
lib/model_zoo/common/get_model.py:9
↓ 1 callers
Method
load_teacher1_model
(self, path, device, resolution=256, load_D=False)
lib/experiments/loss.py:55
↓ 1 callers
Function
main
()
scripts/generate_masks.py:104
↓ 1 callers
Function
main
()
scripts/create_onnx_pipeline.py:278
↓ 1 callers
Function
main
()
scripts/export_inference_model.py:99
↓ 1 callers
Function
main
()
scripts/evaluate_fid_lpips.py:177
↓ 1 callers
Function
main
()
scripts/calculate_flops.py:112
↓ 1 callers
Function
main
()
scripts/demo.py:81
↓ 1 callers
Method
main
(self, batch, net, RANK, **kwargs)
lib/experiments/comodgan_default.py:189
↓ 1 callers
Method
main
(self, batch, net, RANK, **kwargs)
lib/experiments/migan_default.py:177
↓ 1 callers
Function
make_cache_dir_path
(*paths: str)
dnnlib/util.py:124
↓ 1 callers
Function
mapping_D
(k)
lib/experiments/comodgan_default.py:681
↓ 1 callers
Function
mapping_G
(k)
lib/experiments/comodgan_default.py:625
↓ 1 callers
Method
num
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 callers
Function
persistent_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 callers
Method
postprocess
(self, image, mask, model_output)
scripts/create_onnx_pipeline.py:241
↓ 1 callers
Function
preprocess
(img: Image, mask: Image, resolution: int)
scripts/demo.py:56
↓ 1 callers
Method
preprocess
(self, image, mask)
scripts/create_onnx_pipeline.py:233
↓ 1 callers
Function
preprocess_model_args
(args)
lib/model_zoo/common/get_model.py:41
↓ 1 callers
Function
read_mask
(mask_path, invert=False)
scripts/create_onnx_pipeline.py:27
↓ 1 callers
Function
read_mask
(mask_path, invert=False)
scripts/demo.py:26
↓ 1 callers
Method
register
(self, evaf, name)
lib/evaluator/eva_base.py:26
↓ 1 callers
Method
register
(self, model, name, version='x')
lib/model_zoo/common/get_model.py:62
↓ 1 callers
Method
register
(self, ds)
lib/data_factory/common/ds_base.py:66
↓ 1 callers
Method
register
(self, formatf)
lib/data_factory/common/ds_formatter.py:16
↓ 1 callers
Method
register
(self, loadf)
lib/data_factory/common/ds_loader.py:22
↓ 1 callers
Function
report
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 callers
Method
resume
(self, resume_pkl, models)
lib/experiments/comodgan_default.py:155
↓ 1 callers
Method
resume
(self, resume_pkl, models)
lib/experiments/comodgan_default.py:767
↓ 1 callers
Method
resume
(self, resume_pkl, models)
lib/experiments/migan_default.py:148
↓ 1 callers
Function
run_generator
(x)
lib/evaluator/stylegan_metrics/metric_utils.py:261
↓ 1 callers
Method
run_teacher1_G
(self, x, sync)
lib/experiments/loss.py:131
↓ 1 callers
Method
save
(self, pkl_file)
lib/evaluator/stylegan_metrics/metric_utils.py:139
↓ 1 callers
Method
save_cfg
(self, cfg)
lib/cfg_holder.py:25
↓ 1 callers
Function
search_experiment_folder
(root, exid)
lib/cfg_helper.py:606
↓ 1 callers
Method
std
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 callers
Method
sync_
(self, data, rank)
lib/evaluator/eva_base.py:112
Function
PYBIND11_MODULE
torch_utils/ops/bias_act.cpp:94
Function
PYBIND11_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
← previous
next →
101–200 of 526, ranked by callers