Functions — github.com/DIVE128/DMVSNet

CodeHub MCPcopy Create free account

hub / github.com/DIVE128/DMVSNet / functions

Functions151 in github.com/DIVE128/DMVSNet

⨍ Functions 151◇ Types & classes 20 ↳ Endpoints 2

(self, in_channels, base_channels,stage=0)

networks/module.py:351

(self, in_channels, base_channels,stage=0)

networks/module.py:359

(self, in_channels, base_channels,stage=0)

networks/module.py:401

(self, in_channels=32,hid_channels=1,out_channels=1,relu=True)

networks/module.py:438

(self, CropSize=0.1)

datasets/data_io.py:75

(self, datapath, listfile, mode, nviews, ndepths=128, interval_scale=1.06)

datasets/blendedmvs.py:35

(self, datapath, listfile, mode, nviews, img_size=None, ndepths=192, interval_scale=1.06, **kwargs)

datasets/dtu_yao.py:12

(self, datapath, listfile, mode, nviews, ndepths=192, interval_scale=1.06, inverse_depth=False, **kwargs)

datasets/general_eval.py:10

datasets/blendedmvs.py:74

datasets/dtu_yao.py:55

datasets/general_eval.py:58

Functioncheck_geometric_consistency_geomean

(args, depth_ref, intrinsics_ref, extrinsics_ref, depth_src_up,depth_src,depth_src_dn, intrinsics_src, extrins

filter/dypcd_tanks.py:328

Functioncheck_geometric_consistency_pytorch

(depth_ref, intrinsics_ref, extrinsics_ref, depth_src, intrinsics_src, extrinsics_src,alpha=1.0)

filter/pcd.py:203

Functionclassification_loss

(prob_volume, depth_values, interval, depth_gt, mask, weight)

Functioncompute_metrics_for_each_image

Functioncv2_imread

datasets/data_io.py:110

Functionentropy_loss

(prob_volume, depth_gt, mask, depth_value, return_prob_map=False)

Functionentropy_loss_expand

(prob_volume, depth_gt, mask, depth_value, return_prob_map=False)

(self, cost_reg, depth_values, num_depth, interval, prob_volume_init=None,stage=0)

networks/mvsnet.py:15

:param stage_idx: stage :param features: [ref_fea, src_fea1, src_fea2, ...], fea shape: (b, c, h, w) :param proj_matrices: (b

networks/mvsnet.py:111

:param is_flip: augment only for 3D-UNet :param imgs: (b, nview, c, h, w) :param proj_matrices: :param depth_values:

networks/mvsnet.py:188

networks/module.py:57

networks/module.py:102

networks/module.py:151

networks/module.py:196

(self, x_pre, x)

networks/module.py:267

networks/module.py:316

networks/module.py:347

networks/module.py:355

networks/module.py:389

(self, x,stage=0)

networks/module.py:426

:param x: (b, c, d, h, w) :return: (b, 1, d, h, w)

networks/module.py:445

Functiongenerate_pointcloud

Generate a colored point cloud in PLY format from a color and a depth image. Input: rgb_file -- filename of color image depth_file -

Functionget_cur_depth_range_samples_inverse

(last_depth, ndepth, depth_inteval_pixel)

networks/module.py:509

Methodinit_weights

default initialization

networks/module.py:65

Methodinit_weights

default initialization

networks/module.py:113

Methodinit_weights

default initialization

networks/module.py:159

Methodinit_weights

default initialization

networks/module.py:204

Functioninit_worker

Catch Ctrl+C signal to termiante workers

filter/dypcd_tanks.py:385

Functioninit_worker

Catch Ctrl+C signal to termiante workers

filter/pcd.py:381

Functionmake_recursive_func

Functionmvs_loss

(inputs, depth_gt_ms, mask_ms, mode, **kwargs)

Methodread_depth

(self, filename)

datasets/dtu_yao.py:111

Methodread_depth

(self, filename)

datasets/general_eval.py:93

Methodread_mask

(self, filename)

datasets/blendedmvs.py:102

Functionreproject_with_depth

(depth_ref, intrinsics_ref, extrinsics_ref, depth_src, intrinsics_src, extrinsics_src)

filter/pcd.py:96

Methodscale_depth_input

(self,img,max_w, max_h, base=global_base,interp=cv2.INTER_NEAREST)

datasets/general_eval.py:115

Functionsetup_seed

Functionunified_step_focal_loss

(prob_volume, depth_values, interval, depth_gt, mask, weight, gamma, alpha)

Functionwinner_take_all

:param prob_volume: (b, d, h, w) :param depth_values: (b, d, h, w) :return: (b, h, w)

networks/module.py:463

Functionwrapper

← previous101–151 of 151, ranked by callers