MCPcopy Create free account

hub / github.com/10Ring/LAA-Net / functions

Functions375 in github.com/10Ring/LAA-Net

↓ 1 callersMethod_decode_block_string
Get a block through a string notation of arguments. Args: block_string (str): A string notation of arguments.
models/networks/efficientNet.py:298
↓ 1 callersFunction_distance_hm_cls_loss
(cos_sim_ins, hm_preds, hm_gts, label_preds, label_gts, alpha=0.25)
losses/losses.py:112
↓ 1 callersMethod_encode_block_string
Encode a block to a string. Args: block (namedtuple): A BlockArgs type argument. Returns: block_string: A Stri
models/networks/efficientNet.py:331
↓ 1 callersMethod_erase
(self, img: np.array, tops: list, lefts: list, hs: list, ws: list, img_chan: int, **params)
datasets/pipelines/geo_transform.py:182
↓ 1 callersMethod_facial_landmark
(self, image, detector, lm_predictor)
package_utils/geo_landmarks_extraction.py:123
↓ 1 callersMethod_filter_data
(self)
package_utils/bi_online_generation.py:224
↓ 1 callersMethod_gauss_noise
(self, var_limit=(10.0, 50.0), mean=0, per_ch
datasets/pipelines/color_transform.py:77
↓ 1 callersMethod_gaussianblur
(self, blur_limit=(3, 7), sigma_limit=0,
datasets/pipelines/color_transform.py:67
↓ 1 callersMethod_gen_BI
(self, background_face, background_landmark, foreground_face_path, idx=None)
datasets/face_forensic_hm.py:89
↓ 1 callersMethod_get_deconv_cfg
(self, deconv_kernel, index)
models/networks/mrsa_resnet.py:270
↓ 1 callersFunction_get_pixels
(per_pixel, rand_color, patch_size, dtype=np.float32)
datasets/pipelines/functional.py:5
↓ 1 callersMethod_grad_norm
(self)
lib/optimizers/sam.py:66
↓ 1 callersMethod_horizontal_flip
(self, p=0.5, always_apply=False)
datasets/pipelines/geo_transform.py:51
↓ 1 callersMethod_huesaturation
(self, hue_shift_limit=20, sat_shift_limit=20,
datasets/pipelines/color_transform.py:141
↓ 1 callersMethod_img_obj
(self, img_path, file_name, **kwargs)
package_utils/geo_landmarks_extraction.py:115
↓ 1 callersMethod_jpegcompression
(self, quality_lower=70, quality_upper=100,
datasets/pipelines/color_transform.py:89
↓ 1 callersMethod_load_data
(self)
package_utils/geo_landmarks_extraction.py:57
↓ 1 callersMethod_load_data
(self, split)
datasets/face_forensic_binary.py:41
↓ 1 callersMethod_load_data_from_file
Each extension will be treated with particular extension loader
package_utils/geo_landmarks_extraction.py:99
↓ 1 callersMethod_load_data_from_path
Currenly, Using Glob for loading file with regex It might be changed for better performance in large datasets
package_utils/geo_landmarks_extraction.py:72
↓ 1 callersMethod_load_image
(self, img_path)
package_utils/geo_landmarks_extraction.py:119
↓ 1 callersMethod_load_img
(self, img_path)
datasets/face_forensic_binary.py:53
↓ 1 callersMethod_make_branches
(self, num_branches, block, num_blocks, num_channels)
models/networks/pose_hrnet.py:166
↓ 1 callersMethod_make_deconv_layer
(self, num_layers, num_filters, num_kernels)
models/networks/mrsa_resnet.py:283
↓ 1 callersMethod_make_deconv_layer
(self, num_layers, num_filters, num_kernels)
models/networks/pose_efficientNet.py:371
↓ 1 callersMethod_make_fuse_layers
(self)
models/networks/pose_hrnet.py:176
↓ 1 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
models/networks/pose_hrnet.py:375
↓ 1 callersMethod_make_one_branch
(self, branch_index, block, num_blocks, num_channels, stride=1)
models/networks/pose_hrnet.py:128
↓ 1 callersMethod_normalize
(self, p=0.0)
datasets/pipelines/geo_transform.py:48
↓ 1 callersMethod_offset_loss
(self, preds, gts, apply_filter=False)
losses/losses.py:153
↓ 1 callersMethod_random_crop
(self, p=0.5, always_apply=False, crop_limit=0.
datasets/pipelines/geo_transform.py:67
↓ 1 callersMethod_random_erasing
(self, p=0.5, always_apply=False, max
datasets/pipelines/geo_transform.py:80
↓ 1 callersMethod_random_scale
(self, p=0.5, always_apply=False, scale_limi
datasets/pipelines/geo_transform.py:57
↓ 1 callersMethod_randombrightness
(self, brightness_limit=0.1, contrast_limit=0.1,
datasets/pipelines/color_transform.py:119
↓ 1 callersMethod_randomcontrast
(self, limit=0.2, always_apply=False,
datasets/pipelines/color_transform.py:111
↓ 1 callersMethod_randomgamma
(self, gamma_limit=(80, 120), eps=None, alwa
datasets/pipelines/color_transform.py:131
↓ 1 callersMethod_reload_data
(self)
datasets/face_forensic_sbi.py:65
↓ 1 callersMethod_resize
(self)
datasets/pipelines/geo_transform.py:43
↓ 1 callersMethod_rgbshift
(self, r_shift_limit=20, g_shift_limit=20, b_shift_lim
datasets/pipelines/color_transform.py:99
↓ 1 callersFunctionaffine_transform
This function apply the affine transform to each point given by an affine matrix
package_utils/transform.py:63
↓ 1 callersFunctionargs_parse
(args=None)
package_utils/bi_online_generation.py:40
↓ 1 callersFunctionargs_parser
(args=None)
scripts/train.py:30
↓ 1 callersMethodbackward
(ctx, grad_output)
models/networks/efficientNet.py:388
↓ 1 callersFunctionbuild_losses
(cfg, loss_func: Registry, build_func=build_from_cfg, defau
losses/builder.py:11
↓ 1 callersMethodbuild_mask
Override to build the mask
package_utils/deepfake_mask.py:51
↓ 1 callersMethodclear
(self)
package_utils/bi_online_generation.py:210
↓ 1 callersMethodclear_sequences
(self)
package_utils/bi_online_generation.py:214
↓ 1 callersMethodcount
(self)
package_utils/bi_online_generation.py:198
↓ 1 callersFunctioncrop_by_margin
(image, margin=[0, 0])
package_utils/image_utils.py:13
↓ 1 callersMethoddecode
Decode a list of string notations to specify blocks inside the network. Args: string_list (list[str]): A list of strings, each str
models/networks/efficientNet.py:353
↓ 1 callersMethoddeprecated_register_module
(self, cls=None, force=False)
register/register.py:259
↓ 1 callersFunctiondrop_connect
Drop connect. Args: input (tensor: BCWH): Input of this structure. p (float: 0.0~1.0): Probability of drop connection. tra
models/networks/efficientNet.py:83
↓ 1 callersFunctiondynamic_blend
(source,target,mask)
package_utils/deepfake_mask.py:193
↓ 1 callersFunctionefficientnet
Create BlockArgs and GlobalParams for efficientnet model. Args: width_coefficient (float) depth_coefficient (float) image_
models/networks/efficientNet.py:231
↓ 1 callersMethodextract_endpoints
Use convolution layer to extract features from reduction levels i in [1, 2, 3, 4, 5]. Args: inputs (tensor): Input tensor.
models/networks/pose_efficientNet.py:409
↓ 1 callersFunctionf_cstency
(cstency_hm_preds, cstency_hm_gt)
losses/losses.py:28
↓ 1 callersMethodfacial_landmarks
(self, img_paths, detector, lm_predictor)
package_utils/geo_landmarks_extraction.py:158
↓ 1 callersFunctionfreeze_backbone
This func to freeze some specific layers to warm up the models
models/utils.py:53
↓ 1 callersMethodfrom_name
Create an efficientnet model according to name. Args: model_name (str): Name for efficientnet. in_channels (int): Inpu
models/networks/pose_efficientNet.py:610
↓ 1 callersFunctiongen_SBI
(img, landmark)
datasets/sbi/utils.py:67
↓ 1 callersMethodgen_one_datapoint
(self, idx)
package_utils/bi_online_generation.py:231
↓ 1 callersFunctionget_available_masks
Return a list of the available masks for cli
package_utils/deepfake_mask.py:14
↓ 1 callersFunctionget_blend_mask
(mask)
package_utils/deepfake_mask.py:202
↓ 1 callersMethodget_blended_face
(self, background_face_path, background_landmark)
package_utils/bi_online_generation.py:261
↓ 1 callersFunctionget_blocks
(num_layers)
models/networks/arcface.py:105
↓ 1 callersFunctionget_dir
(src_point, rot_rad)
package_utils/transform.py:10
↓ 1 callersMethodget_num_inchannels
(self)
models/networks/pose_hrnet.py:233
↓ 1 callersFunctionget_transforms
()
datasets/sbi/utils.py:56
↓ 1 callersFunctionget_width_and_height_from_size
Obtain height and width from x. Args: x (int, tuple or list): Data size. Returns: size: A tuple or list (H,W).
models/networks/efficientNet.py:394
↓ 1 callersMethodinfer_scope
Infer the scope of registry. The name of the package where registry is defined will be returned. Example: >>> # in mmdet/m
register/register.py:141
↓ 1 callersMethodinit_weights
(self, pretrained=False)
models/networks/xception.py:302
↓ 1 callersMethodinit_weights
(self, pretrained='', **kwargs)
models/networks/pose_hrnet.py:470
↓ 1 callersMethodinit_weights
(self, pretrained=False, advprop=False, verbose=True)
models/networks/pose_efficientNet.py:721
↓ 1 callersFunctionload_file
(file_path)
package_utils/utils.py:141
↓ 1 callersFunctionload_model
(model, model_path, optimizer=None, resume=False, lr=None, lr_step=None, gamma=None)
models/utils.py:76
↓ 1 callersFunctionload_pretrained_weights
Loads pretrained weights from weights path or download using url. Args: model (Module): The whole model of efficientnet. model_nam
models/networks/efficientNet.py:457
↓ 1 callersFunctionmain
()
package_utils/cam_vis.py:25
↓ 1 callersFunctionmake_dir
(dir_path)
package_utils/utils.py:16
↓ 1 callersMethodmerge_mask
Return the mask in requested shape
package_utils/deepfake_mask.py:55
↓ 1 callersFunctionparse_args
(args=None)
scripts/test.py:34
↓ 1 callersFunctionpreset_model
(cfg, model, optimizer=None)
models/utils.py:19
↓ 1 callersFunctionrandaffine
(img, mask)
package_utils/transform.py:113
↓ 1 callersMethodrandomdownscale
(self, img)
datasets/sbi/utils.py:27
↓ 1 callersMethodregister_task
(self, target)
package_utils/bi_online_generation.py:181
↓ 1 callersFunctionreorder_landmark
(landmark)
datasets/sbi/utils.py:119
↓ 1 callersMethodreset
(self)
lib/core_function.py:19
↓ 1 callersFunctionround_repeats
Calculate module's repeat number of a block based on depth multiplier. Use depth_coefficient of global_params. Args: repeats (int):
models/networks/efficientNet.py:67
↓ 1 callersMethodrunning
(self)
package_utils/bi_online_generation.py:204
↓ 1 callersMethodsave2json
(self, data, fn='faceforensics_processed.json')
package_utils/geo_landmarks_extraction.py:224
↓ 1 callersFunctionsave_model
(path, epoch, model, optimizer=None)
models/utils.py:129
↓ 1 callersFunctionsbi_hflip
(img, mask=None, landmark=None, bbox=None)
datasets/sbi/utils.py:127
↓ 1 callersMethodset_swish
Sets swish function as memory efficient (for training) or standard (for export). Args: memory_efficient (bool): Whether to use mem
models/networks/pose_efficientNet.py:140
↓ 1 callersMethodsplit_scope_key
Split scope and key. The first scope will be split from key. Examples: >>> Registry.split_scope_key('mmdet.ResNet')
register/register.py:164
↓ 1 callersMethodstart
(self, p)
package_utils/bi_online_generation.py:188
↓ 1 callersMethodstop
(self)
package_utils/bi_online_generation.py:201
↓ 1 callersMethodterminate
(self)
package_utils/bi_online_generation.py:207
↓ 1 callersFunctiontotal_euclidean_distance
(a,b)
package_utils/bi_online_generation.py:67
↓ 1 callersFunctiontrain
(cfg, model, critetion, optimizer, epoch, data_loader, logger, writer, devices, trainIters, metrics_base='comb
lib/core_function.py:49
↓ 1 callersFunctionunfreeze_backbone
This func to unfreeze all model layers
models/utils.py:67
↓ 1 callersFunctionvalidate
(cfg, model, critetion, epoch, data_loader, logger, writer, devices, valIters, metrics_base='combine')
lib/core_function.py:172
← previousnext →101–200 of 375, ranked by callers