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
↓ 2 callersFunctionpcd_vis
(boxes, save_file="demo.jpg", label_path=None, Tr_velo_to_cam=None)
CUDA-V2XFusion/evaluators/result2kitti.py:145
↓ 2 callersMethodper_tensor
(bits:int, algo:Algorithm)
CUDA-V2XFusion/scripts/tinyq.py:97
↓ 2 callersMethodpoint
(self, cx, cy, radius, color)
libraries/cuOSD/test/cuosd/__init__.py:50
↓ 2 callersFunctionread_json
(path)
CUDA-V2XFusion/evaluators/result2kitti.py:194
↓ 2 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
libraries/3DSparseConvolution/tool/bevfusion-export/funcs.py:96
↓ 2 callersMethodrotationbox
(self, cx, cy, width, height, yaw, thickness, border_color, interpolation, bg_color)
libraries/cuOSD/test/cuosd/__init__.py:54
↓ 2 callersFunctionwrite_kitti_in_txt
(pred_lines, path_txt)
CUDA-V2XFusion/evaluators/result2kitti.py:95
↓ 1 callersFunction__init__
(self)
CUDA-CenterPoint/qat/tools/sparseconv_quantization.py:140
↓ 1 callersFunction__init__
(self)
CUDA-V2XFusion/scripts/quantize.py:114
↓ 1 callersMethod__init__
( self, in_channels: int, out_channels: int, image_size: Tuple[int, int],
CUDA-V2XFusion/mmdet3d/models/vtransforms/base.py:337
↓ 1 callersMethod__init__
( self, in_channels: int, out_channels: int, image_size: Tuple[int, int],
CUDA-V2XFusion/mmdet3d/models/vtransforms/lss.py:36
↓ 1 callersMethod__init__
(self, model)
CUDA-V2XFusion/scripts/export_v2xfusion.py:144
↓ 1 callersMethod__init__
(self, quantization = True)
CUDA-V2XFusion/scripts/quantize.py:62
↓ 1 callersMethod__init__
(self, parent)
CUDA-BEVFusion/qat/export-transfuser.py:226
↓ 1 callersMethod_get_sample_indices
Load annotations from ann_file. Args: ann_file (str): Path of the annotation file. Returns: list[dict]: List
CUDA-V2XFusion/mmdet3d/datasets/v2x_dataset.py:331
↓ 1 callersMethod_quant
(self, input, add_input)
CUDA-CenterPoint/qat/tools/sparseconv_quantization.py:116
↓ 1 callersMethod_set_group_flag
Set flag according to image aspect ratio. Images with aspect ratio greater than 1 will be set as group 1, otherwise group 0. In 3D da
CUDA-V2XFusion/mmdet3d/datasets/v2x_dataset.py:322
↓ 1 callersFunctionallow_sparsity_with_best_practice
(name, weight_shape)
CUDA-V2XFusion/scripts/tinyq.py:991
↓ 1 callersFunctionarg_parser
()
libraries/3DSparseConvolution/tool/export-helper.py:35
↓ 1 callersFunctionarg_parser
()
CUDA-CenterPoint/tool/eval_nusc.py:35
↓ 1 callersFunctionarg_parser
()
CUDA-CenterPoint/tool/export_neck_head.py:46
↓ 1 callersFunctionarg_parser
()
CUDA-BEVFusion/tool/dump-data.py:39
↓ 1 callersFunctionbatch_processor
(model, data, train_mode, return_preds=False, **kwargs)
CUDA-CenterPoint/qat/det3d/torchie/apis/train.py:113
↓ 1 callersMethodbev_pool
(self, geom_feats, x)
CUDA-V2XFusion/mmdet3d/models/vtransforms/base.py:170
↓ 1 callersFunctionbev_transform
(gt_boxes, rotate_angle, scale_ratio, flip_dx, flip_dy)
CUDA-V2XFusion/mmdet3d/datasets/v2x_dataset.py:169
↓ 1 callersFunctionblur_test
()
libraries/cuOSD/test/pytest.py:115
↓ 1 callersMethodboxblur
(self, left, top, right, bottom, kernel_size)
libraries/cuOSD/test/cuosd/__init__.py:53
↓ 1 callersFunctioncenterpoint_qat_test
(nuScense_config, weight_path, use_quantization)
CUDA-CenterPoint/qat/tools/centerpoint_eval.py:123
↓ 1 callersMethodchoose_cams
Choose cameras randomly. Returns: list: Cameras to be used.
CUDA-V2XFusion/mmdet3d/datasets/v2x_dataset.py:753
↓ 1 callersMethodcollect
(self, x)
CUDA-V2XFusion/scripts/tinyq.py:113
↓ 1 callersFunctioncollect_stats
Feed data to the network and collect statistics
CUDA-CenterPoint/qat/tools/sparseconv_quantization.py:178
↓ 1 callersFunctioncollect_stats
Feed data to the network and collect statistics
CUDA-V2XFusion/scripts/quantize.py:251
↓ 1 callersFunctioncollect_stats
Feed data to the network and collect statistics
CUDA-BEVFusion/qat/lean/quantize.py:334
↓ 1 callersFunctioncompare_and_print
(cpp_file, torch_file, detail=False, dense_grid=None)
libraries/3DSparseConvolution/tool/compare.py:71
↓ 1 callersFunctioncompare_and_print
(cpp_file, torch_file, detail=False, dense_grid=None, to_dense=False, do_sort=False)
CUDA-BEVFusion/tool/compare.py:99
↓ 1 callersFunctioncompute_amax
(model, **kwargs)
CUDA-CenterPoint/qat/tools/sparseconv_quantization.py:167
↓ 1 callersFunctioncompute_amax
(model, **kwargs)
CUDA-V2XFusion/scripts/quantize.py:240
↓ 1 callersFunctioncompute_amax
(model, **kwargs)
CUDA-BEVFusion/qat/lean/quantize.py:323
↓ 1 callersMethodconvert
(old: nn.LayerNorm)
CUDA-BEVFusion/qat/export-transfuser.py:259
↓ 1 callersMethodcreate_frustum
(self)
CUDA-V2XFusion/mmdet3d/models/vtransforms/base.py:95
↓ 1 callersMethodcreate_frustum_rays
Generate frustum
CUDA-V2XFusion/mmdet3d/models/vtransforms/base.py:372
↓ 1 callersFunctioncuosd_draw
(image)
libraries/cuOSD/test/pytest.py:53
↓ 1 callersFunctiondepth_transform
Transform depth based on ida augmentation configuration. Args: cam_depth (np array): Nx3, 3: x,y,d. resize (float): Resize factor
CUDA-V2XFusion/mmdet3d/datasets/v2x_dataset.py:196
↓ 1 callersFunctiondisable_quant_layer
(model_finetune)
CUDA-CenterPoint/qat/tools/centerpoint_qat_pipline.py:76
↓ 1 callersFunctiondisable_quant_layer
(model_finetune)
CUDA-CenterPoint/qat/tools/centerpoint_eval.py:97
↓ 1 callersFunctiondump_bin
(args)
CUDA-CenterPoint/tool/eval_nusc.py:47
↓ 1 callersFunctiondump_tensor
(args)
CUDA-BEVFusion/tool/dump-data.py:46
↓ 1 callersFunctionequation_plane
(points)
CUDA-V2XFusion/mmdet3d/datasets/v2x_dataset.py:73
↓ 1 callersFunctioneval_metrics
(args)
CUDA-CenterPoint/tool/eval_nusc.py:77
↓ 1 callersMethodextract_camera_features
( self, x, points, camera2ego, lidar2ego, lidar2camera,
CUDA-V2XFusion/mmdet3d/models/fusion_models/bevfusion.py:109
↓ 1 callersMethodextract_lidar_features
(self, x)
CUDA-V2XFusion/mmdet3d/models/fusion_models/bevfusion.py:155
↓ 1 callersFunctionfill_by_input_format
(task:roiconv.Task, input_format:roiconv.InputFormat, vars:dict)
libraries/roiconvert/unitests/test_all.py:43
↓ 1 callersFunctionflatten_model
(m)
CUDA-CenterPoint/qat/det3d/torchie/apis/train.py:170
↓ 1 callersMethodforward
(self, features)
libraries/3DSparseConvolution/tool/bevfusion-export/export-transfuser.py:217
↓ 1 callersMethodforward
(self, *args, **kwargs)
CUDA-V2XFusion/mmdet3d/models/vtransforms/lss.py:96
↓ 1 callersMethodforward_single
( self, img, points, camera2ego, lidar2ego, lidar2camera,
CUDA-V2XFusion/mmdet3d/models/fusion_models/bevfusion.py:233
↓ 1 callersFunctionfuse_bn_weights
(conv_w_OKI, conv_b, bn_rm, bn_rv, bn_eps, bn_w, bn_b)
libraries/3DSparseConvolution/tool/centerpoint-export/funcs.py:43
↓ 1 callersFunctionfuse_bn_weights
(conv_w_OKI, conv_b, bn_rm, bn_rv, bn_eps, bn_w, bn_b)
libraries/3DSparseConvolution/tool/bevfusion-export/funcs.py:53
↓ 1 callersFunctionfuse_bn_weights
(conv_w_OKI, conv_b, bn_rm, bn_rv, bn_eps, bn_w, bn_b)
CUDA-CenterPoint/qat/onnx_export/funcs.py:43
↓ 1 callersFunctionfuse_bn_weights
(conv_w_OKI, conv_b, bn_rm, bn_rv, bn_eps, bn_w, bn_b)
CUDA-BEVFusion/qat/lean/funcs.py:54
↓ 1 callersFunctionfuse_sparse_basic_block
(self)
libraries/3DSparseConvolution/tool/centerpoint-export/funcs.py:101
↓ 1 callersFunctionfuse_sparse_basic_block
(self)
libraries/3DSparseConvolution/tool/bevfusion-export/funcs.py:119
↓ 1 callersMethodget_M
(self, R, K, R_r, K_r)
CUDA-V2XFusion/mmdet3d/datasets/v2x_dataset.py:368
↓ 1 callersMethodget_bboxes
Generate bboxes from bbox head predictions. Args: preds_dicts (tuple[list[dict]]): Prediction results. Returns:
libraries/3DSparseConvolution/tool/bevfusion-export/export-transfuser.py:184
↓ 1 callersMethodget_bboxes
Generate bboxes from bbox head predictions. Args: preds_dicts (tuple[list[dict]]): Prediction results. Returns:
CUDA-BEVFusion/qat/export-transfuser.py:184
↓ 1 callersFunctionget_cam_calib_intrinsic
(calib_path)
CUDA-V2XFusion/evaluators/result2kitti.py:213
↓ 1 callersMethodget_cam_feats
(self, x)
CUDA-V2XFusion/mmdet3d/models/vtransforms/base.py:166
↓ 1 callersFunctionget_cv_image
(iwidth, iheight, owidth, oheight, interp, format)
libraries/YUVToRGB/compare-with-opencv.py:36
↓ 1 callersMethodget_gt
Generate gt labels from info. Args: info(dict): Infos needed to generate gt labels. cams(list): Camera names.
CUDA-V2XFusion/mmdet3d/datasets/v2x_dataset.py:706
↓ 1 callersMethodget_image
Given data and cam_names, return image data needed. Args: sweeps_data (list): Raw data used to generate the data we needed.
CUDA-V2XFusion/mmdet3d/datasets/v2x_dataset.py:521
↓ 1 callersFunctionget_layer_groups
(m)
CUDA-CenterPoint/qat/det3d/torchie/apis/train.py:174
↓ 1 callersFunctionget_lidar2cam
(calib_path)
CUDA-V2XFusion/evaluators/result2kitti.py:199
↓ 1 callersFunctionget_paddings_indicator
Create boolean mask by actually number of a padded tensor. Args: actual_num ([type]): [description] max_num ([type]): [description
CUDA-V2XFusion/mmdet3d/models/backbones/pillar_encoder.py:49
↓ 1 callersFunctionget_reference_height
(denorm)
CUDA-V2XFusion/mmdet3d/datasets/v2x_dataset.py:110
↓ 1 callersFunctionget_rot
(h)
CUDA-V2XFusion/mmdet3d/datasets/v2x_dataset.py:114
↓ 1 callersFunctionget_sensor2virtual
(denorm)
CUDA-V2XFusion/mmdet3d/datasets/v2x_dataset.py:96
↓ 1 callersFunctionget_velo2cam
(src_denorm_file)
CUDA-V2XFusion/evaluators/result2kitti.py:102
↓ 1 callersFunctionhas_quantizer
(model : nn.Module)
CUDA-V2XFusion/scripts/tinyq.py:754
↓ 1 callersFunctionhas_sparser
(model : nn.Module)
CUDA-V2XFusion/scripts/tinyq.py:758
↓ 1 callersMethodhead_forward
Forward function for CenterPoint. Args: inputs (torch.Tensor): Input feature map with the shape of [B, 512, 128(H)
libraries/3DSparseConvolution/tool/bevfusion-export/export-transfuser.py:49
↓ 1 callersMethodhead_forward
Forward function for CenterPoint. Args: inputs (torch.Tensor): Input feature map with the shape of [B, 512, 128(H)
CUDA-BEVFusion/qat/export-transfuser.py:49
↓ 1 callersFunctionimg_transform
(img, resize, resize_dims, crop, flip, rotate)
CUDA-V2XFusion/mmdet3d/datasets/v2x_dataset.py:138
↓ 1 callersFunctioninitialize
()
CUDA-CenterPoint/qat/tools/sparseconv_quantization.py:161
↓ 1 callersFunctioninsert_quant_add_to_block
(self)
CUDA-CenterPoint/qat/tools/sparseconv_quantization.py:62
↓ 1 callersFunctionkitti_evaluation
(pred_label_path, gt_label_path, current_classes=["Car", "Pedestrian", "Cyclist"], metric_path="metric")
CUDA-V2XFusion/evaluators/result2kitti.py:83
↓ 1 callersFunctionload_checkpoint
(model, file, startsname=None)
libraries/3DSparseConvolution/tool/bevfusion-export/funcs.py:81
↓ 1 callersFunctionload_json
(path)
CUDA-V2XFusion/scripts/dataset_merge.py:29
↓ 1 callersMethodload_lidar2camera_mat
(self, file, root_path)
CUDA-V2XFusion/mmdet3d/datasets/v2x_dataset.py:432
↓ 1 callersFunctionload_model
(cfg, checkpoint_path = None, use_fuse =False, use_quant_spconv= False, use_quant_add=False)
CUDA-CenterPoint/qat/tools/centerpoint_qat_pipline.py:60
↓ 1 callersFunctionload_model
(cfg, checkpoint_path = None)
CUDA-BEVFusion/qat/ptq.py:73
↓ 1 callersMethodload_pcd
(self, file, convert=True)
CUDA-V2XFusion/mmdet3d/datasets/v2x_dataset.py:447
↓ 1 callersFunctionload_tensor
(file)
libraries/YUVToRGB/compare-with-opencv.py:45
↓ 1 callersFunctionloop
()
libraries/cuOSD/test/pytest.py:78
↓ 1 callersFunctionmain
(args)
libraries/3DSparseConvolution/tool/export-helper.py:46
↓ 1 callersFunctionmain
()
CUDA-CenterPoint/qat/tools/centerpoint_qat_pipline.py:122
↓ 1 callersFunctionmain
(args)
CUDA-CenterPoint/tool/export_neck_head.py:155
↓ 1 callersFunctionmain
()
CUDA-V2XFusion/scripts/train.py:41
↓ 1 callersFunctionmain
()
CUDA-V2XFusion/scripts/export_v2xfusion.py:199
↓ 1 callersFunctionmain
()
CUDA-V2XFusion/scripts/ptq_v2xfusion.py:148
↓ 1 callersFunctionmain
()
CUDA-BEVFusion/qat/ptq.py:87
← previousnext →101–200 of 578, ranked by callers