Functions — github.com/NVlabs/CenterPose

CodeHub MCPcopy Create free account

hub / github.com/NVlabs/CenterPose / functions

Functions615 in github.com/NVlabs/CenterPose

⨍ Functions 615◇ Types & classes 95

↓ 1 callersMethodcompute_ray

Computes a ray from camera to box centroid in box frame. For vertex in camera frame V^c, and object unit frame V^o, we have R * Vc

src/tools/objectron_eval/eval_image_official.py:864

↓ 1 callersMethodcompute_ray

Computes a ray from camera to box centroid in box frame. For vertex in camera frame V^c, and object unit frame V^o, we have R * Vc

src/tools/objectron_eval/objectron/dataset/eval.py:212

↓ 1 callersFunctioncompute_rot_loss

(output, target_bin, target_res, mask)

src/lib/models/losses.py:290

↓ 1 callersMethodcompute_scale

Computes scale of the given box sitting on the plane.

src/tools/objectron_eval/eval_video_official.py:1145

↓ 1 callersMethodcompute_scale

Computes scale of the given box sitting on the plane.

src/tools/objectron_eval/eval_image_official.py:795

↓ 1 callersFunctionconv_identify

src/lib/models/networks/DCNv2/testcpu.py:20

↓ 1 callersFunctionconv_identify

src/lib/models/networks/DCNv2/testcuda.py:20

↓ 1 callersFunctiondcn_v2_cpu_backward

src/lib/models/networks/DCNv2/src/cpu/dcn_v2_cpu.cpp:109

↓ 1 callersFunctiondcn_v2_cpu_forward

src/lib/models/networks/DCNv2/src/cpu/dcn_v2_cpu.cpp:17

↓ 1 callersFunctiondcn_v2_psroi_pooling_cpu_backward

src/lib/models/networks/DCNv2/src/cpu/dcn_v2_psroi_pooling_cpu.cpp:350

↓ 1 callersFunctiondcn_v2_psroi_pooling_cpu_forward

src/lib/models/networks/DCNv2/src/cpu/dcn_v2_psroi_pooling_cpu.cpp:278

↓ 1 callersMethoddebug

(self, image_src, num_instances, instances_2d, instances_3d, projection_matrix, boxes, instances_scale,

src/tools/objectron_eval/eval_video_official.py:894

↓ 1 callersMethoddebug

(self, image_src, num_instances, instances_2d, instances_3d, projection_matrix, boxes, instances_scale,

src/tools/objectron_eval/eval_image_official.py:574

↓ 1 callersMethoddebug

(self, debugger, images, dets, output, scale=1, pre_hms=None, pre_hm_hp=None)

src/lib/detectors/base_detector.py:71

↓ 1 callersFunctiondemo

↓ 1 callersFunctiondmcn_get_coordinate_weight_cpu

src/lib/models/networks/DCNv2/src/cpu/dcn_v2_im2col_cpu.cpp:84

↓ 1 callersFunctiondmcn_get_gradient_weight_cpu

src/lib/models/networks/DCNv2/src/cpu/dcn_v2_im2col_cpu.cpp:58

↓ 1 callersFunctiondraw_dense_reg

(regmap, heatmap, center, value, radius, is_offset=False)

src/lib/utils/image.py:181

↓ 1 callersMethodevaluate_2d

Evaluates a pair of 2D projections of 3D boxes. It computes the mean normalized distances of eight vertices of a box. Args: box: A 9*2

data/objectron/dataset/eval.py:171

↓ 1 callersMethodevaluate_2d

Evaluates a pair of 2D projections of 3D boxes. It computes the mean normalized distances of eight vertices of a box. Args:

src/tools/objectron_eval/objectron/dataset/eval.py:171

↓ 1 callersMethodevaluate_3d

Evaluates a box in 3D. It computes metrics of view angle and 3D IoU. Args: box: A predicted box. instance: A 9*3 array of an ann

data/objectron/dataset/eval.py:187

↓ 1 callersMethodevaluate_3d

Evaluates a box in 3D. It computes metrics of view angle and 3D IoU. Args: box: A predicted box. instance: A 9*3

src/tools/objectron_eval/objectron/dataset/eval.py:187

↓ 1 callersMethodevaluate_consistency

src/tools/objectron_eval/eval_video_official.py:811

↓ 1 callersMethodevaluate_iou

Evaluates a 3D box by 3D IoU. It computes 3D IoU of predicted and annotated boxes. Args: box: A 9*3 array of a predicted box. in

data/objectron/dataset/eval.py:367

↓ 1 callersMethodevaluate_iou

Evaluates a 3D box by 3D IoU. It computes 3D IoU of predicted and annotated boxes. Args: box: A 9*3 array of a predicted b

src/tools/objectron_eval/eval_image_official.py:1051

↓ 1 callersMethodevaluate_iou

Evaluates a 3D box by 3D IoU. It computes 3D IoU of predicted and annotated boxes. Args: box: A 9*3 array of a predicted b

src/tools/objectron_eval/objectron/dataset/eval.py:366

↓ 1 callersMethodevaluate_scale

(self, relative_scale, instance)

src/tools/objectron_eval/eval_video_official.py:1008

↓ 1 callersMethodevaluate_scale

(self, relative_scale, instance)

src/tools/objectron_eval/eval_image_official.py:664

↓ 1 callersMethodevaluate_viewpoint

Evaluates a 3D box by viewpoint. Args: box: A 9*3 array of a predicted box. instance: A 9*3 array of an annotated box, in metric leve

data/objectron/dataset/eval.py:288

↓ 1 callersMethodevaluate_viewpoint

Evaluates a 3D box by viewpoint. Args: box: A 9*3 array of a predicted box. instance: A 9*3 array of an annotated box, in

src/tools/objectron_eval/objectron/dataset/eval.py:287

↓ 1 callersFunctionexample_dconv

src/lib/models/networks/DCNv2/testcpu.py:169

↓ 1 callersFunctionexample_dconv

src/lib/models/networks/DCNv2/testcuda.py:169

↓ 1 callersFunctionexample_dpooling

src/lib/models/networks/DCNv2/testcpu.py:183

↓ 1 callersFunctionexample_dpooling

src/lib/models/networks/DCNv2/testcuda.py:183

↓ 1 callersFunctionexample_mdpooling

src/lib/models/networks/DCNv2/testcpu.py:226

↓ 1 callersFunctionexample_mdpooling

src/lib/models/networks/DCNv2/testcuda.py:226

↓ 1 callersFunctionexport_to_ndds_file

( frame=None, filename = "tmp.json", #this has to include path as well sequence=None, frame_id

data/utils.py:60

↓ 1 callersFunctionfill_up_weights

src/lib/models/networks/dlav0.py:457

↓ 1 callersFunctionfill_up_weights

src/lib/models/networks/resnet_dcn.py:112

↓ 1 callersFunctionfill_up_weights

src/lib/models/networks/pose_dla_dcn.py:365

↓ 1 callersMethodfinalize

Computes average precision curves.

data/objectron/dataset/eval.py:458

↓ 1 callersMethodfinalize

Computes average precision curves.

src/tools/objectron_eval/group_report_new.py:79

↓ 1 callersMethodfinalize

Computes average precision curves.

src/tools/objectron_eval/eval_image_official.py:1171

↓ 1 callersMethodfinalize

Computes average precision curves.

src/tools/objectron_eval/group_report_old.py:73

↓ 1 callersMethodfinalize

Computes average precision curves.

src/tools/objectron_eval/objectron/dataset/eval.py:457

↓ 1 callersFunctionfitgaussian

Returns (height, x, y, width_x, width_y) the gaussian parameters of a 2D distribution found by a fit

src/lib/utils/gpfit.py:29

↓ 1 callersMethodfrom_transformation

Constructs an oriented bounding box from transformation and scale.

data/objectron/dataset/box.py:59

↓ 1 callersMethodfrom_transformation

Constructs an oriented bounding box from transformation and scale.

src/tools/objectron_eval/objectron/dataset/box.py:63

↓ 1 callersMethodgenerate_vertexes

src/lib/utils/pnp/cuboid_objectron.py:80

↓ 1 callersFunctionget_dir

(src_point, rot_rad)

src/lib/utils/image.py:82

↓ 1 callersFunctionget_extensions

src/lib/models/networks/DCNv2/setup.py:18

↓ 1 callersMethodget_image

(self, feature, shape=None)

data/objectron/dataset/parser.py:95

↓ 1 callersMethodget_image

(self, feature, shape=None)

src/tools/objectron_eval/objectron/dataset/parser.py:230

↓ 1 callersFunctionget_model_url

(data='imagenet', name='dla34', hash='ba72cf86')

src/lib/models/networks/dlav0.py:19

↓ 1 callersFunctionget_model_url

(data='imagenet', name='dla34', hash='ba72cf86')

src/lib/models/networks/pose_dla_dcn.py:24

↓ 1 callersFunctiongrab_frame

Grab an image frame from the video file.

data/preprocess.py:33

↓ 1 callersFunctiongrayscale

src/lib/utils/image.py:239

↓ 1 callersFunctiongreedy_assignment

src/lib/utils/tracker_baseline.py:301

↓ 1 callersFunctiongreedy_assignment

src/lib/utils/tracker.py:305

↓ 1 callersFunctiongroup_norm

src/lib/models/networks/GN.py:4

↓ 1 callersFunctiongroup_video

(ds, video_ids_split)

src/tools/objectron_eval/prepare_test_video.py:69

↓ 1 callersMethodinit_hidden

(self, batch_size, hidden, shape, device)

src/lib/models/networks/convGRU.py:41

↓ 1 callersMethodinit_offset

src/lib/models/networks/DCNv2/dcn_v2.py:114

↓ 1 callersMethodinit_weights

(self, num_layers, pretrained=True)

src/lib/models/networks/msra_resnet.py:260

↓ 1 callersMethodinit_weights

(self, num_layers)

src/lib/models/networks/resnet_dcn.py:269

↓ 1 callersMethodintersect_box_poly

Clips the polygon against the faces of the axis-aligned box.

data/objectron/dataset/iou.py:93

↓ 1 callersMethodintersect_box_poly

Clips the polygon against the faces of the axis-aligned box.

src/tools/objectron_eval/objectron/dataset/iou.py:96

↓ 1 callersMethodiou

Computes the exact IoU using Sutherland-Hodgman algorithm.

data/objectron/dataset/iou.py:22

↓ 1 callersFunctionis_visible

Determines if a 2D point is visible.

src/lib/utils/pnp/cuboid_pnp_shell.py:74

↓ 1 callersFunctionlighting_

(data_rng, image, alphastd, eigval, eigvec)

src/lib/utils/image.py:243

↓ 1 callersMethodload_pretrained_model

(self, data='imagenet', name='dla34', hash='ba72cf86')

src/lib/models/networks/pose_dla_dcn.py:324

↓ 1 callersFunctionmain

src/main_CenterPoseTrack.py:25

↓ 1 callersFunctionmain

↓ 1 callersFunctionmain

src/main_CenterPose.py:25

↓ 1 callersFunctionmain

src/tools/objectron_eval/eval_video_official.py:1571

↓ 1 callersFunctionmain

src/tools/objectron_eval/eval_image_official.py:1186

↓ 1 callersFunctionmake_cnv_layer

(inp_dim, out_dim)

src/lib/models/networks/large_hourglass.py:126

↓ 1 callersFunctionmake_hg_layer

(kernel, dim0, dim1, mod, layer=convolution, **kwargs)

src/lib/models/networks/large_hourglass.py:290

↓ 1 callersFunctionmake_inter_layer

src/lib/models/networks/large_hourglass.py:122

↓ 1 callersFunctionmake_merge_layer

src/lib/models/networks/large_hourglass.py:100

↓ 1 callersFunctionmake_pool_layer

src/lib/models/networks/large_hourglass.py:107

↓ 1 callersFunctionmake_unpool_layer

src/lib/models/networks/large_hourglass.py:111

↓ 1 callersMethodmatch_box

Matches a detected box with annotated instances. For a predicted box, finds the nearest annotation in instances. This means we always assume

data/objectron/dataset/eval.py:387

↓ 1 callersMethodmatch_box

Matches a detected box with annotated instances. For a predicted box, finds the nearest annotation in instances. This means we always

src/tools/objectron_eval/eval_video_official.py:1428

↓ 1 callersMethodmatch_box

Matches a detected box with annotated instances. For a predicted box, finds the nearest annotation in instances. This means we always

src/tools/objectron_eval/eval_image_official.py:1076

↓ 1 callersMethodmatch_box

Matches a detected box with annotated instances. For a predicted box, finds the nearest annotation in instances. This means we always

src/tools/objectron_eval/objectron/dataset/eval.py:386

↓ 1 callersMethodmerge_outputs

(self, detections)

src/lib/detectors/base_detector.py:68

↓ 1 callersFunctionmodulated_deformable_col2im_coord_cpu

src/lib/models/networks/DCNv2/src/cpu/dcn_v2_im2col_cpu.cpp:375

↓ 1 callersFunctionmodulated_deformable_col2im_coord_cpu_kernel

src/lib/models/networks/DCNv2/src/cpu/dcn_v2_im2col_cpu.cpp:259

↓ 1 callersFunctionmodulated_deformable_col2im_cpu

src/lib/models/networks/DCNv2/src/cpu/dcn_v2_im2col_cpu.cpp:353

↓ 1 callersFunctionmodulated_deformable_col2im_cpu_kernel

src/lib/models/networks/DCNv2/src/cpu/dcn_v2_im2col_cpu.cpp:198

↓ 1 callersFunctionmodulated_deformable_im2col_cpu_kernel

src/lib/models/networks/DCNv2/src/cpu/dcn_v2_im2col_cpu.cpp:127

↓ 1 callersFunctionmoments

Returns (height, x, y, width_x, width_y) the gaussian parameters of a 2D distribution by calculating its moments

src/lib/utils/gpfit.py:13

↓ 1 callersFunctionobject_pose_post_process

(dets, c, s, h, w, opt, Inference=False)

src/lib/utils/post_process.py:12

↓ 1 callersMethodparallel_apply

(self, replicas, inputs, kwargs)

src/lib/models/data_parallel.py:80

↓ 1 callersMethodparse_camera

Parses camera from a tensorflow example.

data/objectron/dataset/parser.py:67

↓ 1 callersMethodparse_example_basic

Parses image and label from a tf.Example proto. Args: example: A tf.Example proto. Returns: video_id

src/tools/objectron_eval/objectron/dataset/parser.py:77

↓ 1 callersMethodparse_plane

Parses plane from a tensorflow example.

data/objectron/dataset/parser.py:83

↓ 1 callersFunctionpartition

src/tools/objectron_eval/prepare_test_video.py:62

↓ 1 callersMethodpost_process

(self, dets, meta, scale=1)

src/lib/detectors/base_detector.py:65

← previous next →201–300 of 615, ranked by callers