Functions — github.com/RozDavid/UnScene3D

CodeHub MCPcopy Create free account

hub / github.com/RozDavid/UnScene3D / functions

Functions1,028 in github.com/RozDavid/UnScene3D

⨍ Functions 1,028◇ Types & classes 235

:param tensor: A 5d tensor of size (batch_size, x, y, z, ch) :return: Positional Encoding Matrix of size (batch_size, x, y, z, ch)

models/mask3d.py:473

(self, tgt, tgt_mask = None, tgt_key_padding_mask = None, quer

models/mask3d.py:538

(self, tgt, memory, memory_mask = None, memory_key_padding_mask = None,

models/mask3d.py:600

models/mask3d.py:648

This performs the loss computation. Parameters: outputs: dict of tensors, see the output specification of the model for the forma

models/criterion.py:239

models/resnet.py:151

models/modules/resnet_block.py:48

models/modules/resnet_block.py:117

x: HW x N x C permute to N x C x HW Apply BN on C permute back

models/modules/helpers_3detr.py:12

models/modules/senet_block.py:21

models/modules/senet_block.py:50

models/modules/senet_block.py:105

x: HW x N x C permute to N x C x HW Apply BN on C permute back

models/modules/3detr_helpers.py:12

(self, input_images)

models/encoders_2d/lseg.py:46

(self, input_images)

models/encoders_2d/dino.py:128

utils/cpp_utils/include/vec3.h:372

utils/cuda_utils/include/cutil_math.h:1239

Functionfreeze_until

Freeze net until param_name https://opendatascience.slack.com/archives/CGK4KQBHD/p1588373239292300?thread_ts=1588105223.275700&cid=CGK4KQBHD

utils/utils.py:131

Methodfrom_json

pseudo_masks/datasets/evaluation/scannet_benchmark_utils/util_3d.py:120

Methodfrom_json

benchmark/util_3d.py:114

Functionfurthestsampling_cuda

utils/pointops2/src/sampling/sampling_cuda.cpp:8

Functiongenerate_splits

Takes preprocessed out path and generate txt files

pseudo_masks/datasets/preprocess/stanford.py:216

utils/cuda_utils/include/cuda_SimpleMatrixUtil.h:1459

utils/cpp_utils/include/vec3.h:271

MethodgetDiagonalMatrix

utils/cuda_utils/include/cuda_SimpleMatrixUtil.h:528

MethodgetFloat3x3

utils/cuda_utils/include/cuda_SimpleMatrixUtil.h:740

MethodgetFloat3x3

utils/cuda_utils/include/cuda_SimpleMatrixUtil.h:1076

MethodgetFloat3x4

utils/cuda_utils/include/cuda_SimpleMatrixUtil.h:1110

utils/cuda_utils/include/cudaUtil.h:86

utils/cuda_utils/include/cudaUtil.h:88

MethodgetIdentity

utils/cuda_utils/include/cuda_SimpleMatrixUtil.h:42

MethodgetIdentity

utils/cuda_utils/include/cuda_SimpleMatrixUtil.h:515

MethodgetIdentity

utils/cuda_utils/include/cuda_SimpleMatrixUtil.h:1202

MethodgetInverse

utils/cuda_utils/include/cuda_SimpleMatrixUtil.h:57

MethodgetInverse

utils/cuda_utils/include/cuda_SimpleMatrixUtil.h:373

MethodgetInverse

utils/cuda_utils/include/cuda_SimpleMatrixUtil.h:761

MethodgetInverse

utils/cuda_utils/include/cuda_SimpleMatrixUtil.h:944

MethodgetInverse

utils/cuda_utils/include/cuda_SimpleMatrixUtil.h:1557

utils/cuda_utils/include/cudaUtil.h:90

utils/cuda_utils/include/cudaUtil.h:92

utils/cuda_utils/include/cuda_SimpleMatrixUtil.h:406

MethodgetTranslation

utils/cuda_utils/include/cuda_SimpleMatrixUtil.h:728

MethodgetTranspose

utils/cuda_utils/include/cuda_SimpleMatrixUtil.h:272

MethodgetTranspose

utils/cuda_utils/include/cuda_SimpleMatrixUtil.h:361

MethodgetTranspose

utils/cuda_utils/include/cuda_SimpleMatrixUtil.h:932

MethodgetTranspose

utils/cuda_utils/include/cuda_SimpleMatrixUtil.h:1268

MethodgetZeroMatrix

utils/cuda_utils/include/cuda_SimpleMatrixUtil.h:522

Functionget_3d_box

box_size is array(l,w,h), heading_angle is radius clockwise from pos x axis, center is xyz of box center output (8,3) array for 3D box cornde

utils/votenet_utils/box_util.py:210

Functionget_3d_box_batch

box_size: [x1,x2,...,xn,3] heading_angle: [x1,x2,...,xn] center: [x1,x2,...,xn,3] Return: [x1,x3,...,xn,8,3]

utils/votenet_utils/box_util.py:227

Methodget_classids

pseudo_masks/datasets/scannet.py:302

Methodget_classnames

pseudo_masks/datasets/scannet.py:305

Methodget_classnames

pseudo_masks/datasets/dataset.py:134

Functionget_clones

models/modules/helpers_3detr.py:115

Functionget_clones

models/modules/3detr_helpers.py:115

Methodget_full_cloud_by_scene_name

(self, scene_name)

pseudo_masks/datasets/scannet.py:321

Functionget_instances

(ids, class_ids, class_labels, id2label)

pseudo_masks/datasets/evaluation/scannet_benchmark_utils/util_3d.py:154

Functionget_instances

(ids, class_ids, class_labels, id2label)

benchmark/util_3d.py:148

Functionget_iou

Compute IoU of two bounding boxes. ** Define your bod IoU function HERE **

utils/votenet_utils/eval_det.py:61

Functionget_iou_obb

utils/votenet_utils/eval_det.py:70

Methodget_output_id

(self, iteration)

pseudo_masks/datasets/scannet.py:299

Functionget_prediction

(dataset, output, target)

utils/utils.py:476

Functionget_torch_device

utils/utils.py:484

Functiongrouping_backward_cuda

utils/pointops2/src/grouping/grouping_cuda.cpp:16

Functiongrouping_forward_cuda

utils/pointops2/src/grouping/grouping_cuda.cpp:8

Methodhisto_summary

Log a histogram of the tensor of values.

utils/votenet_utils/tf_logger.py:49

Methodhook_fn_forward_qkv

(module, input, output)

models/encoders_2d/dino.py:78

Functionimage_grid

A util function for plotting a grid of images. Args: images: (N, H, W, 4/1) array of RGBA images rows: number of rows in the

utils/utils.py:292

Methodindex_to_scene_name

pseudo_masks/datasets/scannet.py:318

Methodindex_to_scene_name

pseudo_masks/datasets/arkit.py:168

Functioninterpolation

input: xyz: (m, 3), new_xyz: (n, 3), feat: (m, c), offset: (b), new_offset: (b) output: (n, c)

utils/pointops2/functions/pointops_ablation.py:165

Functioninterpolation

input: xyz: (m, 3), new_xyz: (n, 3), feat: (m, c), offset: (b), new_offset: (b) output: (n, c)

utils/pointops2/functions/pointops2.py:164

Functioninterpolation

input: xyz: (m, 3), new_xyz: (n, 3), feat: (m, c), offset: (b), new_offset: (b) output: (n, c)

utils/pointops2/functions/pointops.py:752

Functioninterpolation_backward_cuda

utils/pointops2/src/interpolation/interpolation_cuda.cpp:17

Functioninterpolation_forward_cuda

utils/pointops2/src/interpolation/interpolation_cuda.cpp:8

Functioninterpolation_v2

input: xyz: (m, 3), new_xyz: (n, 3), feat: (m, c), offset: (b), new_offset: (b) output: (n, c)

utils/pointops2/functions/pointops.py:769

Methodintrinsics

utils/pc_utils.py:313

utils/cuda_utils/include/cuda_SimpleMatrixUtil.h:939

utils/cpp_utils/include/vec3.h:111

utils/cpp_utils/include/vec3.h:107

Functionis_clockwise

utils/votenet_utils/box_util.py:87

Methodjoint_database

(self, train_modes=("Area_1", "Area_2", "Area_3", "Area_4", "Area_5", "Area_6"))

datasets/preprocessing/s3dis_preprocessing.py:210

FunctionkinectProjToCamera

utils/cuda_utils/include/cudaUtil.h:113

Functionknnquery_cuda

utils/pointops2/src/knnquery/knnquery_cuda.cpp:8

Methodlabel_info

database file containing information labels used by dataset

datasets/outdoor_semseg.py:146

Methodlabel_info

database file containing information labels used by dataset

datasets/freemask_semseg.py:173

Methodlabel_info

database file containing information labels used by dataset

datasets/semseg.py:565

utils/cpp_utils/include/vec3.h:183

utils/cuda_utils/include/cutil_math.h:1028

Functionlidar_3duis

(model, coords, features, device='cuda', z_max = 0.25)

utils/freemask_utils.py:38

Functionload_baseline_model

utils/utils.py:44

Functionload_data

pseudo_masks/datasets/dataset.py:225

Functionload_ids

pseudo_masks/datasets/evaluation/scannet_benchmark_utils/util_3d.py:38

Functionload_ids

benchmark/util_3d.py:38

Functionload_image

(path, scale=1.)

utils/utils.py:202

Methodload_input

pseudo_masks/datasets/dataset.py:128

Functionload_labels

(point_cloud, segments_file, aggregations_file, labels_pd, CLASS_IDs = VALID_C

utils/pc_utils.py:363

Functionload_obj

utils/utils.py:512

pseudo_masks/datasets/dataset.py:200

Methodload_ply_cloud_with_normals

(self, filepath)

datasets/preprocessing/arkit_preprocessing.py:56

Methodload_target

pseudo_masks/datasets/dataset.py:131

← previous next →701–800 of 1,028, ranked by callers