MCPcopy Create free account

hub / github.com/JiawangBian/sc_depth_pl / functions

Functions114 in github.com/JiawangBian/sc_depth_pl

↓ 6 callersFunctionvisualize_image
tensor image: (3, H, W)
visualization.py:7
↓ 4 callersFunctioncompute_depth_errors
Computation of error metrics between predicted and ground truth depths Args: gt (N): ground truth depth pred (N): predicted depth
eval_depth.py:31
↓ 4 callersFunctionsub2ind
(r, c, cols)
losses/normal_ranking_loss.py:55
↓ 4 callersFunctionvisualize_depth
depth: (H, W)
visualization.py:15
↓ 3 callersMethod__init__
(self, in_channels, out_channels, use_refl=True)
models/DepthNet.py:27
↓ 3 callersFunctionget_opts
()
config.py:4
↓ 3 callersFunctionget_training_size
(dataset_name)
config.py:83
↓ 3 callersFunctioninverse_rotation_warp
(img, rot, intrinsics, padding_mode='zeros')
losses/inverse_warp.py:125
↓ 3 callersFunctionload_as_float
(path)
datasets/train_folders.py:8
↓ 2 callersMethodcal_ranking_loss
loss for a given set of pixels: z_A: predicted absolute depth for pixels A z_B: predicted absolute depth for pixels B
losses/mask_ranking_loss.py:85
↓ 2 callersFunctioncompute_pairwise_loss
(tgt_img, ref_img, tgt_depth, ref_depth, pose, intrinsic, hparams)
losses/loss_functions.py:112
↓ 2 callersFunctionedgeGuidedSampling
(inputs, targets, edges_img, thetas_img, masks, h, w)
losses/normal_ranking_loss.py:60
↓ 2 callersFunctioneuler2mat
Convert euler angles to rotation matrix. Reference: https://github.com/pulkitag/pycaffe-utils/blob/master/rot_utils.py#L174 Args: ang
losses/inverse_warp.py:7
↓ 2 callersMethodgetEdge
(self, images)
losses/normal_ranking_loss.py:161
↓ 2 callersFunctionmean_on_mask
(diff, valid_mask)
losses/loss_functions.py:147
↓ 1 callersMethod__init__
(self, num_ch_enc, num_input_features=1, num_frames_to_predict_for=1, stride=1)
models/RectifyNet.py:8
↓ 1 callersMethod__init__
(self, num_layers, pretrained, num_input_images=1)
models/resnet_encoder.py:70
↓ 1 callersMethod__init__
(self, num_layers=18, pretrained=True)
models/PoseNet.py:51
↓ 1 callersFunctioncompute_errors
(gt, pred, dataset)
losses/loss_functions.py:187
↓ 1 callersFunctioncompute_movement_ratio
(frame1, frame2)
generate_valid_frame_index.py:16
↓ 1 callersFunctioncrawl_folder
(folder, dataset='nyu')
datasets/test_folder.py:15
↓ 1 callersFunctioncrawl_folders
(folders_list, dataset='nyu')
datasets/validation_folders.py:15
↓ 1 callersMethodcrawl_folders
(self, sequence_length)
datasets/train_folders.py:63
↓ 1 callersMethodevaluate_depth
evaluate depth result Args: gt_depths: list of gt depth files pred_depths: list of predicted depth files e
eval_depth.py:100
↓ 1 callersMethodgenerate_global_target
(self, depth, pred, theta=0.15)
losses/mask_ranking_loss.py:14
↓ 1 callersFunctiongenerate_index
(scene)
generate_valid_frame_index.py:25
↓ 1 callersMethodgenerate_percentMask_target
(self, depth, pred, invalid_mask, theta=0.15)
losses/mask_ranking_loss.py:41
↓ 1 callersFunctiongenerate_sample_index
(num_frames, skip_frames, sequence_length)
datasets/train_folders.py:12
↓ 1 callersMethodgetNormalEdge
(self, normals)
losses/normal_ranking_loss.py:187
↓ 1 callersFunctionget_smooth_loss
Computes the smoothness loss for a disparity image The color image is used for edge-aware smoothness
losses/loss_functions.py:157
↓ 1 callersMethodget_unreliable
(self, tgt_valid_weight)
losses/mask_ranking_loss.py:100
↓ 1 callersFunctionind2sub
(idx, cols)
losses/normal_ranking_loss.py:49
↓ 1 callersFunctioninverse_warp
Inverse warp a source image to the target image plane. Args: img: the source image (where to sample pixels) -- [B, 3, H, W] d
losses/inverse_warp.py:87
↓ 1 callersFunctionload_sparse_depth
(filename)
eval_depth.py:25
↓ 1 callersFunctionload_sparse_depth
(filename)
datasets/test_folder.py:9
↓ 1 callersFunctionload_sparse_depth
(filename)
datasets/validation_folders.py:9
↓ 1 callersFunctionmain
()
test.py:15
↓ 1 callersFunctionmain
()
generate_valid_frame_index.py:45
↓ 1 callersFunctionmain
()
inference.py:20
↓ 1 callersMethodmain
(self)
eval_depth.py:73
↓ 1 callersFunctionparse_args
()
generate_valid_frame_index.py:8
↓ 1 callersFunctionpose_vec2mat
Convert 6DoF parameters to transformation matrix. Args:s vec: 6DoF parameters in the order of tx, ty, tz, rx, ry, rz -- [B, 6] Re
losses/inverse_warp.py:69
↓ 1 callersFunctionquat2mat
Convert quaternion coefficients to rotation matrix. Args: quat: first three coeff of quaternion of rotation. fourht is then computed to ha
losses/inverse_warp.py:45
↓ 1 callersMethodrectify_imgs
(self, tgt_img, ref_imgs, intrinsics)
SC_DepthV2.py:23
↓ 1 callersFunctionresnet_multiimage_input
Constructs a ResNet model. Args: num_layers (int): Number of resnet layers. Must be 18 or 50 pretrained (bool): If True, returns a
models/resnet_encoder.py:43
↓ 1 callersFunctionupsample
Upsample input tensor by a factor of 2
models/DepthNet.py:42
Method__call__
(self, images, intrinsics=None)
datasets/custom_transforms.py:15
Method__call__
(self, images, intrinsics)
datasets/custom_transforms.py:26
Method__call__
(self, images, intrinsics)
datasets/custom_transforms.py:38
Method__call__
(self, images, intrinsics)
datasets/custom_transforms.py:54
Method__call__
(self, images, intrinsics)
datasets/custom_transforms.py:70
Method__call__
(self, images, intrinsics)
datasets/custom_transforms.py:107
Method__getitem__
(self, index)
datasets/test_folder.py:42
Method__getitem__
(self, index)
datasets/validation_folders.py:52
Method__getitem__
(self, index)
datasets/train_folders.py:101
Method__init__
(self, hparams)
SC_Depth.py:12
Method__init__
(self)
eval_depth.py:56
Method__init__
(self, hparams)
data_modules.py:12
Method__init__
(self, hparams)
SC_DepthV2.py:14
Method__init__
(self, hparams)
SC_DepthV3.py:13
Method__init__
(self, root, transform=None, dataset='nyu')
datasets/test_folder.py:36
Method__init__
(self, root, transform=None, dataset='nyu')
datasets/validation_folders.py:43
Method__init__
(self, root, train=True, sequence_length=3,
datasets/train_folders.py:41
Method__init__
(self, transforms)
datasets/custom_transforms.py:12
Method__init__
(self, mean=[0.45, 0.45, 0.45], std=[0.225, 0.225, 0.225])
datasets/custom_transforms.py:22
Method__init__
(self, output_size=[256, 832])
datasets/custom_transforms.py:104
Method__init__
(self, num_layers=18, pretrained=True)
models/RectifyNet.py:51
Method__init__
(self, in_channels, out_channels)
models/DepthNet.py:11
Method__init__
(self, num_ch_enc, scales=range(4), num_output_channels=1, use_skips=True)
models/DepthNet.py:49
Method__init__
(self, num_layers=18, pretrained=True)
models/DepthNet.py:123
Method__init__
(self, block, layers, num_classes=1000, num_input_images=1)
models/resnet_encoder.py:21
Method__init__
(self, num_ch_enc, num_input_features=1, num_frames_to_predict_for=1, stride=1)
models/PoseNet.py:7
Method__init__
( self, point_pairs=10000, cos_theta1=0.25, cos_theta2=0.98, cos_
losses/normal_ranking_loss.py:140
Method__init__
(self, sample_ratio=0.1, filter_depth=1e-8)
losses/mask_ranking_loss.py:9
Method__init__
(self)
losses/loss_functions.py:19
Method__len__
(self)
datasets/test_folder.py:61
Method__len__
(self)
datasets/validation_folders.py:71
Method__len__
(self)
datasets/train_folders.py:129
Functioncompute_smooth_loss
(tgt_depth, tgt_img)
losses/loss_functions.py:156
Methodconfigure_optimizers
(self)
SC_Depth.py:20
Methodconfigure_optimizers
(self)
SC_DepthV2.py:71
Methodconfigure_optimizers
(self)
SC_DepthV3.py:22
Methodforward
(self, input_features)
models/RectifyNet.py:31
Methodforward
(self, img1, img2)
models/RectifyNet.py:60
Methodforward
(self, x)
models/DepthNet.py:17
Methodforward
(self, x)
models/DepthNet.py:36
Methodforward
(self, input_features)
models/DepthNet.py:96
Methodforward
(self, x)
models/DepthNet.py:132
Methodforward
(self, input_image)
models/resnet_encoder.py:97
Methodforward
(self, input_features)
models/PoseNet.py:30
Methodforward
(self, img1, img2)
models/PoseNet.py:60
Methodforward
inputs and targets: surface normal image images: rgb images
losses/normal_ranking_loss.py:211
Methodforward
(self, pred_depth, gt_depth, tgt_valid_weight)
losses/mask_ranking_loss.py:130
Methodforward
(self, x, y)
losses/loss_functions.py:33
Methodget_textureWeight
(self, tgt_img)
losses/mask_ranking_loss.py:119
Methodinit_weights
(self)
models/RectifyNet.py:57
Methodinit_weights
(self)
models/DepthNet.py:93
Methodinit_weights
(self)
models/DepthNet.py:129
Methodinit_weights
(self)
models/PoseNet.py:57
Functionphoto_and_geometry_loss
(tgt_img, ref_imgs, tgt_depth, ref_depths, intrinsics, poses, poses_inv, hparams)
losses/loss_functions.py:58
next →1–100 of 114, ranked by callers