MCPcopy Create free account

hub / github.com/42dot/VFDepth / functions

Functions160 in github.com/42dot/VFDepth

↓ 13 callersMethodupdate
Display logs with respect to the log frequency
utils/logger.py:109
↓ 12 callersMethodget_current
This function returns samples for current contexts
dataset/nuscenes_dataset.py:63
↓ 11 callersFunctionset_tb_title
This function sets title for tensorboard plot.
utils/logger.py:17
↓ 7 callersFunctionconv2d
This function computes 2d convolutions followed by the specific nonlinear function(LeakyReLU, ELU, or None)
network/blocks.py:41
↓ 6 callersFunctionplot_tb
This function plots images on tensotboard.
utils/logger.py:37
↓ 4 callersFunctioncompute_masked_loss
This function masks losses while avoiding zero division.
models/losses/loss_util.py:21
↓ 4 callersFunctioncompute_photometric_loss
This function calculates photometric reconstruction loss (0.85*SSIM + 0.15*L1)
models/losses/loss_util.py:70
↓ 4 callersFunctionpack_cam_feat
This function packs camera dimension to batch dimension.
network/blocks.py:6
↓ 4 callersMethodprint_perf
This function prints various metrics for depth estimation accuracy.
utils/logger.py:249
↓ 4 callersFunctionunpack_cam_feat
This function unpacks batch dimension into batch and camera dimensions.
network/blocks.py:21
↓ 3 callersFunctioncolormap
This function visualizes disparity map using colormap specified with disparity map variable.
utils/visualize.py:60
↓ 3 callersMethodcompute_reproj_loss
This function computes reprojection loss using auto mask.
models/losses/single_cam_loss.py:17
↓ 3 callersMethodcompute_smooth_loss
This function computes edge-aware smoothness loss for the disparity map.
models/losses/single_cam_loss.py:57
↓ 3 callersFunctionconstruct_dataset
This function constructs datasets.
dataset/base_dataset.py:5
↓ 3 callersMethodget_logs
This function logs depth and pose information for monitoring training process.
models/losses/base_loss.py:27
↓ 3 callersFunctionplot_norm_tb
This function plots normalized images on tensotboard.
utils/logger.py:45
↓ 3 callersMethodprocess_batch
Pass a minibatch through the network and generate images, depth maps, and losses.
models/vfdepth.py:191
↓ 3 callersMethodproject_voxel_into_image
This function projects voxels into 2D image coordinate. [b, feat_dim, n_voxels] -> [b, feat_dim, d, h, w]
network/volumetric_fusionnet.py:232
↓ 3 callersFunctionresize_for_tb
This function resizes images for tensorboard plot.
utils/logger.py:29
↓ 2 callersFunctionalign_dataset
This function reorganize samples to match our trainer configuration.
dataset/data_util.py:46
↓ 2 callersMethodbackproject
This function back-projects 2D image points to 3D.
models/geometry/geometry_util.py:52
↓ 2 callersFunctioncal_depth_error
This function calculates depth error using various metrics.
utils/misc.py:85
↓ 2 callersMethodcompute_depth_losses
This function computes depth metrics, to allow monitoring of training process on validation dataset.
utils/logger.py:193
↓ 2 callersMethodcompute_spatio_loss
This function computes spatial loss.
models/losses/multi_cam_loss.py:16
↓ 2 callersMethodcompute_spatio_tempo_loss
This function computes spatio-temporal loss.
models/losses/multi_cam_loss.py:31
↓ 2 callersFunctionconv1d
This function computes 1d convolutions follwed by the nonlinear function(LeakyReLU, or None)
network/blocks.py:64
↓ 2 callersMethodevaluate
This function evaluates models on full validation dataset.
trainer/vfdepth_trainer.py:113
↓ 2 callersMethodgenerate_depth_map_sf
This function follows structure of dgp_dataset/generate_depth_map in packnet-sfm. Due to the version issue with dgp, minor revision
dataset/ddad_dataset_sf.py:41
↓ 2 callersMethodget_context
This function returns samples for backward and forward contexts
dataset/nuscenes_dataset.py:84
↓ 2 callersMethodget_mean_std
This function returns mean and standard deviation of the overlapped features.
models/geometry/view_rendering.py:30
↓ 2 callersMethodget_norm_image_single
obtain normalized warped images using the mean and the variance from the overlapped regions of the target frame.
models/geometry/view_rendering.py:39
↓ 2 callersMethodget_single_pose
This function computes pose for a single camera.
models/geometry/pose.py:32
↓ 2 callersMethodget_virtual_image
This function warps source image to target image using backprojection and reprojection process.
models/geometry/view_rendering.py:61
↓ 2 callersMethodis_checkpoint
Log less frequently after the early phase steps
utils/logger.py:120
↓ 2 callersMethodlearn
This function sets training process.
trainer/vfdepth_trainer.py:28
↓ 2 callersMethodload_weights
(self)
models/base_model.py:58
↓ 2 callersMethodlog_tb
This function logs outputs for monitoring using tensorboard.
utils/logger.py:138
↓ 2 callersFunctionplot_disp_tb
This function plots disparity maps on tensotboard.
utils/logger.py:53
↓ 2 callersMethodset_train
(self)
models/base_model.py:35
↓ 2 callersMethodset_val
(self)
models/base_model.py:40
↓ 2 callersMethodto_depth
This function transforms disparity value into depth map while multiplying the value with the focal length.
models/vfdepth.py:277
↓ 2 callersMethodtrain
This function trains models.
trainer/vfdepth_trainer.py:56
↓ 2 callersFunctiontransform_mask_sample
This function transforms masks to match input rgb images.
dataset/data_util.py:13
↓ 1 callersMethod__init__
(self, level_in, num_ch_enc, num_ch_dec, scales=range(2), use_skips=False)
network/fusion_depthnet.py:102
↓ 1 callersFunctionaug_depth_params
This function augments camera parameters for depth synthesis.
utils/visualize.py:9
↓ 1 callersMethodaugment_extrinsics
This function augments depth estimation results using augmented extrinsics [batch, cam, 4, 4]
network/volumetric_fusionnet.py:269
↓ 1 callersMethodbackproject_into_voxel
This function backprojects 2D features into 3D voxel coordinate using intrinsic and extrinsic of each camera. Self-occluded regions a
network/volumetric_fusionnet.py:116
↓ 1 callersMethodcalculate_sample_pixel_coords
This function calculates pixel coords for each point([batch, n_voxels, 1, 2]) to sample the per-pixel feature.
network/volumetric_fusionnet.py:166
↓ 1 callersMethodcalculate_valid_mask
This function creates valid mask in voxel coordinate by projecting self-occlusion mask to 3D voxel coords.
network/volumetric_fusionnet.py:183
↓ 1 callersFunctioncamera2ind
This function transforms camera name list to indices
utils/misc.py:13
↓ 1 callersMethodclose_tb
(self)
utils/logger.py:88
↓ 1 callersMethodcompute_aug_losses
This function computes depth augmentation loss(consistency, smoothness).
models/losses/depth_synthesis_loss.py:15
↓ 1 callersFunctioncompute_auto_masks
This function computes auto mask using reprojection loss and identity reprojection loss.
models/losses/loss_util.py:6
↓ 1 callersMethodcompute_depth_maps
This function computes depth map for each viewpoint.
models/vfdepth.py:263
↓ 1 callersFunctioncompute_edg_smooth_loss
This function calculates edge-aware smoothness.
models/losses/loss_util.py:28
↓ 1 callersMethodcompute_losses
This function computes losses.
models/vfdepth.py:290
↓ 1 callersMethodcompute_pose
This function computes multi-camera posse in accordance with the network structure.
models/geometry/pose.py:19
↓ 1 callersMethodcompute_pose_con_loss
This function computes pose consistency loss in "Full surround monodepth from multiple cameras"
models/losses/multi_cam_loss.py:61
↓ 1 callersMethodcompute_relative_cam_poses
This function computes spatio & spatio-temporal transformation for images from different viewpoints.
models/geometry/pose.py:66
↓ 1 callersFunctioncompute_ssim_loss
This function calculates SSIM loss between predicted image and target image.
models/losses/loss_util.py:43
↓ 1 callersMethodcreate_depth_grid
output: [batch, 3, num_depths, height * width]
network/volumetric_fusionnet.py:93
↓ 1 callersMethodcreate_pixel_grid
output: [batch, 3, height * width]
network/volumetric_fusionnet.py:82
↓ 1 callersMethodcreate_voxel_grid
output: [batch, 3, z_dim, y_dim, x_dim] [b, :, z, y, x] contains (x,y,z) 3D point
network/volumetric_fusionnet.py:67
↓ 1 callersMethoddistribute_pose
This function distrubutes pose to each camera by using the canonical pose and camera extrinsics. (default: reference camera 0)
models/geometry/pose.py:44
↓ 1 callersMethodestimate_vfdepth
This function sets dataloader for validation in training.
models/vfdepth.py:206
↓ 1 callersMethodeval_dataloader
(self)
models/base_model.py:32
↓ 1 callersMethodgenerate_depth_map
This function returns depth map for nuscenes dataset, result of depth map is saved in nuscenes/samples/DEPTH_MAP
dataset/nuscenes_dataset.py:104
↓ 1 callersMethodget_filename_sf
This function is defined to meet dgp version(v1.4)
dataset/ddad_dataset_sf.py:74
↓ 1 callersMethodget_metric_names
(self)
utils/logger.py:106
↓ 1 callersFunctionget_relcam
This function returns relative camera indices from given camera list
utils/misc.py:29
↓ 1 callersMethodget_tranformation_mat
This function transforms pose information in accordance with DDAD dataset format
dataset/nuscenes_dataset.py:212
↓ 1 callersMethodget_virtual_depth
This function backward-warp source depth into the target coordinate. src -> target
models/geometry/view_rendering.py:84
↓ 1 callersMethodinit_attrib
(self, cfg)
models/losses/base_loss.py:21
↓ 1 callersMethodinit_geometry
(self, cfg, rank)
models/vfdepth.py:42
↓ 1 callersMethodinit_losses
(self, cfg, rank)
models/vfdepth.py:47
↓ 1 callersMethodinit_project_imgs
(self, rank)
models/geometry/view_rendering.py:24
↓ 1 callersMethodinit_tb
(self)
utils/logger.py:83
↓ 1 callersMethodinit_vis
(self)
utils/logger.py:92
↓ 1 callersMethodinit_weights
(self, cfg)
models/losses/base_loss.py:15
↓ 1 callersMethodlog_result
This function logs outputs for visualization.
utils/logger.py:172
↓ 1 callersMethodlog_time
This function prints epoch, iteration, duration, loss and spent time.
utils/logger.py:128
↓ 1 callersFunctionparse_args
()
eval.py:14
↓ 1 callersFunctionparse_args
()
train.py:32
↓ 1 callersMethodpred_cam_imgs
This function renders projected images using camera parameters and depth information.
models/vfdepth.py:315
↓ 1 callersMethodpredict_depth
This function predicts disparity maps.
models/vfdepth.py:244
↓ 1 callersMethodpredict_pose
This function predicts poses.
models/vfdepth.py:231
↓ 1 callersMethodprepare_dataset
(self, cfg, rank)
models/vfdepth.py:85
↓ 1 callersMethodprepare_model
(self, cfg, rank)
models/vfdepth.py:56
↓ 1 callersMethodpreprocess_non_overlap
This function applies 1x1 convolutions to features from non-overlapping features.
network/volumetric_fusionnet.py:197
↓ 1 callersMethodpreprocess_overlap
This function applies 1x1 convolutions on overlapping features. Camera configuration [0,1,2] or [0,1,2,3,4,5]:
network/volumetric_fusionnet.py:209
↓ 1 callersFunctionpretty_ts
This function prints amount of time taken in user friendly way.
utils/misc.py:75
↓ 1 callersMethodread_config
(self, cfg)
utils/logger.py:78
↓ 1 callersMethodread_config
(self, cfg)
network/fusion_posenet.py:37
↓ 1 callersMethodread_config
(self, cfg)
network/volumetric_fusionnet.py:62
↓ 1 callersMethodread_config
(self, cfg)
network/fusion_depthnet.py:38
↓ 1 callersMethodread_config
(self, cfg)
trainer/vfdepth_trainer.py:23
↓ 1 callersMethodread_config
(self, cfg)
models/vfdepth.py:37
↓ 1 callersMethodread_config
(self, cfg)
models/geometry/view_rendering.py:19
↓ 1 callersMethodread_config
(self, cfg)
models/geometry/pose.py:14
↓ 1 callersMethodreproject
This function reprojects transformed 3D points to 2D image coordinate.
models/geometry/geometry_util.py:62
next →1–100 of 160, ranked by callers