MCPcopy Create free account

hub / github.com/JanghoonChoi/TACT / functions

Functions104 in github.com/JanghoonChoi/TACT

↓ 10 callersFunctiondown2n
(x, n)
utils.py:105
↓ 9 callersFunction_resnet
(arch, block, layers, pretrained, progress, **kwargs)
model/resnet.py:223
↓ 5 callersFunctionimread_to_rgb
(path)
utils.py:14
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1, dilate=False)
model/resnet.py:177
↓ 4 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
model/box_utils.py:51
↓ 3 callersMethod__init__
(self, in_channels, inter_channels=None, sub_sample=True, bn_layer=True)
model/non_local.py:101
↓ 3 callersFunctionconv1x1
1x1 convolution
model/resnet.py:30
↓ 3 callersFunctionconv3x3
3x3 convolution with padding
model/resnet.py:24
↓ 3 callersMethodget_feats_xfa
(self, x, xb)
model/build_model.py:85
↓ 3 callersMethodnormalize_tensor
(self, x, mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
model/build_model.py:30
↓ 3 callersFunctiontpz
(xin, t0,t1,t2,t3)
utils.py:116
↓ 2 callersMethod__init__
(self, block, layers, num_classes=1000, zero_init_residual=False, groups=1, width_per_group=6
model/resnet.py:126
↓ 2 callersFunctionbox_overlap_area
(A,B)
track_utils.py:42
↓ 2 callersFunctionbox_overlap_score
(A,B)
track_utils.py:64
↓ 2 callersMethodfmap_corr
(self,xf,yf,pad=True,dilate=1)
model/rpn_module.py:46
↓ 2 callersMethodforward_box
(self, x,y, xb, xfa=None, add_box=None, nbox=1)
model/build_model.py:55
↓ 2 callersMethodread_img_resize
(self, imf)
tracker.py:47
↓ 2 callersFunctionresult_curve
(result, num_points)
track_utils.py:7
↓ 2 callersFunctionth_rand_rng
pull uniform random sample(s) from [a,b)
th_utils.py:70
↓ 2 callersFunctionunit_rnd
(r)
th_utils.py:80
↓ 1 callersMethod__init__
(self, cfg)
model/fcos.py:63
↓ 1 callersMethod_forward_impl
(self, x)
model/resnet.py:201
↓ 1 callersMethodcorr_head
(self, xfa, yf)
model/rpn_module.py:73
↓ 1 callersMethoddconv_fmap_corr
(self, yf, xf)
model/rpn_module.py:62
↓ 1 callersMethoddraw_vid_seq
(self, seq_res, seq_name)
tracker_batch.py:200
↓ 1 callersMethoddraw_vid_seq
(self, seq_res, seq_name)
tracker.py:154
↓ 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
model/box_utils.py:116
↓ 1 callersMethodforward
(self, x,y, xb, xfa=None, add_box=None)
model/build_model.py:35
↓ 1 callersFunctiongenerate_reg_coords
(cfg)
th_utils.py:103
↓ 1 callersMethodget_cl_loss
(self, cl, gt)
model/loss.py:24
↓ 1 callersMethodget_rcnn_loss
(self, cf, op, bb, br, gb)
model/loss.py:62
↓ 1 callersMethodget_re_loss
(self, re, gr, gt, eps=1e-7)
model/loss.py:37
↓ 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
model/box_utils.py:30
↓ 1 callersMethodload_track_db
(self, name)
tracker_batch.py:74
↓ 1 callersMethodload_track_db
(self, name)
tracker.py:39
↓ 1 callersFunctionnms
Apply non-maximum suppression at test time to avoid detecting too many overlapping bounding boxes for a given object. Args: boxes: (te
model/box_utils.py:176
↓ 1 callersFunctionpoint_form
Convert prior_boxes to (xmin, ymin, xmax, ymax) representation for comparison to point form ground truth data. Args: boxes: (tensor)
model/box_utils.py:6
↓ 1 callersMethodpool_feat
(self, xf, xb_p)
model/rpn_module.py:86
↓ 1 callersMethodread_img_resize
(self, imf)
tracker_batch.py:30
↓ 1 callersMethodread_img_resize
(self, imf)
tracker_batch.py:82
↓ 1 callersFunctionrun_eval
(idx=-1)
test_tracker.py:20
↓ 1 callersMethodrun_track_db
(self, seq_list=None, out_vid=False, calc_auc=True, save_res=False)
tracker.py:116
↓ 1 callersMethodrun_track_seq
(self, seq_name, seq_path, seq_imlist, seq_gt, save_res=False)
tracker_batch.py:96
↓ 1 callersMethodrun_track_seq
(self, seq_name, save_res=False)
tracker.py:61
↓ 1 callersFunctionth_rand
proxy to torch.rand(n)[0]
th_utils.py:62
Method__getitem__
(self, idx)
tracker_batch.py:43
Method__init__
(self, cfg, seq_path, seq_imlist)
tracker_batch.py:21
Method__init__
(self, cfg, db_name=None, idx=-1)
tracker_batch.py:53
Method__init__
(self, cfg, db_name=None, idx=-1)
tracker.py:18
Method__init__
(self,cfg)
model/context.py:13
Method__init__
(self, cfg)
model/boxes.py:11
Method__init__
(self, cfg)
model/loss.py:10
Method__init__
(self, cfg)
model/fcos.py:10
Method__init__
(self,cfg)
model/build_model.py:12
Method__init__
(self,cfg,bb_ch)
model/rpn_module.py:14
Method__init__
(self,cfg)
model/rcnn_module.py:11
Method__init__
:param in_channels: :param inter_channels: :param dimension: :param sub_sample: :param bn_layer:
model/non_local.py:8
Method__init__
(self, in_channels, inter_channels=None, sub_sample=True, bn_layer=True)
model/non_local.py:109
Method__init__
(self, in_channels, inter_channels=None, sub_sample=True, bn_layer=True)
model/non_local.py:117
Method__init__
(self, fdim, reduce=16, use_bias=True)
model/cbam.py:9
Method__init__
(self, gamma=0, alpha=None, size_average=True)
model/focal_loss.py:9
Method__init__
(self, inplanes, planes, stride=1, downsample=None, groups=1, base_width=64, dilation=1, norm
model/resnet.py:38
Method__init__
(self, inplanes, planes, stride=1, downsample=None, groups=1, base_width=64, dilation=1, norm
model/resnet.py:84
Method__len__
(self)
tracker_batch.py:27
Functioncenter_size
Convert prior_boxes to (cx, cy, w, h) representation for comparison to center-size form ground truth data. Args: boxes: (tensor) poin
model/box_utils.py:18
Methodclean_temp_dir
(self, temp_dir='.temp/')
tracker_batch.py:232
Methodclean_temp_dir
(self, temp_dir='.temp/')
tracker.py:186
Functioncrop_img
(I, x, y, w, h, center=False, mfill=False)
utils.py:18
Functiondecode
Decode locations from predictions using priors to undo the encoding we did for offset regression at train time. Args: loc (tensor): lo
model/box_utils.py:141
Functiondist_succ
(v_pred, v_gt, batch_size)
utils.py:83
Functiondraw_bb_img
(img0, x_min,y_min,width,height, color, stroke)
utils.py:62
Methodforward
(self, cl,re, cf,yfa,ybb)
model/context.py:78
Methodforward
(self,cl,re, nms_param=None)
model/boxes.py:21
Methodforward
(self, outs, gts, add_rcnn_loss=True)
model/loss.py:124
Methodforward
(self, x)
model/fcos.py:47
Methodforward
(self, x, out_re=True)
model/fcos.py:97
Methodforward
(self,xf_in,yf_in, xb, xfa_in=None, add_box=None, pool_xf=False)
model/rpn_module.py:107
Methodforward
(self, feats, boxes)
model/rcnn_module.py:32
Methodforward
:param x: (b, c, t, h, w) :param return_nl_map: if True return z, nl_map, else only return z. :return:
model/non_local.py:71
Methodforward
(self, x)
model/cbam.py:29
Methodforward
(self, input, target)
model/focal_loss.py:17
Methodforward
(self, x)
model/resnet.py:56
Methodforward
(self, x)
model/resnet.py:101
Methodforward
(self, x)
model/resnet.py:219
Functionget_dtstr
(sec=True)
utils.py:6
Functiongray2jet
(I)
utils.py:113
Functionlog_sum_exp
Utility function for computing log_sum_exp while determining This will be used to determine unaveraged confidence loss across all examples in
model/box_utils.py:162
Functionmatch
Match each prior box with the ground truth box of the highest jaccard overlap, encode the bounding boxes, then return the matched indices corr
model/box_utils.py:72
Functionresnet101
r"""ResNet-101 model from `"Deep Residual Learning for Image Recognition" <https://arxiv.org/pdf/1512.03385.pdf>`_ Args: pretrained (
model/resnet.py:268
Functionresnet152
r"""ResNet-152 model from `"Deep Residual Learning for Image Recognition" <https://arxiv.org/pdf/1512.03385.pdf>`_ Args: pretrained (
model/resnet.py:280
Functionresnet18
r"""ResNet-18 model from `"Deep Residual Learning for Image Recognition" <https://arxiv.org/pdf/1512.03385.pdf>`_ Args: pretrained (b
model/resnet.py:232
Functionresnet34
r"""ResNet-34 model from `"Deep Residual Learning for Image Recognition" <https://arxiv.org/pdf/1512.03385.pdf>`_ Args: pretrained (b
model/resnet.py:244
Functionresnet50
r"""ResNet-50 model from `"Deep Residual Learning for Image Recognition" <https://arxiv.org/pdf/1512.03385.pdf>`_ Args: pretrained (b
model/resnet.py:256
Functionresnext101_32x8d
r"""ResNeXt-101 32x8d model from `"Aggregated Residual Transformation for Deep Neural Networks" <https://arxiv.org/pdf/1611.05431.pdf>`_ Args
model/resnet.py:306
Functionresnext50_32x4d
r"""ResNeXt-50 32x4d model from `"Aggregated Residual Transformation for Deep Neural Networks" <https://arxiv.org/pdf/1611.05431.pdf>`_ Args:
model/resnet.py:292
Functionresult_curve_px
(result, num_points)
track_utils.py:24
Methodrun_track_db
(self, seq_list=None, out_vid=False, calc_auc=True, save_res=False)
tracker_batch.py:157
Functionth_choice
torch implementation of np.random.choice(), x1.1~1.5 slower than original function
th_utils.py:8
Functionth_choice_mul
(a, n)
th_utils.py:30
Functionth_choice_seq
(a, n)
th_utils.py:44
next →1–100 of 104, ranked by callers