Functions — github.com/alexklwong/mondi-python

Method__init__

(self, image_paths, dense_depth_paths, sparse_depth_paths,

external_src/scaffnet-pytorch/src/datasets.py:347

Method__init__

(self, images_paths, sparse_depth_paths, intrinsics_paths,

external_src/scaffnet-pytorch/src/datasets.py:420

Method__init__

(self, image_paths, sparse_depth_paths, use_image_triplet=T

external_src/scaffnet-pytorch/src/datasets.py:499

Method__init__

(self, n_layer, input_channels=3, n_filters=[32, 64, 128, 2

external_src/scaffnet-pytorch/src/networks.py:29

Method__init__

(self, n_layer, input_channels=3, n_filters=[32, 64, 128, 2

external_src/scaffnet-pytorch/src/networks.py:338

Method__init__

(self, input_channels=256, output_channels=1, n_resolution=

external_src/scaffnet-pytorch/src/networks.py:705

Method__init__

(self, rotation_parameterization, input_channels=256, n_fil

external_src/scaffnet-pytorch/src/networks.py:1084

Method__init__

(self, input_channels, pool_sizes=[3, 5, 7, 9], n_convoluti

external_src/scaffnet-pytorch/src/networks.py:1169

Method__len__

(self)

src/datasets.py:227

Method__len__

(self)

src/datasets.py:471

Method__len__

(self)

external_src/kbnet/src/datasets.py:227

Method__len__

(self)

external_src/kbnet/src/datasets.py:285

Method__len__

(self)

external_src/PENet_ICRA2021/dataloaders/kitti_loader.py:376

Method__len__

(self)

external_src/scaffnet-pytorch/src/datasets.py:208

Method__len__

(self)

external_src/scaffnet-pytorch/src/datasets.py:234

Method__len__

(self)

external_src/scaffnet-pytorch/src/datasets.py:330

Method__len__

(self)

external_src/scaffnet-pytorch/src/datasets.py:396

Method__len__

(self)

external_src/scaffnet-pytorch/src/datasets.py:482

Method__len__

(self)

external_src/scaffnet-pytorch/src/datasets.py:540

Functionabs_rel_err

Absolute relative error Arg(s): src : numpy[float32] source array tgt : numpy[float32] target array

src/eval_utils.py:84

Functionactivation_func

Select activation function Arg(s): activation_fn : str name of activation function

src/net_utils.py:23

Functionactivation_func

Select activation function Arg(s): activation_fn : str name of activation function Returns: torch.nn.Module

external_src/kbnet/src/net_utils.py:23

Functionactivation_func

Select activation function Arg(s): activation_fn : str name of activation function Returns: torch.nn.Module

external_src/scaffnet-pytorch/src/net_utils.py:8

Functionadd_row

(img_merge, row)

external_src/PENet_ICRA2021/vis_utils.py:82

Functionadjust_gamma

Perform gamma correction on an image. Also known as Power Law Transform. Intensities in RGB mode are adjusted based on the following equation

external_src/PENet_ICRA2021/dataloaders/transforms.py:143

Functionadjust_learning_rate

Sets the learning rate to the initial LR decayed by 10 every 5 epochs

external_src/PENet_ICRA2021/helper.py:216

Functionaggregate_teacher_output

Combines ensemble of teacher output Arg(s): losses0x : list[torch.Tensor[float32]] list[N x 1 x H x W] losses for each t

src/loss_utils.py:147

Functionbackproject_to_camera

Backprojects pixel coordinates to 3D camera coordinates Arg(s): depth : torch.Tensor[float32] N x 1 x H x W depth map

external_src/kbnet/src/net_utils.py:1638

Functionbackproject_to_camera

Backprojects pixel coordinates to 3D camera coordinates Arg(s): depth : torch.Tensor[float32] N x 1 x H x W depth map

external_src/scaffnet-pytorch/src/net_utils.py:963

Functionbackproject_to_camera

Backprojects pixel coordinates to 3D camera coordinates Arg(s): depth : torch.Tensor[float32] N x 1 x H x W depth map

external_src/scaffnet-pytorch/src/loss_utils.py:41

Functionbackup_source_code

(backup_directory)

external_src/NLSPN/src/utility.py:87

Functionbatch_exp_map

Applies exponential map for entire batch Arg(s): v : torch.Tensor[float32] N x 3 vectors Returns: torch.Tens

external_src/scaffnet-pytorch/src/net_utils.py:785

Functionbatch_log_map

Applies logarithmic map for entire batch Arg(s): R : torch.Tensor[float32] N x 3 x 3 rotation matrices Returns:

external_src/scaffnet-pytorch/src/net_utils.py:767

Methodcamera_coordinates

(batch, height, width, k)

src/networks.py:697

Methodcamera_coordinates

(batch, height, width, k)

external_src/kbnet/src/networks.py:317

Functioncheck_file_existence

(path_file)

external_src/NLSPN/utils/generate_json_KITTI_DC.py:52

Functioncheck_file_existence

(path_file)

external_src/NLSPN/utils/prepare_KITTI_DC.py:39

Functioncheck_file_existence

(path_file)

external_src/NLSPN/utils/generate_json_NYUDepthV2.py:62

Functioncolor_consistency_loss_func

Computes the color consistency loss Arg(s): src : torch.Tensor[float32] N x 3 x H x W source image tgt : torch.T

src/loss_utils.py:28

Functioncolor_consistency_loss_func

Computes the color consistency loss Arg(s): src : torch.Tensor[float32] N x 3 x H x W source image tgt : torch.T

external_src/kbnet/src/losses.py:23

Functioncolor_consistency_loss_func

Computes the color consistency loss Arg(s): src : torch.Tensor[float32] N x 3 x H x W source image tgt : torch.T

external_src/scaffnet-pytorch/src/losses.py:47

Functioncolorize

Colorizes a 1-channel tensor with matplotlib colormaps Arg(s): T : torch.Tensor[float32] 1-channel tensor colorm

src/log_utils.py:52

Functioncolorize

Colorizes a 1-channel tensor with matplotlib colormaps Arg(s): T : torch.Tensor[float32] 1-channel tensor colorm

external_src/kbnet/src/log_utils.py:48

Functioncolorize

Colorizes a 1-channel tensor with matplotlib colormaps Args: T : tensor 1-channel tensor colormap : str

external_src/scaffnet-pytorch/src/log_utils.py:30

Methodcompute

(self, sample, output)

external_src/NLSPN/src/loss/nlspnloss.py:29

Methodcompute_loss

Compute loss as NLSPN does: 1.0 * L1 + 1.0 * L2 Arg(s): output_depth : torch.Tensor[float32] N x 1 x H x

src/nlspn_model.py:144

Methodcompute_loss

Computes loss function l = w_{ph}l_{ph} + w_{sz}l_{sz} + w_{sm}l_{sm} + w_{tp}l_{tp} Arg(s): output_depth0 : tor

src/fusionnet_models.py:191

Methodcompute_loss

Call the model's compute loss function Currently only supports supervised methods (ENet, PENet, MSGCHN, NLSPN) Unsupervised

src/external_model.py:141

Methodcompute_loss

Compute L2 Loss for PENet Arg(s): output_depth : torch.Tensor[float32] N x 1 x H x W dense output depth

src/penet_model.py:162

Methodcompute_loss

Compute L2 (MSE) Loss Arg(s): output_depth : torch.Tensor[float32] N x 1 x H x W dense output depth alre

src/msg_chn_model.py:154

Methodcompute_loss

Compute L2 loss for ENet with weighing the rgb_depth and d_depth Arg(s): output_depth : torch.Tensor[float32]

src/enet_model.py:166

Methodcompute_loss

Computes loss function Arg(s): loss_func : list[str] loss functions to minimize target_depth

src/scaffnet_models.py:125

Methodcompute_loss

Computes loss function l = w_{ph}l_{ph} + w_{sz}l_{sz} + w_{sm}l_{sm} Arg(s): image0 : torch.Tensor[float32]

src/kbnet_models.py:148

Methoddata_parallel

Allows multi-gpu split along batch

src/nlspn_model.py:228

Methoddata_parallel

Allows multi-gpu split along batch

src/fusionnet_models.py:310

Methoddata_parallel

Allows multi-gpu split along batch

src/penet_model.py:231

Methoddata_parallel

Allows multi-gpu split along batch

src/msg_chn_model.py:217

Methoddata_parallel

Allows multi-gpu split along batch

src/scaffnet_models.py:196

Methoddata_parallel

Allows multi-gpu split along batch

src/kbnet_models.py:251

Functiondeconvbn

(in_channels, out_channels, kernel_size=4, stride=2, padding=1, output_padding=0)

external_src/PENet_ICRA2021/basic.py:78

Functiondeform_conv_backward

external_src/NLSPN/src/model/deformconv/src/deform_conv.h:45

Functiondeform_conv_cpu_backward

external_src/NLSPN/src/model/deformconv/src/cpu/deform_cpu.cpp:27

Functiondeform_conv_cpu_forward

external_src/NLSPN/src/model/deformconv/src/cpu/deform_cpu.cpp:7

Functiondeform_conv_forward

external_src/NLSPN/src/model/deformconv/src/deform_conv.h:10

Functiondeform_psroi_pooling_backward

external_src/NLSPN/src/model/deformconv/src/deform_psroi_pooling.h:44

Functiondeform_psroi_pooling_cpu_backward

external_src/NLSPN/src/model/deformconv/src/cpu/deform_psroi_pooling_cpu.cpp:23

Functiondeform_psroi_pooling_cpu_forward

external_src/NLSPN/src/model/deformconv/src/cpu/deform_psroi_pooling_cpu.cpp:7

Functiondeform_psroi_pooling_forward

external_src/NLSPN/src/model/deformconv/src/deform_psroi_pooling.h:10

Functiondrop_depth_measurements

(depth, prob_keep)

external_src/PENet_ICRA2021/dataloaders/kitti_loader.py:180

Functioneuler_angles

Converts euler angles to rotation matrix Arg(s): v : torch.Tensor[float32] N x 3 vector of rotation angles along x, y, z

external_src/scaffnet-pytorch/src/net_utils.py:803

Methodeval

Sets model to evaluation mode

src/nlspn_model.py:210

Methodeval

Sets model to evaluation mode

src/posenet_model.py:126

Methodeval

Sets model to evaluation mode

src/penet_model.py:212

Methodeval

Sets model to evaluation mode

src/msg_chn_model.py:200

Methodeval

Sets model to evaluation mode

external_src/kbnet/src/posenet_model.py:129

Methodevaluate

(self, sample, output, mode)

external_src/NLSPN/src/metric/nlspnmetric.py:31

Methodforward

Forwards inputs through the network Arg(s): image : torch.Tensor[float32] N x 3 x H x W image

src/nlspn_model.py:84

Methodforward

(self, x)

src/net_utils.py:191

Methodforward

(self, x)

src/net_utils.py:254

Methodforward

(self, x, shape)

src/net_utils.py:302

Methodforward

(self, x)

src/net_utils.py:368

Methodforward

(self, x)

src/net_utils.py:451

Methodforward

(self, x)

src/net_utils.py:533

Methodforward

(self, image, depth, coordinates, fused=None)

src/net_utils.py:616

Methodforward

(self, x, skip=None, shape=None)

src/net_utils.py:720

Methodforward

Forwards inputs through the network Arg(s): image : torch.Tensor[float32] N x 3 x H x W image

src/fusionnet_models.py:125

Methodforward

Forwards inputs through network Arg(s): image : torch.Tensor[float32] N x 3 x H x W image sp

src/external_model.py:124

Methodforward

Forwards inputs through the network Arg(s): image : torch.Tensor[float32] N x 3 x H x W image

src/penet_model.py:58

Methodforward

Forwards inputs through the network Arg(s): image : tensor[float32] N x 3 x H x W image spar

src/msg_chn_model.py:51

Methodforward

Forward input x through the ResNet model Arg(s): x : torch.Tensor Returns: torch.Tensor[float32] : l

src/networks.py:260

Methodforward

Forward input x through the ResNet model Arg(s): image : torch.Tensor[float32] N x 3 x H x W tensor

src/networks.py:350

Methodforward

Forward input x through the ResNet model Arg(s): image : torch.Tensor[float32] N x 3 x H x W tensor

src/networks.py:681

Methodforward

Forward input x through the ResNet model Arg(s): x : torch.Tensor[float32] N x 6 x H x W tensor

src/networks.py:1006

Methodforward

Forward latent vector x through decoder network Arg(s): x : torch.Tensor[float32] latent vector

src/networks.py:1265

Methodforward

Forward latent vector x through decoder network Arg(s): x : torch.Tensor[float32] latent vector

src/networks.py:1550

Methodforward

Forward latent vector x through decoder network Arg(s): x : torch.Tensor[float32] latent vector

src/networks.py:1702

Methodforward

Forward depth and validity maps through sparse to dense pool Arg(s): x : torch.Tensor[float32] Returns:

src/networks.py:1814

Methodforward

Forwards inputs through the network Arg(s): image : torch.Tensor[float32] N x 3 x H x W image

src/enet_model.py:58

Methodforward

Forwards inputs through the network Arg(s): image : torch.Tensor[float32] N x 3 x H x W image

src/scaffnet_models.py:86

Methodforward

Forwards inputs through the network Arg(s): image : torch.Tensor[float32] N x 3 x H x W image

src/kbnet_models.py:88

Functions823 in github.com/alexklwong/mondi-python