MCPcopy Create free account

hub / github.com/amdegroot/ssd.pytorch / functions

Functions110 in github.com/amdegroot/ssd.pytorch

↓ 4 callersFunctionbuild_ssd
(phase, size=300, num_classes=21)
ssd.py:198
↓ 2 callersFunctioncreate_vis_plot
(_xlabel, _ylabel, _title, _legend)
train.py:223
↓ 2 callersFunctionget_voc_results_file_template
(image_set, cls)
eval.py:136
↓ 2 callersMethodpull_item
Args: index (int): Index Returns: tuple: Tuple (image, target, height, width). target is t
data/coco.py:112
↓ 2 callersFunctionupdate_vis_plot
(iteration, loc, conf, window1, window2, update_type, epoch_size=1)
train.py:236
↓ 1 callersMethod__len__
(self)
data/coco.py:109
↓ 1 callersFunctionadd_extras
(cfg, i, batch_norm=False)
ssd.py:149
↓ 1 callersFunctionadjust_learning_rate
Sets the learning rate to the initial LR decayed by 10 at every specified step # Adapted from PyTorch Imagenet example: # https://gith
train.py:202
↓ 1 callersFunctionbase_transform
(image, size, mean)
data/__init__.py:30
↓ 1 callersFunctioncv2_demo
(net, transform)
demo/live.py:20
↓ 1 callersFunctiondecode
Decode locations from predictions using priors to undo the encoding we did for offset regression at train time. Args: loc (tensor): lo
layers/box_utils.py:140
↓ 1 callersFunctiondo_python_eval
(output_dir='output', use_07=True)
eval.py:163
↓ 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
layers/box_utils.py:115
↓ 1 callersFunctionevaluate_detections
(box_list, output_dir, dataset)
eval.py:416
↓ 1 callersMethodforward
Applies network layers and ops on input image(s) x. Args: x: input image or batch of images. Shape: [batch,3,300,300]. R
ssd.py:50
↓ 1 callersFunctionget_label_map
(label_file)
data/coco.py:33
↓ 1 callersFunctionget_output_dir
Return the directory where experimental artifacts are placed. If the directory does not exist, it is created. A canonical path is built using
eval.py:124
↓ 1 callersFunctionintersect
(box_a, box_b)
utils/augmentations.py:9
↓ 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
layers/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
layers/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
utils/augmentations.py:16
↓ 1 callersMethodload_weights
(self, base_file)
ssd.py:113
↓ 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
layers/box_utils.py:161
↓ 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
layers/box_utils.py:71
↓ 1 callersFunctionmultibox
(vgg, extra_layers, cfg, num_classes)
ssd.py:166
↓ 1 callersFunctionnms
Apply non-maximum suppression at test time to avoid detecting too many overlapping bounding boxes for a given object. Args: boxes: (te
layers/box_utils.py:175
↓ 1 callersFunctionparse_rec
Parse a PASCAL VOC xml file
eval.py:104
↓ 1 callersFunctionpoint_form
Convert prior_boxes to (xmin, ymin, xmax, ymax) representation for comparison to point form ground truth data. Args: boxes: (tensor)
layers/box_utils.py:5
↓ 1 callersFunctionpredict
(frame)
demo/live.py:21
↓ 1 callersMethodpull_anno
Returns the original annotation of image at index Note: not using self.__getitem__(), as any transformations passed in could mess up
data/coco.py:156
↓ 1 callersMethodpull_image
Returns the original image object at index in PIL form Note: not using self.__getitem__(), as any transformations passed in could mes
data/coco.py:141
↓ 1 callersMethodpull_image
Returns the original image object at index in PIL form Note: not using self.__getitem__(), as any transformations passed in could mes
data/voc0712.py:142
↓ 1 callersMethodpull_item
(self, index)
data/voc0712.py:122
↓ 1 callersMethodreset_parameters
(self)
layers/modules/l2norm.py:16
↓ 1 callersFunctiontest_net
(save_folder, net, cuda, dataset, transform, top_k, im_size=300, thresh=0.05)
eval.py:364
↓ 1 callersFunctiontest_net
(save_folder, net, cuda, testset, transform, thresh)
test.py:38
↓ 1 callersFunctiontest_voc
()
test.py:79
↓ 1 callersMethodtic
(self)
eval.py:88
↓ 1 callersMethodtoc
(self, average=True)
eval.py:93
↓ 1 callersFunctiontrain
()
train.py:71
↓ 1 callersFunctionvgg
(cfg, i, batch_norm=False)
ssd.py:126
↓ 1 callersFunctionvoc_ap
ap = voc_ap(rec, prec, [use_07_metric]) Compute VOC AP given precision and recall. If use_07_metric is true, uses the VOC 07 11 point met
eval.py:194
↓ 1 callersFunctionvoc_eval
rec, prec, ap = voc_eval(detpath, annopath, imagesetfile, classname,
eval.py:228
↓ 1 callersFunctionwrite_voc_results_file
(all_boxes, dataset)
eval.py:146
↓ 1 callersFunctionxavier
(param)
train.py:213
Method__call__
(self, img, boxes=None, labels=None)
utils/augmentations.py:50
Method__call__
(self, img, boxes=None, labels=None)
utils/augmentations.py:63
Method__call__
(self, image, boxes=None, labels=None)
utils/augmentations.py:68
Method__call__
(self, image, boxes=None, labels=None)
utils/augmentations.py:76
Method__call__
(self, image, boxes=None, labels=None)
utils/augmentations.py:83
Method__call__
(self, image, boxes=None, labels=None)
utils/augmentations.py:94
Method__call__
(self, image, boxes=None, labels=None)
utils/augmentations.py:108
Method__call__
(self, image, boxes=None, labels=None)
utils/augmentations.py:121
Method__call__
(self, image, boxes=None, labels=None)
utils/augmentations.py:133
Method__call__
(self, image, boxes=None, labels=None)
utils/augmentations.py:147
Method__call__
(self, image, boxes=None, labels=None)
utils/augmentations.py:160
Method__call__
(self, image, boxes=None, labels=None)
utils/augmentations.py:178
Method__call__
(self, image, boxes=None, labels=None)
utils/augmentations.py:191
Method__call__
(self, tensor, boxes=None, labels=None)
utils/augmentations.py:199
Method__call__
(self, cvimage, boxes=None, labels=None)
utils/augmentations.py:204
Method__call__
(self, image, boxes=None, labels=None)
utils/augmentations.py:234
Method__call__
(self, image, boxes, labels)
utils/augmentations.py:316
Method__call__
(self, image, boxes, classes)
utils/augmentations.py:341
Method__call__
Args: image (Tensor): image tensor to be transformed Return: a tensor with channels swapped according to swap
utils/augmentations.py:361
Method__call__
(self, image, boxes, labels)
utils/augmentations.py:389
Method__call__
(self, img, boxes, labels)
utils/augmentations.py:416
Method__call__
(self, image, boxes=None, labels=None)
data/__init__.py:42
Method__call__
Args: target (dict): COCO target json annotation as a python dict height (int): height width (int): width
data/coco.py:49
Method__call__
Arguments: target (annotation) : the target annotation to be made usable will be an ET.Element Returns:
data/voc0712.py:49
Method__getitem__
Args: index (int): Index Returns: tuple: Tuple (image, target). target is the object retur
data/coco.py:98
Method__getitem__
(self, index)
data/voc0712.py:114
Method__init__
(self)
eval.py:81
Method__init__
(self, phase, size, base, extras, head, num_classes)
ssd.py:28
Method__init__
(self, transforms)
utils/augmentations.py:47
Method__init__
(self, lambd)
utils/augmentations.py:59
Method__init__
(self, mean)
utils/augmentations.py:73
Method__init__
(self, size=300)
utils/augmentations.py:105
Method__init__
(self, lower=0.5, upper=1.5)
utils/augmentations.py:115
Method__init__
(self, delta=18.0)
utils/augmentations.py:129
Method__init__
(self)
utils/augmentations.py:142
Method__init__
(self, current='BGR', transform='HSV')
utils/augmentations.py:156
Method__init__
(self, lower=0.5, upper=1.5)
utils/augmentations.py:171
Method__init__
(self, delta=32)
utils/augmentations.py:186
Method__init__
(self)
utils/augmentations.py:221
Method__init__
(self, mean)
utils/augmentations.py:313
Method__init__
(self, swaps)
utils/augmentations.py:358
Method__init__
(self)
utils/augmentations.py:377
Method__init__
(self, size=300, mean=(104, 117, 123))
utils/augmentations.py:401
Method__init__
(self, size, mean)
data/__init__.py:38
Method__init__
(self)
data/coco.py:46
Method__init__
(self, root, image_set='trainval35k', transform=None, target_transform=COCOAnnotationTransfor
data/coco.py:86
Method__init__
(self, class_to_ind=None, keep_difficult=False)
data/voc0712.py:44
Method__init__
(self, root, image_sets=[('2007', 'trainval'), ('2012', 'trainval')], transf
data/voc0712.py:97
Method__init__
(self,n_channels, scale)
layers/modules/l2norm.py:8
Method__init__
(self, num_classes, overlap_thresh, prior_for_matching, bkg_label, neg_mining, neg_pos, neg_o
layers/modules/multibox_loss.py:33
Method__init__
(self, num_classes, bkg_label, top_k, conf_thresh, nms_thresh)
layers/functions/detection.py:13
Method__init__
(self, cfg)
layers/functions/prior_box.py:11
Method__len__
(self)
data/voc0712.py:119
Method__repr__
(self)
data/coco.py:172
Functioncenter_size
Convert prior_boxes to (cx, cy, w, h) representation for comparison to center-size form ground truth data. Args: boxes: (tensor) poin
layers/box_utils.py:17
next →1–100 of 110, ranked by callers