MCPcopy Create free account

hub / github.com/baegwangbin/surface_normal_uncertainty / functions

Functions58 in github.com/baegwangbin/surface_normal_uncertainty

⨍ Functions 58◇ Types & classes 14
↓ 8 callersFunctionnorm_normalize
(norm_out)
models/submodules/submodules.py:64
↓ 3 callersFunctiondownload_file_from_google_drive
(id, destination)
download.py:6
↓ 3 callersFunctionsample_points
(init_normal, gt_norm_mask, sampling_ratio, beta)
models/submodules/submodules.py:75
↓ 2 callersMethod__init__
(self, args=None)
models/baseline.py:10
↓ 2 callersMethod__init__
(self, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bia
models/submodules/submodules.py:47
↓ 2 callersMethod__len__
(self)
data/dataloader_nyu.py:59
↓ 2 callersFunctionvalidate
(model, args, test_loader, device, total_iter, where_to_write, vis_dir=None)
train.py:145
↓ 1 callersMethodget_10x_lr_params
(self)
models/NNET.py:18
↓ 1 callersMethodget_1x_lr_params
(self)
models/NNET.py:15
↓ 1 callersFunctionget_confirm_token
(response)
download.py:7
↓ 1 callersFunctionkappa_to_alpha
(pred_kappa)
utils/utils.py:123
↓ 1 callersFunctionmain_worker
(gpu, ngpus_per_node, args)
train.py:186
↓ 1 callersFunctionmakedir
(dirpath)
utils/utils.py:94
↓ 1 callersFunctionnorm_to_rgb
(norm)
utils/utils.py:131
↓ 1 callersFunctionsave_response_content
(response, destination)
download.py:13
↓ 1 callersFunctiontest
(model, test_loader, device, results_dir)
test.py:19
↓ 1 callersFunctiontrain
(model, args, device)
train.py:19
↓ 1 callersFunctionunnormalize
(img_in)
utils/utils.py:113
Method__getitem__
(self, idx)
data/dataloader_nyu.py:62
Method__getitem__
(self, idx)
data/dataloader_custom.py:31
Method__init__
args.loss_fn can be one of following: - L1 - L1 loss (no uncertainty) - L2 - L2 loss (no uncertainty)
utils/losses.py:9
Method__init__
mode: {'train_big', # training set used by GeoNet (CVPR18, 30907 images) 'train', # official train set (795 images)
data/dataloader_nyu.py:19
Method__init__
(self, args, mode)
data/dataloader_nyu.py:48
Method__init__
(self, args, fldr_path)
data/dataloader_custom.py:12
Method__init__
(self, args, fldr_path)
data/dataloader_custom.py:21
Method__init__
(self, args)
models/NNET.py:10
Method__init__
(self)
models/baseline.py:36
Method__init__
(self, num_classes=4)
models/baseline.py:64
Method__init__
(self, args)
models/submodules/decoder.py:8
Method__init__
(self)
models/submodules/encoder.py:7
Method__init__
(self, skip_input, output_features)
models/submodules/submodules.py:11
Method__init__
(self, skip_input, output_features)
models/submodules/submodules.py:29
Method__len__
(self)
data/dataloader_custom.py:28
Function_unnormalize
(img_in)
train.py:136
Functioncolor_augmentation
color augmentation
data/utils.py:18
Functioncompute_normal_errors
(total_normal_errors)
utils/utils.py:63
Functionconcat_image
(image_path_list, concat_image_path)
utils/utils.py:31
Functionconvert_arg_line_to_args
(arg_line)
utils/utils.py:16
Methodforward
(self, *args)
utils/losses.py:28
Methodforward
(self, img, **kwargs)
models/NNET.py:21
Methodforward
(self, x, **kwargs)
models/baseline.py:15
Methodforward
(self, x)
models/baseline.py:51
Methodforward
(self, features)
models/baseline.py:73
Methodforward
(self, features, gt_norm_mask=None, mode='test')
models/submodules/decoder.py:59
Methodforward
(self, x)
models/submodules/encoder.py:22
Methodforward
(self, x, concat_with)
models/submodules/submodules.py:21
Methodforward
(self, x, concat_with)
models/submodules/submodules.py:39
Methodforward
(self, x)
models/submodules/submodules.py:52
Methodforward_R
(self, norm_out, gt_norm, gt_norm_mask)
utils/losses.py:31
Methodforward_UG
(self, pred_list, coord_list, gt_norm, gt_norm_mask)
utils/losses.py:91
Methodget_10x_lr_params
(self)
models/baseline.py:28
Methodget_1x_lr_params
(self)
models/baseline.py:25
Functionload_checkpoint
(fpath, model)
utils/utils.py:47
Functionlog_normal_errors
(metrics, where_to_write, first_line)
utils/utils.py:78
Functionmake_dir_from_list
(dirpath_list)
utils/utils.py:100
Functionrandom_crop
randomly crop the input image & surface normal
data/utils.py:5
Functionsave_args
(args, filename)
utils/utils.py:24
Functionvisualize
(args, img, gt_norm, gt_norm_mask, norm_out_list, total_iter)
utils/utils.py:140