MCPcopy Create free account

hub / github.com/NVIDIA-AI-IOT/Lidar_AI_Solution / functions

Functions578 in github.com/NVIDIA-AI-IOT/Lidar_AI_Solution

⨍ Functions 578◇ Types & classes 103↳ Endpoints 2
↓ 37 callersFunctionget_tensor_id
(obj)
CUDA-CenterPoint/qat/onnx_export/exptool.py:313
↓ 37 callersFunctionget_tensor_id
(obj)
CUDA-BEVFusion/qat/lean/exptool.py:327
↓ 27 callersFunctionget_tensor_id
(obj)
libraries/3DSparseConvolution/tool/centerpoint-export/exptool.py:228
↓ 27 callersFunctionget_tensor_id
(obj)
libraries/3DSparseConvolution/tool/bevfusion-export/exptool.py:252
↓ 20 callersFunctionbuild_dataset
(cfg, default_args=None)
CUDA-V2XFusion/mmdet3d/datasets/builder.py:41
↓ 17 callersMethod__init__
(self, dataloader, num_iter)
CUDA-V2XFusion/scripts/tinyq.py:365
↓ 16 callersMethodinit_quantizer
(self)
CUDA-V2XFusion/scripts/tinyq.py:508
↓ 12 callersFunctionlog
(*msg)
CUDA-V2XFusion/scripts/tinyq.py:44
↓ 11 callersMethodapply
(self, enabled=True)
CUDA-CenterPoint/qat/tools/sparseconv_quantization.py:246
↓ 9 callersMethod_conv_forward
(self, fn, x, output_size: Optional[List[int]] = None)
CUDA-V2XFusion/scripts/tinyq.py:533
↓ 9 callersFunctionregister_tensor
(obj)
CUDA-CenterPoint/qat/onnx_export/exptool.py:309
↓ 9 callersFunctionregister_tensor
(obj)
CUDA-BEVFusion/qat/lean/exptool.py:323
↓ 8 callersMethodapply
(self, fn)
CUDA-V2XFusion/scripts/tinyq.py:300
↓ 8 callersMethodapply
(self, enabled=True)
CUDA-BEVFusion/qat/lean/quantize.py:402
↓ 8 callersMethodexport
( self, img, points, camera2ego, lidar2ego, lidar2camera,
CUDA-V2XFusion/mmdet3d/models/vtransforms/base.py:531
↓ 8 callersMethodline
(self, x0, y0, x1, y1, thickness, color, interpolation)
libraries/cuOSD/test/cuosd/__init__.py:48
↓ 7 callersFunctionfformat
(fmt, vars)
libraries/YUVToRGB/compare-with-opencv.py:77
↓ 7 callersFunctionregister_tensor
(obj)
libraries/3DSparseConvolution/tool/centerpoint-export/exptool.py:224
↓ 7 callersFunctionregister_tensor
(obj)
libraries/3DSparseConvolution/tool/bevfusion-export/exptool.py:248
↓ 6 callersMethodrecompute_mask
(self)
CUDA-V2XFusion/scripts/tinyq.py:934
↓ 5 callersFunctionbuild_optimizer
Build optimizer from configs. Args: model (:obj:`nn.Module`): The model with parameters to be optimized. optimizer_cfg (dict): The
CUDA-CenterPoint/qat/det3d/torchie/apis/train.py:198
↓ 5 callersFunctionexample_to_device
(example, device=None, non_blocking=False)
CUDA-CenterPoint/qat/det3d/torchie/apis/train.py:49
↓ 5 callersFunctionfuse_bn
Given a conv Module `A` and an batch_norm module `B`, returns a conv module `C` such that C(x) == B(A(x)) in inference mode.
CUDA-CenterPoint/qat/onnx_export/funcs.py:62
↓ 5 callersFunctionismodule
(model, node, m)
CUDA-V2XFusion/scripts/tinyq.py:762
↓ 5 callersFunctionload_checkpoint
(model, file, startsname=None)
CUDA-BEVFusion/qat/lean/funcs.py:82
↓ 5 callersMethodrectangle
(self, left, top, right, bottom, thickness, border_color, bg_color)
libraries/cuOSD/test/cuosd/__init__.py:52
↓ 5 callersFunctionreplace_to_quantization_module
(model : torch.nn.Module)
CUDA-V2XFusion/scripts/quantize.py:134
↓ 5 callersFunctionreplace_to_quantization_module
(model : torch.nn.Module)
CUDA-BEVFusion/qat/lean/quantize.py:212
↓ 4 callersFunction__obj_to_id
(obj)
libraries/3DSparseConvolution/tool/centerpoint-export/exptool.py:209
↓ 4 callersFunction__obj_to_id
(obj)
libraries/3DSparseConvolution/tool/bevfusion-export/exptool.py:233
↓ 4 callersFunction__obj_to_id
(obj)
CUDA-CenterPoint/qat/onnx_export/exptool.py:294
↓ 4 callersFunction__obj_to_id
(obj)
CUDA-BEVFusion/qat/lean/exptool.py:308
↓ 4 callersFunction_replace_modules
(model: nn.Module, modules_map: list, ignore_policy: [Callable, List[str]]=None, replace_type=None)
CUDA-V2XFusion/scripts/tinyq.py:692
↓ 4 callersFunctionappend_initializer
(value, name)
CUDA-CenterPoint/qat/onnx_export/exptool.py:282
↓ 4 callersFunctionappend_initializer
(value, name)
CUDA-BEVFusion/qat/lean/exptool.py:296
↓ 4 callersMethodcircle
(self, cx, cy, radius, thickness, border_color, bg_color)
libraries/cuOSD/test/cuosd/__init__.py:51
↓ 4 callersFunctionfuse_bn
Given a conv Module `A` and an batch_norm module `B`, returns a conv module `C` such that C(x) == B(A(x)) in inference mode.
CUDA-BEVFusion/qat/lean/funcs.py:73
↓ 4 callersFunctionreplace_feature
we need to replace x.features = F.relu(x.features) with x = x.replace_feature(F.relu(x.features)) due to limit of torch.fx
CUDA-BEVFusion/qat/lean/funcs.py:97
↓ 4 callersMethodtext
(self, utf8_text, font_size, font, x, y, border_color, bg_color)
libraries/cuOSD/test/cuosd/__init__.py:46
↓ 3 callersMethod__init__
Pillar Feature Net Layer. The Pillar Feature Net could be composed of a series of these layers, but the PointPillars paper results on
CUDA-V2XFusion/mmdet3d/models/backbones/pillar_encoder.py:73
↓ 3 callersFunctionbev_pool
(feats, coords, B, D, H, W, export= False)
CUDA-V2XFusion/mmdet3d/ops/bev_pool/bev_pool.py:83
↓ 3 callersMethodbev_pool
(self, geom_feats, x, export = False)
CUDA-V2XFusion/mmdet3d/models/vtransforms/base.py:456
↓ 3 callersFunctionconnect_io_quantizer
(node, enum_args, rule_name)
CUDA-V2XFusion/scripts/tinyq.py:852
↓ 3 callersFunctiondraw
()
libraries/cuOSD/test/pytest.py:61
↓ 3 callersMethodformat_results
(self, results, img_metas, result_names=[
CUDA-V2XFusion/evaluators/det_evaluators.py:74
↓ 3 callersMethodforward
(self, x)
CUDA-V2XFusion/scripts/tinyq.py:943
↓ 3 callersMethodforward
(self, input1, input2)
CUDA-BEVFusion/qat/lean/quantize.py:70
↓ 3 callersFunctionfuse_bn
Given a conv Module `A` and an batch_norm module `B`, returns a conv module `C` such that C(x) == B(A(x)) in inference mode.
libraries/3DSparseConvolution/tool/centerpoint-export/funcs.py:62
↓ 3 callersFunctionfuse_bn
Given a conv Module `A` and an batch_norm module `B`, returns a conv module `C` such that C(x) == B(A(x)) in inference mode.
libraries/3DSparseConvolution/tool/bevfusion-export/funcs.py:72
↓ 3 callersFunctionfuse_conv_bn
(module)
CUDA-BEVFusion/qat/ptq.py:51
↓ 3 callersFunctionfuse_sparse_basic_block
(self, is_fuse_bn = False, is_fuse_relu=True)
CUDA-CenterPoint/qat/onnx_export/funcs.py:122
↓ 3 callersFunctionfuse_sparse_basic_block
(self, is_fuse_bn = False, is_fuse_relu=True)
CUDA-BEVFusion/qat/lean/funcs.py:132
↓ 3 callersMethodget_cam_feats
(self, x)
CUDA-V2XFusion/mmdet3d/models/vtransforms/base.py:452
↓ 3 callersFunctionget_lidar_3d_8points
(obj_size, yaw_lidar, center_lidar)
CUDA-V2XFusion/evaluators/result2kitti.py:40
↓ 3 callersFunctionignore_match
(ignore_policy : Union[str, List[str], Callable], path : str, target_module: nn.Module=None)
CUDA-V2XFusion/scripts/tinyq.py:675
↓ 3 callersFunctionload_model
(cfg, checkpoint_path = None, use_fuse =False, use_quant_spconv= False, use_quant_add=False)
CUDA-CenterPoint/qat/tools/centerpoint_eval.py:107
↓ 3 callersFunctionset_attr_by_array
(parent, arr)
CUDA-BEVFusion/qat/lean/funcs.py:152
↓ 3 callersFunctionset_attr_by_path
(m, path, newval)
CUDA-BEVFusion/qat/lean/funcs.py:150
↓ 3 callersMethodvoxelize
(self, points)
CUDA-V2XFusion/mmdet3d/models/fusion_models/bevfusion.py:163
↓ 2 callersMethod__enter__
(self)
CUDA-V2XFusion/scripts/tinyq.py:405
↓ 2 callersMethod__exit__
(self, exc_type, exc_value, exc_traceback)
CUDA-V2XFusion/scripts/tinyq.py:415
↓ 2 callersFunction__init__
(self)
CUDA-BEVFusion/qat/lean/quantize.py:128
↓ 2 callersMethod__init__
(self)
CUDA-CenterPoint/qat/tools/sparseconv_quantization.py:53
↓ 2 callersMethod__init__
(self, quantization = True)
CUDA-BEVFusion/qat/lean/quantize.py:61
↓ 2 callersMethod__repr__
(self)
CUDA-V2XFusion/scripts/tinyq.py:77
↓ 2 callersMethod_format_bbox
(self, results, img_metas, jsonfile_prefix=None)
CUDA-V2XFusion/evaluators/det_evaluators.py:127
↓ 2 callersMethod_format_bbox_rope3d
(self, results, jsonfile_prefix=None)
CUDA-V2XFusion/evaluators/det_evaluators.py:199
↓ 2 callersFunction_has_some_module
(model : nn.Module, module_class)
CUDA-V2XFusion/scripts/tinyq.py:747
↓ 2 callersFunctionappend_initializer
(value, name)
libraries/3DSparseConvolution/tool/centerpoint-export/exptool.py:197
↓ 2 callersFunctionappend_initializer
(value, name)
libraries/3DSparseConvolution/tool/bevfusion-export/exptool.py:221
↓ 2 callersMethodapply
(self, data0, data1, width, stride, height, format, stream, gpu_memory, launch)
libraries/cuOSD/test/cuosd/__init__.py:59
↓ 2 callersMethodapply
(self, disabled=True)
CUDA-CenterPoint/qat/tools/sparseconv_quantization.py:231
↓ 2 callersMethodapply
(self, disabled=True)
CUDA-V2XFusion/scripts/quantize.py:304
↓ 2 callersMethodapply
(self, enabled=True)
CUDA-V2XFusion/scripts/quantize.py:319
↓ 2 callersMethodapply
(self, disabled=True)
CUDA-BEVFusion/qat/lean/quantize.py:387
↓ 2 callersFunctionbbbox2bbox
(box3d, Tr_velo_to_cam, camera_intrinsic, img_size=[1920, 1080])
CUDA-V2XFusion/evaluators/result2kitti.py:176
↓ 2 callersFunctionbuild_one_cycle_optimizer
(model, optimizer_config)
CUDA-CenterPoint/qat/det3d/torchie/apis/train.py:178
↓ 2 callersMethodcompute
(self, eps=1 / (1 << 24))
CUDA-V2XFusion/scripts/tinyq.py:130
↓ 2 callersFunctionconvert_point
(point, matrix)
CUDA-V2XFusion/evaluators/result2kitti.py:109
↓ 2 callersMethoddegree2rad
(self, degree)
CUDA-V2XFusion/mmdet3d/datasets/v2x_dataset.py:365
↓ 2 callersFunctioneval_model
(model, data_loader_test, dataset_test)
CUDA-CenterPoint/qat/tools/centerpoint_eval.py:55
↓ 2 callersMethodevaluate
( self, results, metric="bbox", logger=None, **kwargs, )
CUDA-V2XFusion/mmdet3d/datasets/v2x_dataset.py:898
↓ 2 callersFunctiongen_dx_bx
(xbound, ybound, zbound)
CUDA-V2XFusion/mmdet3d/models/vtransforms/base.py:55
↓ 2 callersMethodget_bboxes
(self, preds_dicts)
CUDA-V2XFusion/scripts/export_v2xfusion.py:88
↓ 2 callersFunctionget_call_conv
(model, node)
CUDA-V2XFusion/scripts/tinyq.py:768
↓ 2 callersFunctionget_camera_3d_8points
(obj_size, yaw_lidar, center_lidar, center_in_cam, r_velo2cam, t_velo2cam)
CUDA-V2XFusion/evaluators/result2kitti.py:120
↓ 2 callersFunctionget_denorm
(sweepego2sweepsensor)
CUDA-V2XFusion/mmdet3d/datasets/v2x_dataset.py:89
↓ 2 callersMethodget_geometry
( self, camera2lidar_rots, camera2lidar_trans, intrins, post_rots,
CUDA-V2XFusion/mmdet3d/models/vtransforms/base.py:121
↓ 2 callersMethodget_geometry_rays
Transfer points from camera coord to ego coord. Args: rots(Tensor): Rotation matrix from camera to ego. trans(Tensor)
CUDA-V2XFusion/mmdet3d/models/vtransforms/base.py:406
↓ 2 callersMethodinit_sparse
(self)
CUDA-V2XFusion/scripts/tinyq.py:927
↓ 2 callersMethodinit_weights
(self)
CUDA-V2XFusion/mmdet3d/models/fusion_models/bevfusion.py:105
↓ 2 callersFunctioninverse_model
(model : nn.Module)
libraries/3DSparseConvolution/tool/bevfusion-export/exptool.py:257
↓ 2 callersFunctioninverse_model
(model : nn.Module)
CUDA-BEVFusion/qat/lean/exptool.py:332
↓ 2 callersFunctionisconv_bn_or_conv
(model, node)
CUDA-V2XFusion/scripts/tinyq.py:774
↓ 2 callersFunctionload_tensor
(file, dense_grid, detail=False)
libraries/3DSparseConvolution/tool/compare.py:47
↓ 2 callersFunctionload_tensor
(file, dense_grid, detail=False, to_dense=False)
CUDA-BEVFusion/tool/compare.py:72
↓ 2 callersFunctionmake_model_forward_hook
(self, inverse_indices=False)
libraries/3DSparseConvolution/tool/bevfusion-export/exptool.py:194
↓ 2 callersFunctionmake_model_forward_hook
(self, inverse_indices=False)
CUDA-BEVFusion/qat/lean/exptool.py:269
↓ 2 callersFunctionmerge_v2x_v2xseq
(json_path, data_folder_path, merge_folder_path, data_type, start_idx, train_dataset, val_dataset)
CUDA-V2XFusion/scripts/dataset_merge.py:47
↓ 2 callersFunctionpad
(x)
CUDA-V2XFusion/scripts/export_v2xfusion.py:279
next →1–100 of 578, ranked by callers