Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Wang-xjtu/G2-MonoDepth
/ functions
Functions
69 in github.com/Wang-xjtu/G2-MonoDepth
⨍
Functions
69
◇
Types & classes
19
↓ 6 callers
Function
save_img
( data: Tensor, file_name: Path, )
src/utils.py:43
↓ 2 callers
Method
__init__
(self)
src/losses.py:8
↓ 2 callers
Method
__init__
( self, mid_module: Module, in_channel: int, mid_channel:
src/modules.py:47
↓ 2 callers
Method
_sample
(raw, zero_rate)
src/data_tools.py:161
↓ 2 callers
Function
min_max_norm
(depth)
src/utils.py:36
↓ 2 callers
Method
ordinal_label
(self, sampled_a, sampled_b)
test_utils.py:110
↓ 2 callers
Method
train
(self, cf)
src/src_main.py:123
↓ 1 callers
Method
__gethole__
(hole_dir)
src/data_tools.py:252
↓ 1 callers
Method
__getrgbd__
(rgbd_dir)
src/data_tools.py:239
↓ 1 callers
Method
__gradient_loss__
(self, residual: Tensor)
src/losses.py:54
↓ 1 callers
Method
__init__
(self)
test_utils.py:25
↓ 1 callers
Method
__init__
(self, factor)
src/data_tools.py:73
↓ 1 callers
Method
__min_max_norm__
(depth)
test_utils.py:51
↓ 1 callers
Method
_holes
(self, raw, hole_ls, p=0.8)
src/data_tools.py:132
↓ 1 callers
Method
_noiseblur
(self, raw, hole_gt, p_noise=0.5, p_blur=0.5)
src/data_tools.py:173
↓ 1 callers
Method
absRel
(depth, ground_truth)
test_utils.py:74
↓ 1 callers
Method
adjust_domain
(self, pred)
test_utils.py:57
↓ 1 callers
Method
cleanup
()
src/utils.py:81
↓ 1 callers
Function
demo_metric
(args)
test.py:62
↓ 1 callers
Function
demo_save
(args)
test.py:39
↓ 1 callers
Function
depth_read
(filename)
test_utils.py:15
↓ 1 callers
Function
depth_read
(filename: Path)
src/data_tools.py:22
↓ 1 callers
Method
error
(self, pred, gt)
test_utils.py:144
↓ 1 callers
Method
feedback_module
(self, elapsed, iter_step, loss, summary, global_step)
src/src_main.py:101
↓ 1 callers
Function
get_dataloader
(rgbd_dirs, hole_dirs, batch_size, sizes, rank, num_workers)
src/data_tools.py:279
↓ 1 callers
Function
hole_read
(filename: Path)
src/data_tools.py:31
↓ 1 callers
Method
oe
(depth, ground_truth)
test_utils.py:98
↓ 1 callers
Method
optimize_one_iteration
(self, rgb, gt, raw)
src/src_main.py:75
↓ 1 callers
Function
parse_arguments
()
test.py:18
↓ 1 callers
Function
print_model_parm_nums
(model)
src/utils.py:85
↓ 1 callers
Method
read_data
(self, rgb_path, raw_path)
test_utils.py:40
↓ 1 callers
Method
read_rgbraw
(rgb_path, raw_path)
test_utils.py:29
↓ 1 callers
Function
rgb_read
(filename)
test_utils.py:7
↓ 1 callers
Function
rgb_read
(filename: Path)
src/data_tools.py:14
↓ 1 callers
Method
rmse
(depth, ground_truth)
test_utils.py:67
↓ 1 callers
Method
run_demo
(demo_fn, world_size)
src/utils.py:75
↓ 1 callers
Method
sampling
(self, pred, gt)
test_utils.py:117
↓ 1 callers
Method
save_imgs
(rgb, gt, raw, pred, log_dir, epoch, iter_step)
src/src_main.py:110
↓ 1 callers
Method
setup
(rank, world_size, port)
src/utils.py:62
↓ 1 callers
Method
srmse
(depth, ground_truth)
test_utils.py:81
↓ 1 callers
Method
trans_raw
(self, gt, hole_gt, hole_ls)
src/data_tools.py:139
↓ 1 callers
Method
trans_rgbgt
(self, rgb: Tensor, gt: Tensor)
src/data_tools.py:122
Function
DDP_main
(rank, world_size)
train.py:14
Method
__getitem__
(self, item)
src/data_tools.py:264
Method
__init__
(self, gpus_num)
config.py:5
Method
__init__
(self, t=0.01)
test_utils.py:105
Method
__init__
(self, cf, rank)
src/src_main.py:21
Method
__init__
(self)
src/utils.py:12
Method
__init__
( self, in_channels: int, out_channels: int, rezero: bool,
src/custom_blocks.py:7
Method
__init__
( self, )
src/losses.py:24
Method
__init__
( self, k: int = 4, sobel: bool = True, )
src/losses.py:40
Method
__init__
( self, in_channels: int, out_channels: int, rezero: bool,
src/modules.py:8
Method
__init__
( self, in_channel: int, out_channel: int, rezero: bool,
src/modules.py:28
Method
__init__
(self, layer_num: int = 7, rezero: bool = True)
src/networks.py:10
Method
__init__
(self, size)
src/data_tools.py:94
Method
__init__
(self, rgbd_dir, hole_dir, size)
src/data_tools.py:204
Method
__len__
(self)
src/data_tools.py:258
Method
__masked_mean_robust_standardization__
(depth, mask, eps=1e-6)
src/utils.py:17
Method
forward
(self, depth, gt, mask_hole)
src/utils.py:28
Method
forward
(self, x: Tensor)
src/custom_blocks.py:44
Method
forward
(self, x: Tensor)
src/losses.py:17
Method
forward
(self, output: Tensor, target: Tensor, hole: Tensor)
src/losses.py:31
Method
forward
(self, output: Tensor, target: Tensor, hole_target: Tensor)
src/losses.py:66
Method
forward
(self, x: Tensor)
src/modules.py:22
Method
forward
(self, x: Tensor)
src/modules.py:41
Method
forward
(self, x: Tensor)
src/modules.py:63
Method
forward
( self, rgb: Tensor, point: Tensor, hole_point: Tensor,
src/networks.py:31
Method
forward
Args: img (PIL Image or Tensor): Image to be cropped and resized. Returns: PIL Image or Tensor: Randoml
src/data_tools.py:40
Method
forward
Args: depth (Tensor): depth map to be scaled and shifted. Returns: Tensor: scaled and shifted depth.
src/data_tools.py:77