Functions — github.com/autonomousvision/unimatch

CodeHub MCPcopy Create free account

hub / github.com/autonomousvision/unimatch / functions

Functions377 in github.com/autonomousvision/unimatch

⨍ Functions 377◇ Types & classes 110

(self, aug_params=None, split='training', root='datasets/Sintel', dstype='

dataloader/flow/datasets.py:142

(self, aug_params=None, split='train', root='datasets/FlyingChairs_release/data',

dataloader/flow/datasets.py:174

(self, aug_params=None, root='datasets/FlyingThings3D', dstype='frames_cle

dataloader/flow/datasets.py:193

(self, aug_params=None, root='datasets/VKITTI2', )

dataloader/flow/datasets.py:243

(self, aug_params=None, split='training', root='datasets/KITTI', )

dataloader/flow/datasets.py:279

(self, aug_params=None, split='training', root='datasets/KITTI12')

dataloader/flow/datasets.py:301

(self, transforms)

dataloader/stereo/transforms.py:11

(self, no_normalize=False)

dataloader/stereo/transforms.py:23

(self, mean, std)

dataloader/stereo/transforms.py:50

(self, img_height, img_width)

dataloader/stereo/transforms.py:67

(self, asymmetric_color_aug=True, )

dataloader/stereo/transforms.py:155

(self, asymmetric_color_aug=True, )

dataloader/stereo/transforms.py:177

(self, asymmetric_color_aug=True, )

dataloader/stereo/transforms.py:199

(self, asymmetric_color_aug=True, )

dataloader/stereo/transforms.py:221

(self, asymmetric_color_aug=True, )

dataloader/stereo/transforms.py:243

(self, asymmetric_color_aug=True, )

dataloader/stereo/transforms.py:265

(self, min_scale=-0.4, max_scale=0.4, crop_width=512,

dataloader/stereo/transforms.py:297

Resize low-resolution data to high-res for mixed dataset training

dataloader/stereo/transforms.py:333

dataloader/stereo/transforms.py:361

dataloader/stereo/transforms.py:379

(self, p=0.5, occlusion_mask_zero=False)

dataloader/stereo/transforms.py:410

(self, transform=None, is_vkitti2=False, is_sintel=False

dataloader/stereo/datasets.py:16

(self, data_dir='datasets/FlyingThings3D', mode='TRAIN',

dataloader/stereo/datasets.py:94

(self, data_dir='datasets/Monkaa', split='frames_finalpass',

dataloader/stereo/datasets.py:117

(self, data_dir='datasets/Driving', split='frames_finalpass',

dataloader/stereo/datasets.py:138

(self, data_dir='datasets/KITTI/stereo/kitti_2015', mode='training',

dataloader/stereo/datasets.py:159

(self, data_dir='datasets/KITTI/stereo/kitti_2012', mode='training',

dataloader/stereo/datasets.py:194

(self, data_dir='datasets/VKITTI2', transform=None, )

dataloader/stereo/datasets.py:226

(self, data_dir='datasets/DrivingStereo', transform=None,

dataloader/stereo/datasets.py:248

(self, data_dir='datasets/SintelStereo', split='clean',

dataloader/stereo/datasets.py:273

(self, data_dir='datasets/ETH3D', mode='train', transfor

dataloader/stereo/datasets.py:307

(self, data_dir='datasets/Middlebury/MiddEval3', mode='training',

dataloader/stereo/datasets.py:348

(self, data_dir='datasets/Middlebury/2005', transform=None,

dataloader/stereo/datasets.py:389

(self, data_dir='datasets/Middlebury/2014', transform=None,

dataloader/stereo/datasets.py:433

(self, data_dir='datasets/Middlebury/2021/data', transform=None,

dataloader/stereo/datasets.py:462

(self, data_dir='datasets/CREStereo/stereo_trainset/crestereo', transform=

dataloader/stereo/datasets.py:514

(self, data_dir='datasets/Tartanair', transform=None, )

dataloader/stereo/datasets.py:538

(self, data_dir='datasets/InStereo2K', transform=None, )

dataloader/stereo/datasets.py:587

(self, data_dir='datasets/FallingThings', transform=None,

dataloader/stereo/datasets.py:612

(self, transforms)

dataloader/depth/augmentation.py:10

(self, mean, std)

dataloader/depth/augmentation.py:39

(self, crop_size)

dataloader/depth/augmentation.py:67

(self, asymmetric=True)

dataloader/depth/augmentation.py:96

(self, min_size, min_scale=-0.2, max_scale=0.2)

dataloader/depth/augmentation.py:125

(self, asymmetric=False)

dataloader/depth/augmentation.py:198

(self, asymmetric=False)

dataloader/depth/augmentation.py:216

(self, asymmetric=False)

dataloader/depth/augmentation.py:234

(self, asymmetric=False)

dataloader/depth/augmentation.py:252

(self, asymmetric=False)

dataloader/depth/augmentation.py:270

(self, data_dir='datasets/Scannet', mode='train', transform

dataloader/depth/datasets.py:12

(self, data_dir='datasets/Demon', mode='train', transforms=

dataloader/depth/datasets.py:121

Args: scale_factors (list[float]): list of scaling factors to upsample or downsample the input features for creat

unimatch/vit_fpn.py:13

Init. Args: features (int): number of features

unimatch/dpt_head.py:65

Init. Args: features (int): number of features

unimatch/dpt_head.py:134

( self, num_scales=1, feature_channels=128, upsample_factor=8, lowest_

unimatch/unimatch_depthsplat.py:19

( self, in_channels, out_channels, kernel_size,

unimatch/trident_conv.py:11

(self, in_channels, **kwargs, )

unimatch/attention.py:172

(self, d_model=128, nhead=1, ffn_dim_expansion=4,

unimatch/transformer.py:150

(self, num_layers=6, d_model=128, nhead=1,

unimatch/transformer.py:204

(self, hidden_dim=128, input_dim=192 + 128, kernel_size=5, )

unimatch/reg_refine.py:23

(self, corr_channels=324, flow_channels=2, )

unimatch/reg_refine.py:56

(self, corr_channels=324, hidden_dim=128, context_dim=128,

unimatch/reg_refine.py:79

(self, num_pos_feats=64, temperature=10000, normalize=True, scale=None)

unimatch/position.py:15

(self, num_scales=1, feature_channels=128, upsample_fact

unimatch/unimatch.py:17

(self, in_planes, planes, norm_layer=nn.InstanceNorm2d, stride=1, dilation=1, )

unimatch/backbone.py:7

( self, spacial_dim: int, embed_dim: int, num_heads_channels: int, out

unimatch/ldm_unet/unet.py:41

(self, channels, out_channels=None, ks=5)

unimatch/ldm_unet/unet.py:131

(self, channels, use_conv, dims=2, out_channels=None,padding=1, downsample_3ddim=False, # downsample all

unimatch/ldm_unet/unet.py:151

( self, channels, emb_channels, dropout, out_channels=None, us

unimatch/ldm_unet/unet.py:194

( self, channels, num_heads=1, num_head_channels=-1, use_checkpoint=Fa

unimatch/ldm_unet/unet.py:319

( self, channels, condition_channels, num_heads=8, proj_channels=512,

unimatch/ldm_unet/unet.py:394

(self, n_heads, n_frames=2, use_cross_view_self_attn=False)

unimatch/ldm_unet/unet.py:536

(self, n_heads)

unimatch/ldm_unet/unet.py:581

( self, image_size, in_channels, model_channels, out_channels,

unimatch/ldm_unet/unet.py:640

(self, in_channels, model_channels, out_channels,

unimatch/ldm_unet/unet.py:1137

(self, in_dim1, in_dim2, dim=128, out_dim=None, num_heads=4,

unimatch/ldm_unet/cross_attention.py:65

(self, dim, dim_out=None, mult=4, glu=False, dropout=0.)

unimatch/ldm_unet/attention.py:46

(self, dim, heads=4, dim_head=32)

unimatch/ldm_unet/attention.py:79

(self, in_channels)

unimatch/ldm_unet/attention.py:98

(self, query_dim, context_dim=None, heads=8, dim_head=64, dropout=0.)

unimatch/ldm_unet/attention.py:151

(self, dim, n_heads, d_head, dropout=0., context_dim=None, gated_ff=True, checkpoint=False)

unimatch/ldm_unet/attention.py:195

(self, in_channels, n_heads, d_head, depth=1, dropout=0., context_dim=None)

unimatch/ldm_unet/attention.py:226

dataloader/flow/datasets.py:137

dataloader/stereo/datasets.py:84

dataloader/depth/datasets.py:99

dataloader/depth/datasets.py:199

dataloader/flow/datasets.py:131

dataloader/stereo/datasets.py:87

dataloader/depth/datasets.py:114

(self, x, emb=None)

unimatch/ldm_unet/unet.py:289

(self, x, context=None)

unimatch/ldm_unet/attention.py:211

Functionbad_pixel_metric

(d_est, d_gt, mask, abs_threshold=10, rel_threshold=0.1,

loss/stereo_metric.py:31

Functionbetas_for_alpha_bar

Create a beta schedule that discretizes the given alpha_t_bar function, which defines the cumulative product of (1-beta) over time from t = [

unimatch/ldm_unet/util.py:77

Functionbilinear_upflow

(flow, scale_factor=8)

utils/utils.py:133

Functionconvert_module_to_f16

unimatch/ldm_unet/unet.py:28

Functionconvert_module_to_f32

unimatch/ldm_unet/unet.py:31

Methodconvert_to_fp16

Convert the torso of the model to float16.

unimatch/ldm_unet/unet.py:1054

Methodconvert_to_fp32

Convert the torso of the model to float32.

unimatch/ldm_unet/unet.py:1062

Functioncoords_grid_np

utils/utils.py:71

Methodcount_flops

unimatch/ldm_unet/unet.py:572

← previous next →201–300 of 377, ranked by callers