MCPcopy Index your code

hub / github.com/endo-yuki-t/Fewshot-SMIS / functions

Functions246 in github.com/endo-yuki-t/Fewshot-SMIS

↓ 15 callersMethod__init__
(self, kernel, pad, upsample_factor=1)
models/stylegan2/model.py:72
↓ 12 callersFunctionget_block
(in_channel, depth, num_units, stride=2)
models/encoders/helpers.py:25
↓ 7 callersFunctionnms
Non-maximum suppression. Arguments: boxes: a float numpy array of shape [n, 5], where each row is (xmin, ymin, xmax, ymax
models/mtcnn/mtcnn_pytorch/src/box_utils.py:5
↓ 6 callersMethod__init__
(self, *, image_size, patch_size, num_classes, dim, depth, heads, mlp_dim, channels = 3, attn_dropout = 0., ff
vit_pytorch/vit_pytorch.py:81
↓ 6 callersFunctioncalibrate_box
Transform bounding boxes to be more like true bounding boxes. 'offsets' is one of the outputs of the nets. Arguments: bboxes: a f
models/mtcnn/mtcnn_pytorch/src/box_utils.py:94
↓ 6 callersMethodcheckpoint_me
(self, loss_dict, is_best)
training/coach.py:257
↓ 5 callersMethod__init__
(self, opts)
configs/transforms_config.py:8
↓ 5 callersFunctiontformfwd
Function: ---------- apply affine transform 'trans' to uv Parameters: ---------- @trans: 3x3 np.array
models/mtcnn/mtcnn_pytorch/src/matlab_cp2tform.py:19
↓ 4 callersMethod__init__
(self)
criteria/lpips/networks.py:89
↓ 4 callersFunctionconvert_to_square
Convert bounding boxes to a square form. Arguments: bboxes: a float numpy array of shape [n, 5]. Returns: a float nump
models/mtcnn/mtcnn_pytorch/src/box_utils.py:71
↓ 4 callersFunctionget_blocks
(num_layers)
models/encoders/helpers.py:29
↓ 4 callersFunctionget_image_boxes
Cut out boxes from the image. Arguments: bounding_boxes: a float numpy array of shape [n, 5]. img: an instance of PIL.Image.
models/mtcnn/mtcnn_pytorch/src/box_utils.py:127
↓ 4 callersMethodparse
(self)
options/test_options.py:30
↓ 4 callersMethodparse_and_log_images
(self, id_logs, x, y, y_hat, title, subscript=None, display_count=2)
training/coach.py:357
↓ 4 callersFunctiontensor2im
(var)
utils/common.py:18
↓ 3 callersMethod__init__
(self, in_c, out_c, spatial)
models/encoders/psp_encoders.py:11
↓ 3 callersMethod__init__
(self)
models/mtcnn/mtcnn_pytorch/src/get_nets.py:34
↓ 3 callersMethod__load_latent_avg
(self, ckpt, repeat=None)
models/psp.py:295
↓ 3 callersMethodbackward
(ctx, grad_output)
models/stylegan2/op/upfirdn2d.py:124
↓ 3 callersMethodcalc_loss
(self, x, y, y_hat, latent)
training/coach.py:299
↓ 3 callersMethodextract_feats
(self, x)
criteria/id_loss.py:16
↓ 3 callersFunctionfindNonreflectiveSimilarity
(uv, xy, options=None)
models/mtcnn/mtcnn_pytorch/src/matlab_cp2tform.py:68
↓ 3 callersMethodforward
(self, x, resize=True, latent_mask=None, input_code=False, randomize_noise=True, inject_latent=N
models/psp.py:74
↓ 3 callersMethodget_transforms
(self)
configs/transforms_config.py:12
↓ 3 callersMethodlog_metrics
(self, metrics_dict, prefix)
training/coach.py:348
↓ 3 callersFunctionmake_kernel
(k)
models/stylegan2/model.py:18
↓ 3 callersMethodprint_metrics
(self, metrics_dict, prefix)
training/coach.py:352
↓ 3 callersMethodset_requires_grad
(self, state: bool)
criteria/lpips/networks.py:46
↓ 3 callersMethodstep
(self, closure=None)
training/ranger.py:78
↓ 3 callersMethodtrain
(self)
training/coach.py:73
↓ 3 callersFunctionupfirdn2d
(input, kernel, up=1, down=1, pad=(0, 0))
models/stylegan2/op/upfirdn2d.py:142
↓ 2 callersMethod__init__
(self, channels, reduction)
models/encoders/helpers.py:57
↓ 2 callersFunction_preprocess
Preprocessing step before feeding the network. Arguments: img: a float numpy array of shape [h, w, c]. Returns: a floa
models/mtcnn/mtcnn_pytorch/src/box_utils.py:226
↓ 2 callersMethod_upsample_add
Upsample and add two feature maps. Args: x: (Variable) top feature map to be upsampled. y: (Variable) lateral feature m
models/encoders/psp_encoders.py:72
↓ 2 callersMethodalign
(self, img)
models/mtcnn/mtcnn.py:23
↓ 2 callersMethoddetect_faces
Arguments: image: an instance of PIL.Image. min_face_size: a float number. thresholds: a list of leng
models/mtcnn/mtcnn.py:45
↓ 2 callersFunctionfused_leaky_relu
(input, bias, negative_slope=0.2, scale=2 ** 0.5)
models/stylegan2/op/fused_act.py:84
↓ 2 callersFunctionget_keys
(d, name)
models/psp.py:14
↓ 2 callersFunctionget_reference_facial_points
Function: ---------- get reference 5 key points according to crop settings: 0. Set default crop_size: if def
models/mtcnn/mtcnn_pytorch/src/align_trans.py:32
↓ 2 callersFunctionget_similarity_transform
Function: ---------- Find Similarity Transform Matrix 'trans': u = src_pts[:, 0] v = src_pts[:, 1]
models/mtcnn/mtcnn_pytorch/src/matlab_cp2tform.py:159
↓ 2 callersFunctionlog_input_image
(x, opts)
utils/common.py:9
↓ 2 callersFunctionrun_first_stage
Run P-Net, generate bounding boxes, and do NMS. Arguments: image: an instance of PIL.Image. net: an instance of pytorch's nn.
models/mtcnn/mtcnn_pytorch/src/first_stage.py:12
↓ 2 callersMethodvalidate
(self)
training/coach.py:228
↓ 2 callersFunctionwarp_and_crop_face
Function: ---------- apply affine transform 'trans' to uv Parameters: ---------- @src_img: 3x3 np.array
models/mtcnn/mtcnn_pytorch/src/align_trans.py:210
↓ 1 callersFunctionIR_101
Constructs a ir-101 model.
models/encoders/model_irse.py:57
↓ 1 callersMethod__all_idx
(self, idx, axis)
datasets/augmentations.py:20
↓ 1 callersMethod__get_save_dict
(self)
training/coach.py:384
↓ 1 callersMethod__init__
(self, n_classes=None)
datasets/augmentations.py:10
↓ 1 callersFunction_generate_bboxes
Generate bounding boxes at places where there is probably a face. Arguments: probs: a float numpy array of shape [n, m].
models/mtcnn/mtcnn_pytorch/src/first_stage.py:51
↓ 1 callersFunctionalign_face
:param filepath: str :return: PIL Image
scripts/align_all_parallel.py:52
↓ 1 callersMethodbicubic_kernel
This equation is exactly copied from the website below: https://clouard.users.greyc.fr/Pantheon/experiments/rescaling/index-en.html#bicubic
datasets/augmentations.py:45
↓ 1 callersMethodcalc_latent_loss
(self, latent, w)
training/coach.py:332
↓ 1 callersFunctionchunks
Yield successive n-sized chunks from lst.
scripts/calc_id_loss_parallel.py:22
↓ 1 callersFunctionchunks
Yield successive n-sized chunks from lst.
scripts/align_all_parallel.py:141
↓ 1 callersMethodcompute_rep_vec
(self, x_list, codes_list, input_code=False, randomize_noise=True, return_latents=False)
models/psp.py:113
↓ 1 callersMethodcompute_rep_vecs
(self, x_list, codes_list, input_code=False, randomize_noise=True, return_latents=False)
models/psp.py:196
↓ 1 callersMethodconfigure_datasets
(self)
training/coach.py:278
↓ 1 callersMethodconfigure_optimizers
(self)
training/coach.py:268
↓ 1 callersFunctioncorrect_bboxes
Crop boxes that are too big and get coordinates with respect to cutouts. Arguments: bboxes: a float numpy array of shape [n, 5],
models/mtcnn/mtcnn_pytorch/src/box_utils.py:162
↓ 1 callersFunctioncvt_tform_mat_for_cv2
Function: ---------- Convert Transform Matrix 'trans' into 'cv2_trans' which could be directly used by cv2.warpAffine():
models/mtcnn/mtcnn_pytorch/src/matlab_cp2tform.py:199
↓ 1 callersMethoddense_pseudo_labeling
(self, x, resize=True, latent_mask=None, input_code=False, randomize_noise=True, inject_latent=N
models/psp.py:144
↓ 1 callersFunctionfindSimilarity
(uv, xy, options=None)
models/mtcnn/mtcnn_pytorch/src/matlab_cp2tform.py:119
↓ 1 callersFunctionget_affine_transform_matrix
Function: ---------- get affine transform matrix 'tfm' from src_pts to dst_pts Parameters: ---------- @src_pts:
models/mtcnn/mtcnn_pytorch/src/align_trans.py:163
↓ 1 callersFunctionget_colors
()
utils/common.py:45
↓ 1 callersFunctionget_landmark
get landmark with dlib :return: np.array shape=(68, 2)
scripts/align_all_parallel.py:32
↓ 1 callersFunctionget_network
(net_type: str)
criteria/lpips/networks.py:12
↓ 1 callersFunctionget_similarity_transform_for_cv2
Function: ---------- Find Similarity Transform Matrix 'cv2_trans' which could be directly used by cv2.warpAffine():
models/mtcnn/mtcnn_pytorch/src/matlab_cp2tform.py:227
↓ 1 callersFunctionget_state_dict
(net_type: str = 'alex', version: str = '0.1')
criteria/lpips/utils.py:11
↓ 1 callersMethodinitialize
(self)
options/train_options.py:11
↓ 1 callersMethodinitialize
(self)
options/test_options.py:10
↓ 1 callersMethodinvert_training_img
(self, overwrite)
training/coach.py:181
↓ 1 callersFunctionis_image_file
(filename)
utils/data_utils.py:13
↓ 1 callersFunctionl2_norm
(input, axis=1)
models/encoders/helpers.py:15
↓ 1 callersMethodload_weights
(self)
models/psp.py:51
↓ 1 callersMethodlog_images
(self, name, im_data, subscript=None, log_latest=False)
training/coach.py:371
↓ 1 callersFunctionmain
()
scripts/train.py:16
↓ 1 callersFunctionnormalize_activation
(x, eps=1e-10)
criteria/lpips/utils.py:6
↓ 1 callersMethodonehot_initialization
(self, a)
datasets/augmentations.py:13
↓ 1 callersFunctionparse_args
()
scripts/calc_id_loss_parallel.py:71
↓ 1 callersFunctionparse_args
()
scripts/align_all_parallel.py:167
↓ 1 callersFunctionparse_args
()
scripts/calc_losses_on_images.py:18
↓ 1 callersFunctionpositionalencoding2d
:param d_model: dimension of the model :param height: height of the positions :param width: width of the positions :return: d_model*h
vit_pytorch/positionalembedding2d.py:24
↓ 1 callersFunctionrun
()
scripts/generate_stylegan_samples.py:22
↓ 1 callersFunctionrun
(args)
scripts/calc_id_loss_parallel.py:80
↓ 1 callersFunctionrun
()
scripts/style_mixing.py:21
↓ 1 callersFunctionrun
(args)
scripts/align_all_parallel.py:175
↓ 1 callersFunctionrun
(args)
scripts/calc_losses_on_images.py:29
↓ 1 callersFunctionrun
()
scripts/inference.py:22
↓ 1 callersFunctionrun_on_batch
(inputs, net, opts)
scripts/generate_stylegan_samples.py:52
↓ 1 callersFunctionrun_on_batch
(inputs, net, opts)
scripts/inference.py:114
↓ 1 callersMethodset_encoder
(self)
models/psp.py:40
↓ 1 callersMethodset_opts
(self, opts)
models/psp.py:292
↓ 1 callersMethodsparse_pseudo_labeling
(self, x, resize=True, latent_mask=None, input_code=False, randomize_noise=True, inject_latent=N
models/psp.py:221
↓ 1 callersFunctiontensor2map
(var)
utils/common.py:27
↓ 1 callersFunctiontensor2sketch
(var)
utils/common.py:37
↓ 1 callersFunctiontforminv
Function: ---------- apply the inverse of affine transform 'trans' to uv Parameters: ---------- @trans: 3x3 n
models/mtcnn/mtcnn_pytorch/src/matlab_cp2tform.py:45
↓ 1 callersMethodtrain_fewshot
(self)
training/coach.py:114
↓ 1 callersFunctionvis_faces_no_id
(hooks_dict, fig, gs, i)
utils/common.py:85
↓ 1 callersFunctionvis_faces_with_id
(hooks_dict, fig, gs, i)
utils/common.py:73
↓ 1 callersMethodz_score
(self, x: torch.Tensor)
criteria/lpips/networks.py:50
next →1–100 of 246, ranked by callers