MCPcopy Create free account

hub / github.com/xinntao/facexlib / functions

Functions245 in github.com/xinntao/facexlib

↓ 13 callersFunctionconv_dw
(inp, oup, stride, leaky=0.1)
facexlib/detection/retinaface_net.py:25
↓ 12 callersFunctionget_block
(in_channel, depth, num_units, stride=2)
facexlib/recognition/arcface_arch.py:81
↓ 8 callersMethod__init__
(self, channels, reduction)
facexlib/recognition/arcface_arch.py:23
↓ 8 callersMethoddetect_faces
( self, image, conf_threshold=0.8, nms_threshold=0.4, use_origin_size=
facexlib/detection/retinaface.py:192
↓ 8 callersFunctioninit_detection_model
(model_name, half=False, device='cuda', model_rootpath=None)
facexlib/detection/__init__.py:8
↓ 7 callersFunctionload_file_from_url
Ref:https://github.com/1adrianb/face-alignment/blob/master/face_alignment/utils.py
facexlib/utils/misc.py:59
↓ 6 callersMethod__init__
(self, in_channels)
facexlib/matting/modnet.py:16
↓ 5 callersMethod__init__
(self, *args)
facexlib/assessment/hyperiqa_net.py:11
↓ 5 callersMethod__init__
(self, in_channel, out_channel)
facexlib/detection/retinaface_net.py:38
↓ 5 callersMethod__init__
(self, num_class)
facexlib/parsing/bisenet.py:112
↓ 5 callersMethod__init__
(self, num_modules=1, end_relu=False, gray_scale=False, num_landmarks=68, device='cuda')
facexlib/alignment/awing_arch.py:271
↓ 5 callersFunctionconv3x3
3x3 convolution with padding
facexlib/alignment/awing_arch.py:130
↓ 5 callersFunctionconv_bn
(inp, oup, stride=1, leaky=0)
facexlib/detection/retinaface_net.py:6
↓ 5 callersFunctionimg2tensor
Numpy array to tensor. Args: imgs (list[ndarray] | ndarray): Input images. bgr2rgb (bool): Whether to change bgr to rgb.
facexlib/utils/misc.py:30
↓ 5 callersFunctiontformfwd
Function: ---------- apply affine transform 'trans' to uv Parameters: ---------- @trans: 3x3 np.array tr
facexlib/detection/matlab_cp2tform.py:13
↓ 4 callersMethod__init__
(self, in_size=128, out_size=128, min_feat_size=32,
facexlib/parsing/parsenet.py:142
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
facexlib/assessment/hyperiqa_net.py:194
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
facexlib/headpose/hopenet_arch.py:29
↓ 4 callersFunctioncreate_layer_basic
(in_chan, out_chan, bnum, stride=1)
facexlib/parsing/resnet.py:41
↓ 4 callersFunctionget_location
(val, length)
facexlib/utils/face_restoration_helper.py:14
↓ 4 callersMethodupdate
This method must be called once for each frame even with empty detections. NOTE:as in practical realtime MOT, the detector doesn't run
facexlib/tracking/sort.py:20
↓ 3 callersFunction_make_divisible
(v, divisor, min_value=None)
facexlib/matting/mobilenetv2.py:12
↓ 3 callersMethodbin2degree
(self, predict)
facexlib/headpose/hopenet_arch.py:50
↓ 3 callersFunctionconv_bn1X1
(inp, oup, stride, leaky=0)
facexlib/detection/retinaface_net.py:19
↓ 3 callersFunctionconv_bn_no_relu
(inp, oup, stride)
facexlib/detection/retinaface_net.py:12
↓ 3 callersFunctionfindNonreflectiveSimilarity
(uv, xy, options=None)
facexlib/detection/matlab_cp2tform.py:60
↓ 3 callersFunctioninit_parsing_model
(model_name='bisenet', half=False, device='cuda', model_rootpath=None)
facexlib/parsing/__init__.py:8
↓ 2 callersMethod__detect_faces
(self, inputs)
facexlib/detection/retinaface.py:145
↓ 2 callersFunction_totensor
(img, bgr2rgb, float32)
facexlib/utils/misc.py:43
↓ 2 callersMethodalign_warp_face
Align and warp faces with face template.
facexlib/utils/face_restoration_helper.py:234
↓ 2 callersMethodclean_all
(self)
facexlib/utils/face_restoration_helper.py:367
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
facexlib/parsing/resnet.py:5
↓ 2 callersFunctionconvert_bbox_to_z
Takes a bounding box in the form [x1,y1,x2,y2] and returns z in the form [x,y,s,r] where x,y is the centre of the box and s is the scale/area and
facexlib/tracking/kalman_tracker.py:5
↓ 2 callersFunctionconvert_x_to_bbox
Takes a bounding box in the centre form [x,y,s,r] and returns it in the form [x1,y1,x2,y2] where x1,y1 is the top left and x2,y2 is the bottom
facexlib/tracking/kalman_tracker.py:19
↓ 2 callersMethodget_face_landmarks_5
(self, only_keep_largest=False, only_center_face=Fal
facexlib/utils/face_restoration_helper.py:123
↓ 2 callersFunctionget_largest_face
(det_faces, h, w)
facexlib/utils/face_restoration_helper.py:12
↓ 2 callersFunctionget_reference_facial_points
Function: ---------- get reference 5 key points according to crop settings: 0. Set default crop_size: if default_
facexlib/detection/align_trans.py:19
↓ 2 callersFunctionget_similarity_transform
Function: ---------- Find Similarity Transform Matrix 'trans': u = src_pts[:, 0] v = src_pts[:, 1]
facexlib/detection/matlab_cp2tform.py:130
↓ 2 callersFunctionimwrite
Write image to file. Args: img (ndarray): Image array to be written. file_path (str): Image file path. params (None or li
facexlib/utils/misc.py:11
↓ 2 callersFunctionl2_norm
(input, axis=1)
facexlib/recognition/arcface_arch.py:15
↓ 2 callersFunctionpy_cpu_nms
Pure Python NMS baseline.
facexlib/detection/retinaface_utils.py:39
↓ 2 callersMethodread_image
img can be image path or cv2 loaded image.
facexlib/utils/face_restoration_helper.py:108
↓ 2 callersMethodupdate
Updates the state vector with observed bbox.
facexlib/tracking/kalman_tracker.py:78
↓ 2 callersFunctionvisualize_detection
Visualize detection results. Args: img (Numpy array): Input image. CHW, BGR, [0, 255], uint8.
facexlib/visualization/vis_detection.py:5
↓ 1 callersMethod__align_multi
(self, image, boxes, landmarks, limit=None)
facexlib/detection/retinaface.py:235
↓ 1 callersMethod__init__
(self)
facexlib/parsing/resnet.py:50
↓ 1 callersMethod__init__
(self, in_channels)
facexlib/matting/backbone.py:12
↓ 1 callersMethod__init__
(self, in_channels, alpha=1.0, expansion=6, num_classes=1000)
facexlib/matting/mobilenetv2.py:84
↓ 1 callersMethod_forward
(self, level, inp)
facexlib/alignment/awing_arch.py:242
↓ 1 callersMethod_generate_network
(self, level)
facexlib/alignment/awing_arch.py:230
↓ 1 callersMethod_init_conv
(self, conv)
facexlib/matting/modnet.py:259
↓ 1 callersMethod_init_norm
(self, norm)
facexlib/matting/modnet.py:264
↓ 1 callersMethod_init_weights
(self)
facexlib/matting/mobilenetv2.py:179
↓ 1 callersFunction_minimal_ext_cmd
(cmd)
setup.py:20
↓ 1 callersFunction_scandir
(dir_path, suffix, recursive)
facexlib/utils/misc.py:100
↓ 1 callersFunctionalign_crop_face_landmarks
Align and crop face with landmarks. The output_size and transform_size are based on width. The height is adjusted based on shrink_ratio_h/shr
facexlib/utils/face_utils.py:31
↓ 1 callersMethodalign_multi
(self, img, conf_threshold=0.8, limit=None)
facexlib/detection/retinaface.py:253
↓ 1 callersFunctionassociate_detections_to_trackers
Assigns detections to tracked object (both represented as bounding boxes) Returns: 3 lists of matches, unmatched_detections and unmatched
facexlib/tracking/data_association.py:29
↓ 1 callersFunctionbatched_decode
Decode locations from predictions using priors to undo the encoding we did for offset regression at train time. Args: b_loc (tensor):
facexlib/detection/retinaface_utils.py:297
↓ 1 callersFunctionbatched_decode_landm
Decode landm from predictions using priors to undo the encoding we did for offset regression at train time. Args: pre (tensor): landm
facexlib/detection/retinaface_utils.py:320
↓ 1 callersMethodbatched_transform
Arguments: frames: a list of PIL.Image, or torch.Tensor(shape=[n, h, w, c], type=np.float32, BGR format).
facexlib/detection/retinaface.py:261
↓ 1 callersFunctioncalculate_points
(heatmaps)
facexlib/alignment/awing_arch.py:8
↓ 1 callersFunctionconv_1x1_bn
(inp, oup)
facexlib/matting/mobilenetv2.py:26
↓ 1 callersFunctionconv_bn
(inp, oup, stride)
facexlib/matting/mobilenetv2.py:22
↓ 1 callersFunctioncrop_one_img
(img, save_cropped_path=None)
scripts/crop_faces_5landmarks.py:7
↓ 1 callersFunctioncvt_tform_mat_for_cv2
Function: ---------- Convert Transform Matrix 'trans' into 'cv2_trans' which could be directly used by cv2.warpAffine():
facexlib/detection/matlab_cp2tform.py:170
↓ 1 callersFunctiondecode
Decode locations from predictions using priors to undo the encoding we did for offset regression at train time. Args: loc (tensor): lo
facexlib/detection/retinaface_utils.py:254
↓ 1 callersFunctiondecode_landm
Decode landm from predictions using priors to undo the encoding we did for offset regression at train time. Args: pre (tensor): landm
facexlib/detection/retinaface_utils.py:274
↓ 1 callersFunctiondraw_and_save
Visualize results
scripts/get_ffhq_template.py:14
↓ 1 callersFunctiondraw_axis
draw head pose axis.
facexlib/visualization/vis_headpose.py:6
↓ 1 callersFunctiondraw_pose_cube
draw head pose cube. Where (tdx, tdy) is the translation of the face. For pose we have [pitch yaw roll tdx tdy tdz scale_factor]
facexlib/visualization/vis_headpose.py:35
↓ 1 callersFunctionencode
Encode the variances from the priorbox layers into the ground truth boxes we have matched (based on jaccard overlap) with the prior boxes. Arg
facexlib/detection/retinaface_utils.py:200
↓ 1 callersFunctionencode_landm
Encode the variances from the priorbox layers into the ground truth boxes we have matched (based on jaccard overlap) with the prior boxes. Arg
facexlib/detection/retinaface_utils.py:224
↓ 1 callersFunctionfindSimilarity
(uv, xy, options=None)
facexlib/detection/matlab_cp2tform.py:94
↓ 1 callersMethodforward
(self)
facexlib/detection/retinaface_utils.py:19
↓ 1 callersMethodforward
(self, x)
facexlib/alignment/awing_arch.py:324
↓ 1 callersMethodforward
(self, x)
facexlib/matting/modnet.py:25
↓ 1 callersFunctiongenerate_config
(network_name)
facexlib/detection/retinaface.py:15
↓ 1 callersFunctionget_affine_transform_matrix
Function: ---------- get affine transform matrix 'tfm' from src_pts to dst_pts Parameters: ---------- @src_pts: Kx2 n
facexlib/detection/align_trans.py:112
↓ 1 callersFunctionget_blocks
(num_layers)
facexlib/recognition/arcface_arch.py:85
↓ 1 callersFunctionget_center_face
(det_faces, h=0, w=0, center=None)
facexlib/utils/face_restoration_helper.py:34
↓ 1 callersFunctionget_center_landmark
(landmarks, center)
scripts/extract_detection_info_ffhq.py:52
↓ 1 callersFunctionget_git_hash
()
setup.py:18
↓ 1 callersFunctionget_hash
()
setup.py:43
↓ 1 callersMethodget_landmarks
(self, img)
facexlib/alignment/awing_arch.py:359
↓ 1 callersFunctionget_requirements
(filename='requirements.txt')
setup.py:75
↓ 1 callersFunctionget_similarity_transform_for_cv2
Function: ---------- Find Similarity Transform Matrix 'cv2_trans' which could be directly used by cv2.warpAffine():
facexlib/detection/matlab_cp2tform.py:198
↓ 1 callersMethodget_state
Returns the current bounding box estimate.
facexlib/tracking/kalman_tracker.py:106
↓ 1 callersFunctionget_version
()
setup.py:69
↓ 1 callersFunctioninit_alignment_model
(model_name, half=False, device='cuda', model_rootpath=None)
facexlib/alignment/__init__.py:10
↓ 1 callersFunctioninit_assessment_model
(model_name, half=False, device='cuda', model_rootpath=None)
facexlib/assessment/__init__.py:7
↓ 1 callersFunctioninit_headpose_model
(model_name, half=False, device='cuda', model_rootpath=None)
facexlib/headpose/__init__.py:7
↓ 1 callersFunctioninit_matting_model
(model_name='modnet', half=False, device='cuda', model_rootpath=None)
facexlib/matting/__init__.py:8
↓ 1 callersFunctionintersect
We resize both tensors to [A,B,2] without new malloc: [A,2] -> [A,1,2] -> [A,B,2] [B,2] -> [1,B,2] -> [A,B,2] Then we compute the area of
facexlib/detection/retinaface_utils.py:79
↓ 1 callersFunctioniou
Computes IOU between two bboxes in the form [x1,y1,x2,y2]
facexlib/tracking/data_association.py:14
↓ 1 callersFunctionjaccard
Compute the jaccard overlap of two sets of boxes. The jaccard overlap is simply the intersection over union of two boxes. Here we operate on
facexlib/detection/retinaface_utils.py:98
↓ 1 callersFunctionlandmark_98_to_68
Transfer 98 landmark positions to 68 landmark positions. Args: landmark_98(numpy array): Polar coordinates of 98 landmarks, (98, 2) Re
facexlib/alignment/convert_98_to_68_landmarks.py:40
↓ 1 callersMethodload_pretrained_ckpt
(self)
facexlib/matting/backbone.py:22
↓ 1 callersFunctionload_txt_file
Load data or string from txt file.
facexlib/alignment/convert_98_to_68_landmarks.py:4
↓ 1 callersFunctionmain
(args)
inference/inference_alignment.py:9
next →1–100 of 245, ranked by callers