MCPcopy Create free account

hub / github.com/cunjian/pytorch_face_landmark / functions

Functions362 in github.com/cunjian/pytorch_face_landmark

↓ 46 callersMethodappend
(self, numbers)
utils/logger.py:61
↓ 36 callersFunctionconv_dw
(inp, oup, stride, leaky=0.1)
Retinaface/models/net.py:29
↓ 21 callersMethodload
(self, model)
vision/ssd/ssd.py:137
↓ 11 callersMethodwrite
(self, s)
utils/progress/progress/helpers.py:37
↓ 8 callersFunctiongauss
(x,a,b,c)
utils/visualize.py:18
↓ 7 callersFunctionSeperableConv2d
Replace Conv2d with a depthwise Conv2d and Pointwise Conv2d.
vision/ssd/mb_tiny_fd.py:9
↓ 7 callersFunctionSeperableConv2d
Replace Conv2d with a depthwise Conv2d and Pointwise Conv2d.
vision/ssd/mb_tiny_RFB_fd.py:9
↓ 7 callersFunctionconv_bn
(inp, oup, stride = 1, leaky = 0)
Retinaface/models/net.py:9
↓ 6 callersMethod__init__
(self, embedding_size)
models/mobilefacenet.py:89
↓ 6 callersFunction_convert
(image, alpha=1, beta=0)
Retinaface/data/data_augment.py:82
↓ 5 callersFunctiondrawLandmark_multiple
Input: - img: gray or RGB - bbox: type of BBox - landmark: reproject landmark of (5L, 2L) Output: - img marked with lan
common/utils.py:49
↓ 5 callersFunctionmake_image
(img, mean=(0,0,0), std=(1,1,1))
utils/visualize.py:12
↓ 5 callersFunctionnms
Dispatch to either CPU or GPU NMS implementations.
FaceBoxes/utils/nms_wrapper.py:13
↓ 5 callersMethodreprojectLandmark
(self, landmark)
common/utils.py:26
↓ 5 callersFunctionsleep
()
utils/progress/test_progress.py:16
↓ 5 callersFunctiontformfwd
Function: ---------- apply affine transform 'trans' to uv Parameters: ---------- @trans: 3x3 np.array tra
utils/matlab_cp2tform.py:11
↓ 4 callersMethod__init__
(self, channels, reduction)
models/basenet.py:33
↓ 4 callersMethoditer
(self, it)
utils/progress/progress/__init__.py:75
↓ 3 callersMethod__init__
(self, in_channels, out_channels, **kwargs)
FaceBoxes/models/faceboxes.py:51
↓ 3 callersMethod__init__
(self,inchannels=512,num_anchors=3)
Retinaface/models/retinaface.py:27
↓ 3 callersMethod__init__
(self)
MTCNN/get_nets.py:29
↓ 3 callersFunctionarea_of
Compute the areas of rectangles given two corners. Args: left_top (N, 2): left top corner. right_bottom (N, 2): right bottom corn
vision/utils/box_utils_numpy.py:40
↓ 3 callersFunctionarea_of
Compute the areas of rectangles given two corners. Args: left_top (N, 2): left top corner. right_bottom (N, 2): right bottom corn
vision/utils/box_utils.py:68
↓ 3 callersFunctioncalibrate_box
Transform bounding boxes to be more like true bounding boxes. 'offsets' is one of the outputs of the nets. Arguments: bboxes: a float
MTCNN/box_utils.py:94
↓ 3 callersFunctionconv_bn1X1
(inp, oup, stride, leaky=0)
Retinaface/models/net.py:22
↓ 3 callersFunctionconv_bn_no_relu
(inp, oup, stride)
Retinaface/models/net.py:16
↓ 3 callersFunctionfindNonreflectiveSimilarity
(uv, xy, options=None)
utils/matlab_cp2tform.py:56
↓ 3 callersMethodfinish
(self)
utils/progress/progress/__init__.py:64
↓ 3 callersFunctioniou_of
Return intersection-over-union (Jaccard index) of boxes. Args: boxes0 (N, 4): ground truth boxes. boxes1 (N or 1, 4): predicted b
vision/utils/box_utils.py:82
↓ 3 callersMethodnext
(self, n=1)
utils/progress/progress/__init__.py:67
↓ 3 callersMethodsave
(self, model_path)
vision/ssd/ssd.py:140
↓ 2 callersMethod__init__
(self, in_channel, out_channel)
Retinaface/models/net.py:41
↓ 2 callersMethod__init__
(self, message=None, **kwargs)
utils/progress/progress/helpers.py:25
↓ 2 callersMethod__init__
(self, in_planes, out_planes, stride=1, scale=0.1, map_reduce=8, vision=1, groups=1)
vision/nn/mb_tiny_RFB.py:31
↓ 2 callersMethod_get_annotation
(self, image_id)
vision/datasets/voc_dataset.py:88
↓ 2 callersFunction_preprocess
Preprocessing step before feeding the network. Arguments: img: a float numpy array of shape [h, w, c]. Returns: a float nump
MTCNN/box_utils.py:226
↓ 2 callersMethod_read_image
(self, image_id)
vision/datasets/voc_dataset.py:115
↓ 2 callersMethodcompute_header
(self, i, x)
vision/ssd/ssd.py:103
↓ 2 callersFunctionconvert_to_square
Convert bounding boxes to a square form. Arguments: bboxes: a float numpy array of shape [n, 5]. Returns: a float numpy arra
MTCNN/box_utils.py:71
↓ 2 callersFunctiondetect_faces
Arguments: image: an instance of PIL.Image. min_face_size: a float number. thresholds: a list of length 3. nms_th
MTCNN/detector.py:9
↓ 2 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.
MTCNN/box_utils.py:127
↓ 2 callersFunctionget_reference_facial_points
Function: ---------- get reference 5 key points according to crop settings: 0. Set default crop_size: if default_
utils/align_trans.py:25
↓ 2 callersFunctionget_similarity_transform
Function: ---------- Find Similarity Transform Matrix 'trans': u = src_pts[:, 0] v = src_pts[:, 1]
utils/matlab_cp2tform.py:149
↓ 2 callersFunctionintersect
(box_a, box_b)
vision/transforms/transforms.py:13
↓ 2 callersMethodplot
(self, names=None)
utils/logger.py:70
↓ 2 callersFunctionremove_prefix
Old style model is stored with all names of parameters sharing common prefix 'module.'
FaceBoxes/utils/functions.py:20
↓ 2 callersFunctionremove_prefix
Old style model is stored with all names of parameters sharing common prefix 'module.'
Retinaface/Retinaface.py:40
↓ 2 callersMethodtic
(self)
FaceBoxes/utils/timer.py:23
↓ 2 callersMethodtoc
(self, average=True)
FaceBoxes/utils/timer.py:28
↓ 2 callersMethodupdate
(self)
utils/progress/progress/__init__.py:58
↓ 2 callersMethodwriteln
(self, line)
utils/progress/progress/helpers.py:65
↓ 1 callersMethod__init__
(self, *args, **kwargs)
utils/progress/progress/__init__.py:31
↓ 1 callersMethod__init__
(self)
models/pfld_compressed.py:62
↓ 1 callersFunction_crop
(image, boxes, labels, landm, img_dim)
Retinaface/data/data_augment.py:7
↓ 1 callersFunction_distort
(image)
Retinaface/data/data_augment.py:80
↓ 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]. offs
MTCNN/first_stage.py:48
↓ 1 callersMethod_initialize_weights
(self)
models/mobilefacenet.py:125
↓ 1 callersMethod_make_bbox_head
(self,fpn_num=3,inchannels=64,anchor_num=2)
Retinaface/models/retinaface.py:95
↓ 1 callersMethod_make_class_head
(self,fpn_num=3,inchannels=64,anchor_num=2)
Retinaface/models/retinaface.py:89
↓ 1 callersMethod_make_landmark_head
(self,fpn_num=3,inchannels=64,anchor_num=2)
Retinaface/models/retinaface.py:101
↓ 1 callersFunction_mirror
(image, boxes, landms)
Retinaface/data/data_augment.py:167
↓ 1 callersFunction_pad_to_square
(image, rgb_mean, pad_image_flag)
Retinaface/data/data_augment.py:189
↓ 1 callersMethod_read_image_ids
(image_sets_file)
vision/datasets/voc_dataset.py:81
↓ 1 callersFunction_resize_subtract_mean
(image, insize, rgb_mean)
Retinaface/data/data_augment.py:200
↓ 1 callersFunctioncheck_keys
(model, pretrained_state_dict)
FaceBoxes/utils/functions.py:7
↓ 1 callersFunctioncheck_keys
(model, pretrained_state_dict)
Retinaface/Retinaface.py:27
↓ 1 callersMethodclear
(self)
Retinaface/utils/timer.py:35
↓ 1 callersMethodclearln
(self)
utils/progress/progress/helpers.py:61
↓ 1 callersMethodclose
(self)
utils/logger.py:79
↓ 1 callersFunctionconv_bn
(inp, oup, kernel, stride, padding=1)
models/pfld_compressed.py:14
↓ 1 callersMethodconv_dw
(inp, oup, stride)
vision/nn/mb_tiny.py:18
↓ 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],
MTCNN/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():
utils/matlab_cp2tform.py:187
↓ 1 callersFunctiondecode
Decode locations from predictions using priors to undo the encoding we did for offset regression at train time. Args: loc (tensor): lo
FaceBoxes/utils/box_utils.py:177
↓ 1 callersFunctiondecode
Decode locations from predictions using priors to undo the encoding we did for offset regression at train time. Args: loc (tensor): lo
Retinaface/utils/box_utils.py:209
↓ 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
Retinaface/utils/box_utils.py:229
↓ 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
FaceBoxes/utils/box_utils.py:152
↓ 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
Retinaface/utils/box_utils.py:156
↓ 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
Retinaface/utils/box_utils.py:179
↓ 1 callersMethodend
(self, key="default")
vision/utils/misc.py:17
↓ 1 callersFunctionfindSimilarity
(uv, xy, options=None)
utils/matlab_cp2tform.py:108
↓ 1 callersMethodfinish
(self)
utils/progress/progress/helpers.py:45
↓ 1 callersMethodforward
(self)
FaceBoxes/utils/prior_box.py:20
↓ 1 callersMethodforward
(self)
Retinaface/layers/functions/prior_box.py:17
↓ 1 callersMethodforward
(self, x: torch.Tensor)
vision/ssd/ssd.py:42
↓ 1 callersFunctiongenerate_priors
(feature_map_list, shrinkage_list, image_size, min_boxes, clamp=True)
vision/utils/box_utils.py:6
↓ 1 callersFunctionget_affine_transform_matrix
Function: ---------- get affine transform matrix 'tfm' from src_pts to dst_pts Parameters: ---------- @src_pts: Kx2 n
utils/align_trans.py:156
↓ 1 callersFunctionget_similarity_transform_for_cv2
Function: ---------- Find Similarity Transform Matrix 'cv2_trans' which could be directly used by cv2.warpAffine():
utils/matlab_cp2tform.py:213
↓ 1 callersMethodgoto
(self, index)
utils/progress/progress/__init__.py:112
↓ 1 callersFunctionhard_nms
Args: box_scores (N, 5): boxes in corner-form and probabilities. iou_threshold: intersection over union threshold. top_k
vision/utils/box_utils.py:168
↓ 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
FaceBoxes/utils/box_utils.py:31
↓ 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
Retinaface/utils/box_utils.py:29
↓ 1 callersFunctioniou_of
Return intersection-over-union (Jaccard index) of boxes. Args: boxes0 (N, 4): ground truth boxes. boxes1 (N or 1, 4): predicted b
vision/utils/box_utils_numpy.py:54
↓ 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
FaceBoxes/utils/box_utils.py:52
↓ 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
Retinaface/utils/box_utils.py:50
↓ 1 callersFunctionjaccard_numpy
Compute the jaccard overlap of two sets of boxes. The jaccard overlap is simply the intersection over union of two boxes. E.g.: A ∩ B
vision/transforms/transforms.py:20
↓ 1 callersFunctionload_model
()
test_batch_detections.py:46
↓ 1 callersFunctionload_model
(model, pretrained_path, load_to_cpu)
FaceBoxes/utils/functions.py:27
↓ 1 callersFunctionload_model
(model, pretrained_path, load_to_cpu)
Retinaface/Retinaface.py:47
↓ 1 callersFunctionlog_sum_exp
Utility function for computing log_sum_exp while determining This will be used to determine unaveraged confidence loss across all examples in
Retinaface/utils/box_utils.py:250
next →1–100 of 362, ranked by callers