MCPcopy Create free account

hub / github.com/Banconxuan/RTM3D / functions

Functions537 in github.com/Banconxuan/RTM3D

Method__init__
(self, spatial_scale, pooled_size, output_dim,
src/lib/models/networks/DCNv2/dcn_v2.py:189
Method__init__
(self, spatial_scale, pooled_size, output_dim,
src/lib/models/networks/DCNv2/dcn_v2.py:225
Method__len__
(self)
src/lib/datasets/dataset/nusceneshp.py:67
Method_get_losses
(self, opt)
src/lib/trains/car_pose.py:63
Method_make_level
(self, block, inplanes, planes, blocks, stride=1)
src/lib/models/networks/dlav0.py:260
Method_make_level
(self, block, inplanes, planes, blocks, stride=1)
src/lib/models/networks/pose_dla_dcn.py:257
Function_neg_loss
Modified focal loss. Exactly the same as CornerNet. Runs faster and costs a little bit more memory Arguments: pred (batch x c x
src/lib/models/losses.py:91
Function_not_faster_neg_loss
(pred, gt)
src/lib/models/losses.py:119
Function_read_imageset_file
(path)
src/tools/prepare_data.py:10
Function_slow_neg_loss
focal loss from CornerNet
src/lib/models/losses.py:66
Function_slow_reg_loss
(regr, gt_regr, mask)
src/lib/models/losses.py:137
Method_to_float
(self, x)
src/lib/datasets/dataset/nusceneshp.py:64
Method_to_float
(self, x)
src/lib/datasets/dataset/kittihp.py:68
Methodadd_2d_detection
( self, img, dets, show_box=False, show_txt=True, center_thresh=0.5, img_id='det')
src/lib/utils/debugger.py:512
Methodadd_bird_view
(self, dets, center_thresh=0.3, img_id='bird')
src/lib/utils/debugger.py:525
Methodadd_bird_views
(self, dets_dt, dets_gt, center_thresh=0.3, img_id='bird')
src/lib/utils/debugger.py:549
Methodadd_ct_detection
( self, img, dets, show_box=False, show_txt=True, center_thresh=0.5, img_id='det')
src/lib/utils/debugger.py:423
Methodadd_mask
(self, mask, bg, imgId = 'default', trans = 0.8)
src/lib/utils/debugger.py:120
Methodadd_points
(self, points, img_id='default')
src/lib/utils/debugger.py:348
Methodadd_rect
(self, rect1, rect2, c, conf=1, img_id='default')
src/lib/utils/debugger.py:206
Functionagnex_ct_decode
( t_heat, l_heat, b_heat, r_heat, ct_heat, t_regr=None, l_regr=None, b_regr=None, r_regr=None, K
src/lib/models/decode.py:122
Functionalpha2rot_y
Get rotation_y by alpha + theta - 180 alpha : Observation angle of object, ranging [-pi..pi] x : Object center x to the camera center (x-
src/tools/prepare_data.py:16
Methodbackward
(ctx, grad_output)
src/lib/models/networks/DCNv2/dcn_v2.py:163
Functionbox2D_sourround_box3dimimage
(box3d_in_image)
src/lib/utils/kitti_read.py:565
Functionbox2D_sourround_box3dimimage_mat
(box3d_in_image)
src/lib/utils/kitti_read.py:577
Functionbox3DOverlap
measure overlap between 3D bounding boxes, parametrized by (ry, h, w, l, tx, ty, tz)
src/tools/kitti_eval/evaluate_object_3d_offline.cpp:318
Functionbox3DOverlap
measure overlap between 3D bounding boxes, parametrized by (ry, h, w, l, tx, ty, tz)
src/tools/kitti_eval/evaluate_object_3d.cpp:318
Functionboxes3d_to_bev_torch
:param boxes3d: (N, 7) [x, y, z, h, w, l, ry] :return: boxes_bev: (N, 5) [x1, y1, x2, y2, ry]
src/lib/utils/kitti_utils_torch.py:134
Functionboxes3d_to_corners3d
:param boxes3d: (N, 7) [x, y, z, h, w, l, ry] :param rotate: :return: corners3d: (N, 8, 3)
src/lib/utils/kitti_utils_torch.py:66
Functionboxes3d_to_corners3d_torch
:param boxes3d: (N, 7) [x, y, z, h, w, l, ry] :return: corners_rotated: (N, 8, 3)
src/lib/utils/kitti_utils_torch.py:104
Functionboxes_iou3d_gpu
:param boxes_a: (N, 7) [x, y, z, h, w, l, ry] :param boxes_b: (M, 7) [x, y, z, h, w, l, ry] :return: ans_iou: (M, N)
src/lib/utils/iou3d/iou3d_utils.py:21
Functionboxes_iou_bev
:param boxes_a: (M, 5) :param boxes_b: (N, 5) :return: ans_iou: (M, N)
src/lib/utils/iou3d/iou3d_utils.py:6
Functionboxes_iou_bev
:param boxes_a: (M, 5) :param boxes_b: (N, 5) :return: ans_iou: (M, N)
src/lib/models/losses.py:19
Functionboxes_iou_bev_gpu
src/lib/utils/iou3d/src/iou3d.cpp:52
Functionboxes_overlap_bev_gpu
src/lib/utils/iou3d/src/iou3d.cpp:31
Functionbrightness_
(data_rng, image, gs, gs_mean, var)
src/lib/utils/image.py:214
Functioncalib2P
(frame_calib)
src/lib/utils/kitti_read.py:764
Methodcompose_vis_add
( self, img_path, dets, calib, center_thresh, pred, bev, img_id='out')
src/lib/utils/debugger.py:484
Functioncompute_orientation_3d
(dim, location, rotation_y)
src/lib/utils/ddd_utils.py:83
Functioncontrast_
(data_rng, image, gs, gs_mean, var)
src/lib/utils/image.py:218
Functionconv3x3
3x3 convolution with padding
src/lib/models/networks/dlav0.py:22
Functionconv3x3
3x3 convolution with padding
src/lib/models/networks/pose_dla_dcn.py:25
Functioncrop
(img, center, scale, output_size, rot=0)
src/lib/utils/image.py:84
Functionctdet_decode
(heat, wh, reg=None, cat_spec_wh=False, K=100)
src/lib/models/decode.py:464
Functionctdet_post_process
(dets, c, s, h, w, num_classes)
src/lib/utils/post_process.py:83
Functiondata_parallel
r"""Evaluates module(input) in parallel across the GPUs given in device_ids. This is the functional version of the DataParallel module. Args
src/lib/models/data_parallel.py:87
Functiondcn_v2_backward
src/lib/models/networks/DCNv2/src/dcn_v2.h:41
Functiondcn_v2_cpu_backward
src/lib/models/networks/DCNv2/src/cpu/dcn_v2_cpu.cpp:26
Functiondcn_v2_cpu_forward
src/lib/models/networks/DCNv2/src/cpu/dcn_v2_cpu.cpp:7
Functiondcn_v2_forward
src/lib/models/networks/DCNv2/src/dcn_v2.h:9
Functiondcn_v2_psroi_pooling_backward
src/lib/models/networks/DCNv2/src/dcn_v2.h:109
Functiondcn_v2_psroi_pooling_cpu_backward
src/lib/models/networks/DCNv2/src/cpu/dcn_v2_cpu.cpp:58
Functiondcn_v2_psroi_pooling_cpu_forward
src/lib/models/networks/DCNv2/src/cpu/dcn_v2_cpu.cpp:42
Functiondcn_v2_psroi_pooling_forward
src/lib/models/networks/DCNv2/src/dcn_v2.h:75
Functionddd_decode
(heat, rot, depth, dim, wh=None, reg=None, K=40)
src/lib/models/decode.py:426
Functionddd_post_process
(dets, c, s, calibs, opt)
src/lib/utils/post_process.py:75
Methoddebug
(self, batch, output, iter_id)
src/lib/trains/car_pose.py:69
Methoddebug
(self, debugger, images, dets, output, scale=1)
src/lib/detectors/car_pose.py:89
Functiondist_to_plane
Calculates the signed distance from a 3D plane to each point in a list of points :param plane: (a, b, c, d) :param points: (N, 3) :re
src/lib/utils/kitti_utils_torch.py:15
Functiondla102
(pretrained=None, **kwargs)
src/lib/models/networks/dlav0.py:378
Functiondla102x
(pretrained=None, **kwargs)
src/lib/models/networks/dlav0.py:387
Functiondla102x2
(pretrained=None, **kwargs)
src/lib/models/networks/dlav0.py:396
Functiondla169
(pretrained=None, **kwargs)
src/lib/models/networks/dlav0.py:405
Functiondla34
(pretrained, **kwargs)
src/lib/models/networks/dlav0.py:319
Functiondla34
(pretrained=True, **kwargs)
src/lib/models/networks/pose_dla_dcn.py:309
Functiondla46_c
(pretrained=None, **kwargs)
src/lib/models/networks/dlav0.py:328
Functiondla46x_c
(pretrained=None, **kwargs)
src/lib/models/networks/dlav0.py:338
Functiondla60
(pretrained=None, **kwargs)
src/lib/models/networks/dlav0.py:358
Functiondla60x
(pretrained=None, **kwargs)
src/lib/models/networks/dlav0.py:368
Functiondla60x_c
(pretrained, **kwargs)
src/lib/models/networks/dlav0.py:348
Functiondraw_Box3D
(box3d, ax, color_set='b')
src/lib/utils/vis_3d_utils.py:327
Functiondraw_Box3DinImage
(box2d, img, color_set=(255, 0, 0))
src/lib/utils/vis_3d_utils.py:341
Functiondraw_box2D
(box, image, color_set=(255, 0, 0))
src/lib/utils/vis_3d_utils.py:488
Functiondraw_ellipse2D
(param, img, color_set=(100, 0, 100))
src/lib/utils/vis_3d_utils.py:468
Functiondraw_ellipsoid3D
(ellipsodi_param, ax, color_set='r')
src/lib/utils/vis_3d_utils.py:271
Functiondraw_ellipsoid3D_fasle
(ellipsodi_param, ax, color_set='r')
src/lib/utils/vis_3d_utils.py:220
Methoddraw_features
(self,width, height, x, savename)
src/lib/models/networks/msra_resnet.py:233
Functiondraw_msra_gaussian
(heatmap, center, sigma)
src/lib/utils/image.py:175
Functiondraw_umich_gaussian
(heatmap, center, radius, k=1)
src/lib/utils/image.py:126
Functionenlarge_box3d
:param boxes3d: (N, 7) [x, y, z, h, w, l, ry]
src/lib/utils/kitti_utils_torch.py:150
Functionevaluate
(label_path, result_path, label_split_file, current_class=0,
src/tools/kitti-object-eval-python/evaluate.py:22
Functionexct_decode
( t_heat, l_heat, b_heat, r_heat, ct_heat, t_regr=None, l_regr=None, b_regr=None, r_regr=None, K
src/lib/models/decode.py:273
Functionfilter_annos_low_score
(image_annos, thresh)
src/tools/kitti-object-eval-python/kitti_common.py:190
Functionfilter_kitti_anno
(image_anno, used_classes, used_difficulty=None,
src/tools/kitti-object-eval-python/kitti_common.py:151
Functionfind_bottom_keypoint
(BOX3D_in_image,BOX3D_param,box_2D)
src/lib/utils/kitti_read.py:1100
Functionflip
(img)
src/lib/utils/image.py:16
Methodforward
(self, batch,unlabel=False,phase=None)
src/lib/trains/base_trainer.py:28
Methodforward
(self, outputs, batch,phase=None)
src/lib/trains/car_pose.py:27
Methodforward
(self, out, target)
src/lib/models/losses.py:174
Methodforward
(self, output, mask, ind, target)
src/lib/models/losses.py:190
Methodforward
(self, output, mask, ind, target)
src/lib/models/losses.py:200
Methodforward
(self, output, mask, ind, target)
src/lib/models/losses.py:213
Methodforward
(self, output, mask, ind, target,dep)
src/lib/models/losses.py:227
Methodforward
(self, output, mask, ind, target)
src/lib/models/losses.py:260
Methodforward
(self, output, mask, ind, target)
src/lib/models/losses.py:271
Methodforward
(self, output, mask, ind, rotbin, rotres)
src/lib/models/losses.py:283
Methodforward
(self, output, batch,phase=None)
src/lib/models/losses.py:312
Methodforward
(self, output, batch)
src/lib/models/losses.py:481
Methodforward
(self, output)
src/lib/models/losses.py:502
Methodforward
(self, *inputs, **kwargs)
src/lib/models/data_parallel.py:64
← previousnext →301–400 of 537, ranked by callers