unik3d/datasets/__init__.py

from ._2d3ds import d2D3DS
from ._4dor import d4DOR
from .a2d2 import A2D2
from .adt import ADT
from .aimotive import aiMotive
from .argoverse import Argoverse
from .argoverse2 import Argoverse2
from .arkit import ARKit
from .ase import ASE
from .base_dataset import BaseDataset
from .bdd import BDD
from .bedlam import BEDLAM
from .behave import Behave
from .blendedmvg import BlendedMVG
from .cityscape import Cityscape
from .ddad import DDAD
from .deep360 import Deep360
from .dense import DENSE
from .diml import DIML
from .diode import DiodeIndoor, DiodeIndoor_F
from .dl3dv import DL3DV
from .driving_stereo import DrivingStereo
from .dtu_rmvd import DTURMVD
from .dummy import Dummy
from .dynamic_replica import DynReplica
from .eden import EDEN
from .eth3d import ETH3D, ETH3D_F, ETH3DRMVD
from .facedepth import FaceDepth
from .flsea import FLSea
from .futurehouse import FutureHouse
from .gibson import Gibson
from .hammer import HAMMER
from .hm3d import HM3D
from .hoi4d import HOI4D
from .hypersim import HyperSim
from .ibims import IBims, IBims_F
from .ken_burns import KenBurns
from .kitti import KITTI, KITTIRMVD, KITTIBenchmark
from .kitti360 import KITTI360
from .lyft import Lyft
from .mapillary import Mapillary
from .matrix_city import MatrixCity
from .matterport3d import Matterport3D
from .megadepth import MegaDepth
from .megadepth_s import MegaDepthS
from .midair import MidAir
from .mip import MIP
from .ms2 import MS2
from .mvimgnet import MVImgNet
from .mvsynth import MVSynth
from .nerds360 import NeRDS360
from .niantic_mapfree import NianticMapFree
from .nuscenes import Nuscenes
from .nyuv2 import NYUv2Depth
from .point_odyssey import PointOdyssey
from .proteus import Proteus
from .samplers import (DistributedSamplerNoDuplicate,
                       DistributedSamplerWrapper, ShardedInfiniteSampler)
from .scannet import ScanNet
from .scannetpp import ScanNetpp, ScanNetpp_F
from .sintel import Sintel
from .sunrgbd import SUNRGBD
from .synscapes import Synscapes
from .tartanair import TartanAir
from .taskonomy import Taskonomy
from .tat_rmvd import TATRMVD
from .theo import Theo
from .unrealstereo4k import UnrealStereo4K
from .urbansyn import UrbanSyn
from .utils import ConcatDataset, collate_fn, get_weights
from .vkitti import VKITTI
from .void import VOID
from .waymo import Waymo
from .wildrgbd import WildRGBD

__all__ = [
    "Dummy",
    "BaseDataset",
    "get_weights" "DistributedSamplerNoDuplicate",
    "ShardedInfiniteSampler",
    "DistributedSamplerWrapper",
    "ConcatDataset",
    "PairDataset",
    "collate_fn",
    # additional, do not count
    "WaymoImage",
    "MegaDepth",
    "COCO2017",
    "ImageNet",
    "OASISv2",
    # image based
    "Argoverse",
    "DDAD",
    "IBims",
    "NYUv2Depth",
    "DrivingStereo",
    "VOID",
    "Mapillary",
    "ScanNet",
    "Taskonomy",
    "BDD",
    "A2D2",
    "Nuscenes",
    "SUNRGBD",
    "ETH3D",
    "HAMMER",
    "Cityscape",
    "KITTI",
    "DENSE",
    "DIML",
    "DiodeIndoor",
    "FLSea",
    "ARKitScenes",
    "Lyft",
    "HyperSim",
    "KenBurns",
    "HRWSI",
    "UrbanSyn",
    "Synscapes",
    "Gibson",
    "Matterport3D",
    "_2D3DS",
    # sequence based
    "TartanAir",
    "WildRGBD",
    "ScanNetS",
    "ScanNetpp",
    "MVImgNet",
    "NianticMapFree",
    "DL3DV",
    "PointOdyssey",
    "KITTIMulti",
    "Waymo",
    "Argoverse2",
    "UnrealStereo4K",
    "MatrixCity",
    "HM3D",
    "MVSynth",
    "EDEN",
    # sequence based, but not usable for seq, only image
    "BEDLAM",
    "NeRDS360",
    "BlendedMVG",
    "DynReplica",
    "ARKitS",
    "Sintel",
    "VKITTI",
    "MegaDepthS",
    # benchmarks
    "KITTIBenchmark",
    "ETH3DRMVD",
    "DTURMVD",
    "KITTIRMVD",
    "TATRMVD",
    "DiodeIndoor_F",
    "IBims_F",
    "ETH3D_F",
    "KITTI360",
    "ScanNetpp_F",
    "ADT",
]