MCPcopy Create free account

hub / github.com/biubug6/Pytorch_Retinaface / functions

Functions80 in github.com/biubug6/Pytorch_Retinaface

↓ 13 callersFunctionconv_dw
(inp, oup, stride, leaky=0.1)
models/net.py:29
↓ 6 callersFunction_convert
(image, alpha=1, beta=0)
data/data_augment.py:82
↓ 5 callersFunctionconv_bn
(inp, oup, stride = 1, leaky = 0)
models/net.py:9
↓ 4 callersMethodforward
(self, input)
models/net.py:55
↓ 4 callersMethodtic
(self)
utils/timer.py:20
↓ 4 callersMethodtoc
(self, average=True)
utils/timer.py:25
↓ 3 callersMethod__init__
(self,inchannels=512,num_anchors=3)
models/retinaface.py:27
↓ 3 callersFunctionconv_bn1X1
(inp, oup, stride, leaky=0)
models/net.py:22
↓ 3 callersFunctionconv_bn_no_relu
(inp, oup, stride)
models/net.py:16
↓ 3 callersFunctiondecode
Decode locations from predictions using priors to undo the encoding we did for offset regression at train time. Args: loc (tensor): lo
utils/box_utils.py:209
↓ 3 callersFunctiondecode_landm
Decode landm from predictions using priors to undo the encoding we did for offset regression at train time. Args: pre (tensor): landm
utils/box_utils.py:229
↓ 3 callersFunctionpy_cpu_nms
Pure Python NMS baseline.
utils/nms/py_cpu_nms.py:10
↓ 2 callersMethod__init__
(self, in_channel, out_channel)
models/net.py:41
↓ 2 callersFunctionremove_prefix
Old style model is stored with all names of parameters sharing common prefix 'module.'
test_widerface.py:46
↓ 2 callersFunctionremove_prefix
Old style model is stored with all names of parameters sharing common prefix 'module.'
detect.py:43
↓ 2 callersFunctionremove_prefix
Old style model is stored with all names of parameters sharing common prefix 'module.'
convert_to_onnx.py:39
↓ 2 callersFunctionremove_prefix
Old style model is stored with all names of parameters sharing common prefix 'module.'
test_fddb.py:45
↓ 1 callersFunction_crop
(image, boxes, labels, landm, img_dim)
data/data_augment.py:7
↓ 1 callersFunction_distort
(image)
data/data_augment.py:80
↓ 1 callersMethod_make_bbox_head
(self,fpn_num=3,inchannels=64,anchor_num=2)
models/retinaface.py:95
↓ 1 callersMethod_make_class_head
(self,fpn_num=3,inchannels=64,anchor_num=2)
models/retinaface.py:89
↓ 1 callersMethod_make_landmark_head
(self,fpn_num=3,inchannels=64,anchor_num=2)
models/retinaface.py:101
↓ 1 callersFunction_mirror
(image, boxes, landms)
data/data_augment.py:167
↓ 1 callersFunction_pad_to_square
(image, rgb_mean, pad_image_flag)
data/data_augment.py:189
↓ 1 callersFunction_resize_subtract_mean
(image, insize, rgb_mean)
data/data_augment.py:200
↓ 1 callersFunctionadjust_learning_rate
Sets the learning rate # Adapted from PyTorch Imagenet example: # https://github.com/pytorch/examples/blob/master/imagenet/main.py
train.py:145
↓ 1 callersFunctioncheck_keys
(model, pretrained_state_dict)
test_widerface.py:33
↓ 1 callersFunctioncheck_keys
(model, pretrained_state_dict)
detect.py:30
↓ 1 callersFunctioncheck_keys
(model, pretrained_state_dict)
convert_to_onnx.py:26
↓ 1 callersFunctioncheck_keys
(model, pretrained_state_dict)
test_fddb.py:32
↓ 1 callersMethodclear
(self)
utils/timer.py:35
↓ 1 callersFunctiondataset_pr_info
(thresh_num, pr_curve, count_face)
widerface_evaluate/evaluation.py:199
↓ 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
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
utils/box_utils.py:179
↓ 1 callersFunctionevaluation
(pred, gt_path, iou_thresh=0.5)
widerface_evaluate/evaluation.py:227
↓ 1 callersFunctionget_gt_boxes
gt dir: (wider_face_val.mat, wider_easy_val.mat, wider_medium_val.mat, wider_hard_val.mat)
widerface_evaluate/evaluation.py:18
↓ 1 callersFunctionget_preds
(pred_dir)
widerface_evaluate/evaluation.py:104
↓ 1 callersFunctionimage_eval
single image evaluation pred: Nx5 gt: Nx4 ignore:
widerface_evaluate/evaluation.py:146
↓ 1 callersFunctionimg_pr_info
(thresh_num, pred_info, proposal_list, pred_recall)
widerface_evaluate/evaluation.py:182
↓ 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
utils/box_utils.py:29
↓ 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
utils/box_utils.py:50
↓ 1 callersFunctionload_model
(model, pretrained_path, load_to_cpu)
test_widerface.py:53
↓ 1 callersFunctionload_model
(model, pretrained_path, load_to_cpu)
detect.py:50
↓ 1 callersFunctionload_model
(model, pretrained_path, load_to_cpu)
convert_to_onnx.py:46
↓ 1 callersFunctionload_model
(model, pretrained_path, load_to_cpu)
test_fddb.py:52
↓ 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
utils/box_utils.py:250
↓ 1 callersFunctionmatch
Match each prior box with the ground truth box of the highest jaccard overlap, encode the bounding boxes, then return the matched indices corr
utils/box_utils.py:96
↓ 1 callersFunctionmatrix_iof
return iof of a and b, numpy version for data augenmentation
utils/box_utils.py:84
↓ 1 callersFunctionnorm_score
norm score pred {key: [[x1,y1,x2,y2,s]]}
widerface_evaluate/evaluation.py:121
↓ 1 callersFunctionpoint_form
Convert prior_boxes to (xmin, ymin, xmax, ymax) representation for comparison to point form ground truth data. Args: boxes: (tensor)
utils/box_utils.py:5
↓ 1 callersFunctionread_pred_file
(filepath)
widerface_evaluate/evaluation.py:82
↓ 1 callersFunctiontrain
()
train.py:89
↓ 1 callersFunctionvoc_ap
(rec, prec)
widerface_evaluate/evaluation.py:207
Method__call__
(self, image, targets)
data/data_augment.py:215
Method__getitem__
(self, index)
data/wider_face.py:40
Method__init__
(self)
utils/timer.py:13
Method__init__
(self, txt_path, preproc=None)
data/wider_face.py:10
Method__init__
(self, img_dim, rgb_means)
data/data_augment.py:211
Method__init__
(self, num_classes, overlap_thresh, prior_for_matching, bkg_label, neg_mining, neg_pos, neg_overlap, encode_ta
layers/modules/multibox_loss.py:32
Method__init__
(self, cfg, image_size=None, phase='train')
layers/functions/prior_box.py:8
Method__init__
(self,in_channels_list,out_channels)
models/net.py:69
Method__init__
(self)
models/net.py:103
Method__init__
(self,inchannels=512,num_anchors=3)
models/retinaface.py:15
Method__init__
(self,inchannels=512,num_anchors=3)
models/retinaface.py:38
Method__init__
:param cfg: Network related settings. :param phase: train or test.
models/retinaface.py:49
Method__len__
(self)
data/wider_face.py:37
Function_expand
(image, boxes, fill, p)
data/data_augment.py:141
Functioncenter_size
Convert prior_boxes to (cx, cy, w, h) representation for comparison to center-size form ground truth data. Args: boxes: (tensor) poin
utils/box_utils.py:17
Functiondetection_collate
Custom collate fn for dealing with batches of images that have a different number of associated object annotations (bounding boxes). Argument
data/wider_face.py:79
Methodforward
Multibox Loss Args: predictions (tuple): A tuple containing loc preds, conf preds, and prior boxes from SSD net.
layers/modules/multibox_loss.py:44
Methodforward
(self)
layers/functions/prior_box.py:17
Methodforward
(self, input)
models/net.py:81
Methodforward
(self, x)
models/net.py:128
Methodforward
(self,x)
models/retinaface.py:20
Methodforward
(self,x)
models/retinaface.py:31
Methodforward
(self,x)
models/retinaface.py:42
Methodforward
(self,inputs)
models/retinaface.py:107
Functionget_gt_boxes_from_txt
(gt_path, cache_dir)
widerface_evaluate/evaluation.py:37
Functionmatrix_iou
return iou of a and b, numpy version for data augenmentation
utils/box_utils.py:71
Functionnms
Apply non-maximum suppression at test time to avoid detecting too many overlapping bounding boxes for a given object. Args: boxes: (te
utils/box_utils.py:264