MCPcopy Create free account

hub / github.com/M2219/ESMStereo / functions

Functions258 in github.com/M2219/ESMStereo

Method__getitem__
(self, index)
datasets/ETH3D_loader.py:90
Method__getitem__
(self, index)
datasets/kitti_dataset.py:50
Method__getitem__
(self, index)
datasets/sceneflow_dataset.py:44
Method__getitem__
(self, index)
datasets/KITTI2012loader.py:94
Method__getitem__
(self, index)
datasets/KITTIRawLoader.py:56
Method__init__
(self)
train_kitti.py:240
Method__init__
(self)
train_sceneflow.py:280
Method__init__
(self)
utils/experiment.py:129
Method__init__
(self)
utils/experiment.py:143
Method__init__
( self, left, right, left_disp, training, imgloader=img_loader
datasets/KITTI2015loader.py:71
Method__init__
( self, left, right, left_disp, training, imgloader=img_loader
datasets/middlebury_loader.py:56
Method__init__
( self, left, right, disp_gt, occ_mask, training, imgl
datasets/ETH3D_loader.py:66
Method__init__
( self, datapath_12: str, datapath_15: str, list_filename: str, training: bool )
datasets/kitti_dataset.py:16
Method__init__
(self, datapath: str, list_filename: str, training: bool)
datasets/sceneflow_dataset.py:18
Method__init__
( self, left, right, left_disp, training, imgloader=img_loader
datasets/KITTI2012loader.py:78
Method__init__
(self, left, right, imgloader=img_loader)
datasets/KITTIRawLoader.py:46
Method__init__
(self, co_transforms)
datasets/flow_transforms.py:13
Method__init__
(self, size, order=2)
datasets/flow_transforms.py:25
Method__init__
(self, size)
datasets/flow_transforms.py:56
Method__init__
(self, angle, px, diff_angle=0, order=2, reshape=False)
datasets/flow_transforms.py:80
Method__init__
(self, in_channels: int, out_channels: int, deconv: bool=False, is_3d: bool=False, bn: bool=True, gelu: bool=T
models/submodule.py:14
Method__init__
(self, inplanes, planes, stride, downsample, pad, dilation)
models/submodule.py:43
Method__init__
(self)
models/ESMStereo.py:22
Method__init__
(self, backbone: str)
models/ESMStereo.py:41
Method__init__
(self, in_channels: int, add_channel: int)
models/ESMStereo.py:130
Method__init__
(self, C: int, cf1: int, cf2: int)
models/ESMStereo.py:186
Method__init__
(self)
models/ESMStereo.py:243
Method__init__
(self)
models/ESMStereo.py:321
Method__init__
(self)
models/ESMStereo.py:431
Method__init__
(self, maxdisp: int, gwc: bool=False, norm_correlation: bool=True, backbone: str="efficientnet_b2", cv_scale
models/ESMStereo.py:512
Method__init__
(self)
models/ESMStereo_confidence.py:22
Method__init__
(self, backbone: str)
models/ESMStereo_confidence.py:41
Method__init__
(self, in_channels: int, add_channel: int)
models/ESMStereo_confidence.py:130
Method__init__
(self, C: int, cf1: int, cf2: int)
models/ESMStereo_confidence.py:186
Method__init__
(self)
models/ESMStereo_confidence.py:243
Method__init__
(self)
models/ESMStereo_confidence.py:321
Method__init__
(self)
models/ESMStereo_confidence.py:431
Method__init__
(self, C: int, fc: int)
models/ESMStereo_confidence.py:512
Method__init__
(self, C)
models/ESMStereo_confidence.py:552
Method__init__
(self, maxdisp: int, gwc: bool=False, norm_correlation: bool=True, backbone: str="efficientnet_b2", cv_scale
models/ESMStereo_confidence.py:747
Method__init__
(self)
models/ESMStereo_trt.py:22
Method__init__
(self, backbone: str)
models/ESMStereo_trt.py:41
Method__init__
(self, in_channels: int, add_channel: int)
models/ESMStereo_trt.py:130
Method__init__
(self, C: int, cf1: int, cf2: int)
models/ESMStereo_trt.py:186
Method__init__
(self)
models/ESMStereo_trt.py:243
Method__init__
(self)
models/ESMStereo_trt.py:321
Method__init__
(self)
models/ESMStereo_trt.py:431
Method__init__
(self, maxdisp: int, gwc: bool=False, norm_correlation: bool=True, backbone: str="efficientnet_b2", cv_scale
models/ESMStereo_trt.py:512
Method__init__
(self, dim, mlp_ratio=4)
models/shufflemixer.py:11
Method__init__
(self, dim: int, mlp_ratio: int=2)
models/shufflemixer.py:24
Method__init__
(self, normalized_shape: int)
models/shufflemixer.py:48
Method__init__
(self, normalized_shape)
models/shufflemixer.py:65
Method__init__
(self, dim: int, LayerNorm_type: str='BiasFree')
models/shufflemixer.py:84
Method__init__
(self, dim: int, kernel_size: int, mlp_ratio: int=2)
models/shufflemixer.py:117
Method__init__
(self, n_feats, kernel_size, n_blocks, mlp_ratio, upscaling_factor)
models/shufflemixer.py:145
Method__len__
(self)
datasets/KITTI2015loader.py:132
Method__len__
(self)
datasets/middlebury_loader.py:130
Method__len__
(self)
datasets/ETH3D_loader.py:134
Method__len__
(self)
datasets/kitti_dataset.py:47
Method__len__
(self)
datasets/sceneflow_dataset.py:41
Method__len__
(self)
datasets/KITTI2012loader.py:133
Method__len__
(self)
datasets/KITTIRawLoader.py:73
Functionbuild_concat_volume
(refimg_fea, targetimg_fea, maxdisp)
models/submodule.py:129
Functionbuild_gwc_volume_gather
(ref: torch.Tensor, tgt: torch.Tensor, maxdisp: int,
models/submodule.py:245
Functionbuild_gwc_volume_norm
(refimg_fea, targetimg_fea, maxdisp, num_groups)
models/submodule.py:174
Functionbuild_substract_volume
(refimg_fea, targetimg_fea, maxdisp, num_groups)
models/submodule.py:116
FunctioncomputeD1
virtual_kitti_publisher/src/virtual_kitti_publisher_cuda_node.cpp:95
FunctioncomputeD1
kitti_publisher_ess/src/kitti_publisher_ess_cuda_node.cpp:106
Functioncompute_metric_for_each_image
(metric_func: Callable)
utils/metrics.py:17
Functioncontext_upsample
(depth_low, up_weights)
models/submodule.py:202
Functioncount_parameters_in_MB
(model)
models/ESMStereo.py:18
Functioncount_parameters_in_MB
(model)
models/ESMStereo_confidence.py:18
Functioncount_parameters_in_MB
(model)
models/ESMStereo_trt.py:18
Functiondataloader
(filepath)
datasets/listfiles.py:10
Functiondisparity_loader
(path)
datasets/KITTI2015loader.py:65
Functiondisparity_loader
(path)
datasets/middlebury_loader.py:50
Functiondisparity_loader
(path)
datasets/ETH3D_loader.py:60
Functiondisparity_loader
(path)
datasets/KITTI2012loader.py:64
Functiondisparity_loader
(path)
datasets/KITTIRawLoader.py:40
Functionet_loader
(filepath: str)
datasets/ETH3D_loader.py:29
Methodforward
(self, x: torch.Tensor)
models/submodule.py:32
Methodforward
(self, x)
models/submodule.py:52
Methodforward
(self, x: torch.Tensor, rem: torch.Tensor)
models/submodule.py:91
Methodforward
(self, x: torch.Tensor)
models/ESMStereo.py:68
Methodforward
(self, featL: List[ torch.Tensor], featR: Optional[List[ torch.Tensor]]=None)
models/ESMStereo.py:100
Methodforward
(self, x: torch.Tensor)
models/ESMStereo.py:165
Methodforward
(self, disp: torch.Tensor, left_f1x: torch.Tensor, left_f2x: torch.Tensor)
models/ESMStereo.py:221
Methodforward
(self, left_f1x: torch.Tensor, left_f2x: torch.Tensor, left_f4x: torch.Tensor, init_disp: torch.Tensor)
models/ESMStereo.py:296
Methodforward
(self, left_f2x, left_f4x, left_f8x, stem_f2, init_disp)
models/ESMStereo.py:396
Methodforward
(self, left_f1x: torch.Tensor, left_f2x: torch.Tensor, left_f4x: torch.Tensor, left_f8x: torch.Tensor, init_di
models/ESMStereo.py:484
Methodforward
(self, left: torch.Tensor, right: torch.Tensor, train_status: bool)
models/ESMStereo.py:638
Methodforward
(self, x: torch.Tensor)
models/ESMStereo_confidence.py:68
Methodforward
(self, featL: List[ torch.Tensor], featR: Optional[List[ torch.Tensor]]=None)
models/ESMStereo_confidence.py:100
Methodforward
(self, x: torch.Tensor)
models/ESMStereo_confidence.py:165
Methodforward
(self, disp: torch.Tensor, left_f1x: torch.Tensor, left_f2x: torch.Tensor)
models/ESMStereo_confidence.py:221
Methodforward
(self, left_f1x: torch.Tensor, left_f2x: torch.Tensor, left_f4x: torch.Tensor, init_disp: torch.Tensor)
models/ESMStereo_confidence.py:296
Methodforward
(self, left_f2x, left_f4x, left_f8x, stem_f2, init_disp)
models/ESMStereo_confidence.py:396
Methodforward
(self, left_f1x: torch.Tensor, left_f2x: torch.Tensor, left_f4x: torch.Tensor, left_f8x: torch.Tensor, init_di
models/ESMStereo_confidence.py:484
Methodforward
(self, left_f1x: torch.Tensor, init_conf: torch.Tensor)
models/ESMStereo_confidence.py:532
Methodforward
(self, cost, disp, imag, left_f1x, left_f2x, device)
models/ESMStereo_confidence.py:651
← previousnext →101–200 of 258, ranked by callers