SYMBOL INDEX (3758 symbols across 426 files) FILE: AutoReconForDens3R/nerfstudio/cameras/camera_optimizers.py class CameraOptimizerConfig (line 39) | class CameraOptimizerConfig(cfg.InstantiateConfig): class CameraOptimizer (line 64) | class CameraOptimizer(nn.Module): method __init__ (line 69) | def __init__( method forward (line 99) | def forward( FILE: AutoReconForDens3R/nerfstudio/cameras/camera_paths.py function get_interpolated_camera_path (line 30) | def get_interpolated_camera_path(cameras: Cameras, steps: int) -> Cameras: function get_spiral_path (line 48) | def get_spiral_path( function get_path_from_json (line 114) | def get_path_from_json(camera_path: Dict[str, Any]) -> Cameras: function get_path_from_npz (line 152) | def get_path_from_npz(camera_path: Dict[str, Any]) -> Cameras: FILE: AutoReconForDens3R/nerfstudio/cameras/camera_utils.py function unit_vector (line 30) | def unit_vector(data, axis: Optional[int] = None) -> np.ndarray: function quaternion_from_matrix (line 49) | def quaternion_from_matrix(matrix, isprecise: bool = False) -> np.ndarray: function quaternion_slerp (line 105) | def quaternion_slerp(quat0, quat1, fraction: float, spin: int = 0, short... function quaternion_matrix (line 139) | def quaternion_matrix(quaternion) -> np.ndarray: function get_interpolated_poses (line 161) | def get_interpolated_poses(pose_a, pose_b, steps: int = 10) -> List[float]: function get_interpolated_k (line 185) | def get_interpolated_k(k_a, k_b, steps: int = 10) -> TensorType[3, 4]: function get_interpolated_poses_many (line 202) | def get_interpolated_poses_many( function normalize (line 228) | def normalize(x) -> TensorType[...]: function viewmatrix (line 233) | def viewmatrix(lookat, up, pos) -> TensorType[...]: function get_distortion_params (line 252) | def get_distortion_params( function _compute_residual_and_jacobian (line 276) | def _compute_residual_and_jacobian( function radial_and_tangential_undistort (line 342) | def radial_and_tangential_undistort( function rotation_matrix (line 382) | def rotation_matrix(a: TensorType[3], b: TensorType[3]) -> TensorType[3,... function auto_orient_and_center_poses (line 410) | def auto_orient_and_center_poses( FILE: AutoReconForDens3R/nerfstudio/cameras/cameras.py class CameraType (line 36) | class CameraType(Enum): class Cameras (line 56) | class Cameras(TensorDataclass): method __init__ (line 89) | def __init__( method _init_get_fc_xy (line 145) | def _init_get_fc_xy(self, fc_xy, name): method _init_get_camera_type (line 166) | def _init_get_camera_type( method _init_get_height_width (line 208) | def _init_get_height_width( method _init_get_times (line 240) | def _init_get_times(self, times): method device (line 252) | def device(self): method image_height (line 257) | def image_height(self) -> TensorType["num_cameras":..., 1]: method image_width (line 262) | def image_width(self) -> TensorType["num_cameras":..., 1]: method is_jagged (line 267) | def is_jagged(self): method get_image_coords (line 276) | def get_image_coords( method generate_rays (line 304) | def generate_rays( # pylint: disable=too-many-statements method _generate_rays_from_coords (line 459) | def _generate_rays_from_coords( method to_json (line 698) | def to_json( method get_intrinsics_matrices (line 733) | def get_intrinsics_matrices(self) -> TensorType["num_cameras":..., 3, 3]: method rescale_output_resolution (line 747) | def rescale_output_resolution( FILE: AutoReconForDens3R/nerfstudio/cameras/lie_groups.py function exp_map_SO3xR3 (line 23) | def exp_map_SO3xR3(tangent_vector: TensorType["b", 6]) -> TensorType["b"... function exp_map_SE3 (line 61) | def exp_map_SE3(tangent_vector: TensorType["b", 6]) -> TensorType["b", 3... FILE: AutoReconForDens3R/nerfstudio/cameras/rays.py class Frustums (line 39) | class Frustums(TensorDataclass): method get_positions (line 55) | def get_positions(self) -> TensorType[..., 3]: method set_offsets (line 66) | def set_offsets(self, offsets): method get_start_positions (line 70) | def get_start_positions(self) -> TensorType[..., 3]: method get_gaussian_blob (line 84) | def get_gaussian_blob(self) -> Gaussians: method get_mock_frustum (line 103) | def get_mock_frustum(cls, device="cpu") -> "Frustums": class RaySamples (line 119) | class RaySamples(TensorDataclass): method __post_init__ (line 140) | def __post_init__(self) -> None: method get_weights (line 153) | def get_weights(self, densities: TensorType[..., "num_samples", 1]) ->... method get_weights_and_transmittance (line 179) | def get_weights_and_transmittance( method get_weights_from_alphas (line 205) | def get_weights_from_alphas(self, alphas: TensorType[..., "num_samples... method get_weights_and_transmittance_from_alphas (line 224) | def get_weights_and_transmittance_from_alphas( class RayBundle (line 250) | class RayBundle(TensorDataclass): method set_camera_indices (line 274) | def set_camera_indices(self, camera_index: int) -> None: method __len__ (line 282) | def __len__(self): method sample (line 286) | def sample(self, num_rays: int) -> "RayBundle": method get_row_major_sliced_ray_bundle (line 299) | def get_row_major_sliced_ray_bundle(self, start_idx: int, end_idx: int... method get_ray_samples (line 312) | def get_ray_samples( FILE: AutoReconForDens3R/nerfstudio/configs/base_config.py class PrintableConfig (line 44) | class PrintableConfig: # pylint: disable=too-few-public-methods method __str__ (line 47) | def __str__(self): class InstantiateConfig (line 62) | class InstantiateConfig(PrintableConfig): # pylint: disable=too-few-pub... method setup (line 67) | def setup(self, **kwargs) -> Any: class MachineConfig (line 74) | class MachineConfig(PrintableConfig): class LocalWriterConfig (line 90) | class LocalWriterConfig(InstantiateConfig): method setup (line 108) | def setup(self, banner_messages: Optional[List[str]] = None, **kwargs)... class LoggingConfig (line 118) | class LoggingConfig(PrintableConfig): class TrainerConfig (line 137) | class TrainerConfig(PrintableConfig): class ViewerConfig (line 173) | class ViewerConfig(PrintableConfig): class Config (line 207) | class Config(PrintableConfig): method set_global (line 242) | def set_global(self) -> None: method is_viewer_enabled (line 253) | def is_viewer_enabled(self) -> bool: method is_wandb_enabled (line 257) | def is_wandb_enabled(self) -> bool: method is_tensorboard_enabled (line 261) | def is_tensorboard_enabled(self) -> bool: method set_timestamp (line 265) | def set_timestamp(self) -> None: method set_experiment_name (line 270) | def set_experiment_name(self) -> None: method get_base_dir (line 275) | def get_base_dir(self) -> Path: method get_checkpoint_dir (line 282) | def get_checkpoint_dir(self) -> Path: method print_to_terminal (line 286) | def print_to_terminal(self) -> None: method save_config (line 292) | def save_config(self) -> None: FILE: AutoReconForDens3R/nerfstudio/configs/config_utils.py function to_immutable_dict (line 30) | def to_immutable_dict(d: Dict[str, Any]): function convert_markup_to_ansi (line 39) | def convert_markup_to_ansi(markup_string: str) -> str: FILE: AutoReconForDens3R/nerfstudio/engine/callbacks.py class TrainingCallbackAttributes (line 28) | class TrainingCallbackAttributes: class TrainingCallbackLocation (line 45) | class TrainingCallbackLocation(Enum): class TrainingCallback (line 52) | class TrainingCallback: method __init__ (line 66) | def __init__( method run_callback (line 85) | def run_callback(self, step: int): method run_callback_at_location (line 98) | def run_callback_at_location(self, step: int, location: TrainingCallba... FILE: AutoReconForDens3R/nerfstudio/engine/optimizers.py class OptimizerConfig (line 36) | class OptimizerConfig(base_config.PrintableConfig): method setup (line 50) | def setup(self, params) -> Any: class AdamOptimizerConfig (line 59) | class AdamOptimizerConfig(OptimizerConfig): class RAdamOptimizerConfig (line 67) | class RAdamOptimizerConfig(OptimizerConfig): class AdamWOptimizerConfig (line 74) | class AdamWOptimizerConfig(OptimizerConfig): function setup_optimizers (line 81) | def setup_optimizers(config: base_config.Config, param_groups: Dict[str,... class Optimizers (line 104) | class Optimizers: method __init__ (line 112) | def __init__(self, config: Dict[str, Any], param_groups: Dict[str, Lis... method optimizer_step (line 129) | def optimizer_step(self, param_group_name: str) -> None: method scheduler_step (line 137) | def scheduler_step(self, param_group_name: str) -> None: method zero_grad_all (line 146) | def zero_grad_all(self) -> None: method optimizer_scaler_step_all (line 151) | def optimizer_scaler_step_all(self, grad_scaler: GradScaler) -> None: method optimizer_step_all (line 171) | def optimizer_step_all(self): method scheduler_step_all (line 180) | def scheduler_step_all(self, step: int) -> None: method load_optimizers (line 192) | def load_optimizers(self, loaded_state: Dict[str, Any]) -> None: method load_schedulers (line 201) | def load_schedulers(self, loaded_state: Dict[str, Any]) -> None: FILE: AutoReconForDens3R/nerfstudio/engine/schedulers.py class SchedulerConfig (line 27) | class SchedulerConfig(InstantiateConfig): method setup (line 36) | def setup(self, optimizer=None, lr_init=None, **kwargs) -> Any: class ExponentialDecaySchedule (line 41) | class ExponentialDecaySchedule(lr_scheduler.LambdaLR): method __init__ (line 58) | def __init__(self, optimizer, lr_init, lr_final, max_steps, lr_delay_s... class DelayerScheduler (line 76) | class DelayerScheduler(lr_scheduler.LambdaLR): method __init__ (line 79) | def __init__( class DelayedExponentialScheduler (line 99) | class DelayedExponentialScheduler(DelayerScheduler): method __init__ (line 102) | def __init__( class MultiStepSchedulerConfig (line 120) | class MultiStepSchedulerConfig(InstantiateConfig): method setup (line 126) | def setup(self, optimizer=None, lr_init=None, **kwargs) -> Any: class ExponentialSchedulerConfig (line 136) | class ExponentialSchedulerConfig(InstantiateConfig): method setup (line 143) | def setup(self, optimizer=None, lr_init=None, **kwargs) -> Any: class NeuSSchedulerConfig (line 152) | class NeuSSchedulerConfig(InstantiateConfig): method setup (line 160) | def setup(self, optimizer=None, lr_init=None, **kwargs) -> Any: class NeuSScheduler (line 170) | class NeuSScheduler(lr_scheduler.LambdaLR): method __init__ (line 173) | def __init__(self, optimizer, warm_up_end, learning_rate_alpha, max_st... class AutoReconSchedulerConfig (line 187) | class AutoReconSchedulerConfig(InstantiateConfig): method setup (line 195) | def setup(self, optimizer=None, lr_init=None, **kwargs) -> Any: class AutoReconScheduler (line 203) | class AutoReconScheduler(lr_scheduler.LambdaLR): method __init__ (line 206) | def __init__(self, optimizer, warm_up_end) -> None: FILE: AutoReconForDens3R/nerfstudio/engine/trainer.py class Trainer (line 52) | class Trainer: method __init__ (line 74) | def __init__(self, config: cfg.Config, local_rank: int = 0, world_size... method setup (line 106) | def setup(self, test_mode: Literal["test", "val", "inference"] = "val"): method train (line 131) | def train(self) -> None: method _always_render (line 189) | def _always_render(self, step): method _check_viewer_warnings (line 196) | def _check_viewer_warnings(self) -> None: method _init_viewer_state (line 206) | def _init_viewer_state(self) -> None: method _update_viewer_state (line 217) | def _update_viewer_state(self, step: int): method _update_viewer_rays_per_sec (line 237) | def _update_viewer_rays_per_sec(self, train_t: TimeWriter, vis_t: Time... method _load_checkpoint (line 253) | def _load_checkpoint(self) -> None: method save_checkpoint (line 286) | def save_checkpoint(self, step: int) -> None: method train_iteration (line 317) | def train_iteration(self, step: int) -> Tuple[torch.Tensor, Dict[str, ... method eval_iteration (line 338) | def eval_iteration(self, step): FILE: AutoReconForDens3R/nerfstudio/exporter/exporter_utils.py class Mesh (line 50) | class Mesh: function get_mesh_from_pymeshlab_mesh (line 65) | def get_mesh_from_pymeshlab_mesh(mesh: pymeshlab.Mesh) -> Mesh: function get_mesh_from_filename (line 78) | def get_mesh_from_filename(filename: str, target_num_faces: Optional[int... function generate_point_cloud (line 89) | def generate_point_cloud( function render_trajectory (line 211) | def render_trajectory( FILE: AutoReconForDens3R/nerfstudio/exporter/mesh_culling_utils.py class MeshCullerConfig (line 28) | class MeshCullerConfig(InstantiateConfig): class MeshCuller (line 50) | class MeshCuller: method __init__ (line 53) | def __init__(self, config: MeshCullerConfig): method main (line 58) | def main(self, mesh: Mesh, pipeline: Pipeline, output_path: Path): method _get_ray_sampler (line 108) | def _get_ray_sampler(self): method _build_ray_bundle (line 125) | def _build_ray_bundle(self, vertices, faces, face_normals) -> RayBundle: method _get_ray_length (line 142) | def _get_ray_length(self, vertices, faces): FILE: AutoReconForDens3R/nerfstudio/exporter/texture_utils.py function get_parallelogram_area (line 43) | def get_parallelogram_area( function get_texture_image (line 59) | def get_texture_image(num_pixels_w, num_pixels_h, device): function unwrap_mesh_per_uv_triangle (line 78) | def unwrap_mesh_per_uv_triangle( function unwrap_mesh_with_xatlas (line 210) | def unwrap_mesh_with_xatlas( function export_textured_mesh (line 326) | def export_textured_mesh( FILE: AutoReconForDens3R/nerfstudio/exporter/tsdf_utils.py class TSDF (line 42) | class TSDF: method to (line 62) | def to(self, device: str): method device (line 77) | def device(self): method truncation (line 82) | def truncation(self): method from_aabb (line 89) | def from_aabb(aabb: TensorType[2, 3], volume_dims: TensorType[3]): method get_mesh (line 116) | def get_mesh(self) -> Mesh: method export_mesh (line 139) | def export_mesh(cls, mesh: Mesh, filename: str): method integrate_tsdf (line 168) | def integrate_tsdf( function export_tsdf_mesh (line 273) | def export_tsdf_mesh( FILE: AutoReconForDens3R/nerfstudio/field_components/activations.py class _TruncExp (line 24) | class _TruncExp(Function): # pylint: disable=abstract-method method forward (line 29) | def forward(ctx, x): # pylint: disable=arguments-differ method backward (line 35) | def backward(ctx, g): # pylint: disable=arguments-differ FILE: AutoReconForDens3R/nerfstudio/field_components/base_field_component.py class FieldComponent (line 25) | class FieldComponent(nn.Module): method __init__ (line 33) | def __init__(self, in_dim: Optional[int] = None, out_dim: Optional[int... method build_nn_modules (line 38) | def build_nn_modules(self) -> None: method set_in_dim (line 42) | def set_in_dim(self, in_dim: int) -> None: method get_out_dim (line 52) | def get_out_dim(self) -> int: method forward (line 59) | def forward(self, in_tensor: TensorType["bs":..., "input_dim"]) -> Ten... FILE: AutoReconForDens3R/nerfstudio/field_components/embedding.py class Embedding (line 26) | class Embedding(FieldComponent): method __init__ (line 35) | def __init__(self, in_dim: int, out_dim: int) -> None: method build_nn_modules (line 41) | def build_nn_modules(self) -> None: method mean (line 44) | def mean(self, dim=0): method forward (line 48) | def forward(self, in_tensor: TensorType[..., "input_dim"]) -> TensorTy... FILE: AutoReconForDens3R/nerfstudio/field_components/encodings.py class Encoding (line 41) | class Encoding(FieldComponent): method __init__ (line 48) | def __init__(self, in_dim: int) -> None: method forward (line 54) | def forward(self, in_tensor: TensorType["bs":..., "input_dim"]) -> Ten... class Identity (line 63) | class Identity(Encoding): method get_out_dim (line 66) | def get_out_dim(self) -> int: method forward (line 71) | def forward(self, in_tensor: TensorType["bs":..., "input_dim"]) -> Ten... class ScalingAndOffset (line 75) | class ScalingAndOffset(Encoding): method __init__ (line 84) | def __init__(self, in_dim: int, scaling: float = 1.0, offset: float = ... method get_out_dim (line 90) | def get_out_dim(self) -> int: method forward (line 95) | def forward(self, in_tensor: TensorType["bs":..., "input_dim"]) -> Ten... class NeRFEncoding (line 99) | class NeRFEncoding(Encoding): method __init__ (line 111) | def __init__( method get_out_dim (line 121) | def get_out_dim(self) -> int: method forward (line 129) | def forward( class RFFEncoding (line 162) | class RFFEncoding(Encoding): method __init__ (line 172) | def __init__(self, in_dim: int, num_frequencies: int, scale: float, in... method get_out_dim (line 185) | def get_out_dim(self) -> int: method forward (line 188) | def forward( class HashEncoding (line 220) | class HashEncoding(Encoding): method __init__ (line 234) | def __init__( method get_out_dim (line 286) | def get_out_dim(self) -> int: method hash_fn (line 289) | def hash_fn(self, in_tensor: TensorType["bs":..., "num_levels", 3]) ->... method pytorch_fwd (line 308) | def pytorch_fwd(self, in_tensor: TensorType["bs":..., "input_dim"]) ->... method forward (line 351) | def forward(self, in_tensor: TensorType["bs":..., "input_dim"]) -> Ten... class TensorCPEncoding (line 357) | class TensorCPEncoding(Encoding): method __init__ (line 366) | def __init__(self, resolution: int = 256, num_components: int = 24, in... method get_out_dim (line 375) | def get_out_dim(self) -> int: method forward (line 378) | def forward(self, in_tensor: TensorType["bs":..., "input_dim"]) -> Ten... method upsample_grid (line 393) | def upsample_grid(self, resolution: int) -> None: class TensorVMEncoding (line 407) | class TensorVMEncoding(Encoding): method __init__ (line 419) | def __init__( method get_out_dim (line 437) | def get_out_dim(self) -> int: method index_fn (line 440) | def index_fn(self, x: torch.Tensor, y: torch.Tensor, width: int, heigh... method grid_sample_2d (line 453) | def grid_sample_2d(self, feature, coord, type="plane"): method forward (line 496) | def forward(self, in_tensor: TensorType["bs":..., "input_dim"]) -> Ten... method upsample_grid (line 529) | def upsample_grid(self, resolution: int) -> None: class SHEncoding (line 545) | class SHEncoding(Encoding): method __init__ (line 552) | def __init__(self, levels: int = 4) -> None: method get_out_dim (line 560) | def get_out_dim(self) -> int: method forward (line 564) | def forward(self, in_tensor: TensorType["bs":..., "input_dim"]) -> Ten... class PeriodicVolumeEncoding (line 568) | class PeriodicVolumeEncoding(Encoding): method __init__ (line 581) | def __init__( method get_out_dim (line 616) | def get_out_dim(self) -> int: method hash_fn (line 619) | def hash_fn(self, in_tensor: TensorType["bs":..., "num_levels", 3]) ->... method pytorch_fwd (line 640) | def pytorch_fwd(self, in_tensor: TensorType["bs":..., "input_dim"]) ->... method forward (line 686) | def forward(self, in_tensor: TensorType["bs":..., "input_dim"]) -> Ten... method get_total_variation_loss (line 689) | def get_total_variation_loss(self): FILE: AutoReconForDens3R/nerfstudio/field_components/field_heads.py class FieldHeadNames (line 28) | class FieldHeadNames(Enum): class FieldHead (line 49) | class FieldHead(FieldComponent): method __init__ (line 59) | def __init__( method set_in_dim (line 76) | def set_in_dim(self, in_dim: int) -> None: method _construct_net (line 81) | def _construct_net(self): method forward (line 84) | def forward(self, in_tensor: TensorType["bs":..., "in_dim"]) -> Tensor... class DensityFieldHead (line 101) | class DensityFieldHead(FieldHead): method __init__ (line 109) | def __init__(self, in_dim: Optional[int] = None, activation: Optional[... class RGBFieldHead (line 113) | class RGBFieldHead(FieldHead): method __init__ (line 121) | def __init__(self, in_dim: Optional[int] = None, activation: Optional[... class SHFieldHead (line 125) | class SHFieldHead(FieldHead): method __init__ (line 135) | def __init__( class UncertaintyFieldHead (line 143) | class UncertaintyFieldHead(FieldHead): method __init__ (line 151) | def __init__(self, in_dim: Optional[int] = None, activation: Optional[... class TransientRGBFieldHead (line 155) | class TransientRGBFieldHead(FieldHead): method __init__ (line 163) | def __init__(self, in_dim: Optional[int] = None, activation: Optional[... class TransientDensityFieldHead (line 167) | class TransientDensityFieldHead(FieldHead): method __init__ (line 175) | def __init__(self, in_dim: Optional[int] = None, activation: Optional[... class SemanticFieldHead (line 181) | class SemanticFieldHead(FieldHead): method __init__ (line 190) | def __init__(self, num_classes: int, in_dim: Optional[int] = None) -> ... class PredNormalsFieldHead (line 194) | class PredNormalsFieldHead(FieldHead): method __init__ (line 202) | def __init__(self, in_dim: Optional[int] = None, activation: Optional[... method forward (line 205) | def forward(self, in_tensor: TensorType["bs":..., "in_dim"]) -> Tensor... FILE: AutoReconForDens3R/nerfstudio/field_components/mlp.py class MLP (line 27) | class MLP(FieldComponent): method __init__ (line 39) | def __init__( method build_nn_modules (line 63) | def build_nn_modules(self) -> None: method forward (line 80) | def forward(self, in_tensor: TensorType["bs":..., "in_dim"]) -> Tensor... FILE: AutoReconForDens3R/nerfstudio/field_components/progressive_encoding.py class ProgressiveEncoding (line 11) | class ProgressiveEncoding(nn.Module): method __init__ (line 14) | def __init__( method _get_init_weights (line 41) | def _get_init_weights(self): method forward (line 46) | def forward(self, positions): FILE: AutoReconForDens3R/nerfstudio/field_components/spatial_distortions.py class SpatialDistortion (line 27) | class SpatialDistortion(nn.Module): method forward (line 30) | def forward( class SceneContraction (line 42) | class SceneContraction(SpatialDistortion): method __init__ (line 63) | def __init__( method forward (line 72) | def forward(self, positions): class ForegroundAwareSceneContraction (line 107) | class ForegroundAwareSceneContraction(SpatialDistortion): method __init__ (line 116) | def __init__( method forward (line 134) | def forward(self, positions): FILE: AutoReconForDens3R/nerfstudio/field_components/temporal_distortions.py class TemporalDistortion (line 28) | class TemporalDistortion(nn.Module): method forward (line 31) | def forward(self, positions: TensorType["bs":..., 3], times: Optional[... class TemporalDistortionKind (line 42) | class TemporalDistortionKind(Enum): method to_temporal_distortion (line 47) | def to_temporal_distortion(self, config: Dict[str, Any]) -> TemporalDi... class DNeRFDistortion (line 54) | class DNeRFDistortion(TemporalDistortion): method __init__ (line 64) | def __init__( method forward (line 87) | def forward(self, positions, times=None): FILE: AutoReconForDens3R/nerfstudio/fields/base_field.py class FieldConfig (line 33) | class FieldConfig(InstantiateConfig): class Field (line 40) | class Field(nn.Module): method __init__ (line 43) | def __init__(self) -> None: method density_fn (line 48) | def density_fn(self, positions: TensorType["bs":..., 3]) -> TensorType... method get_density (line 68) | def get_density(self, ray_samples: RaySamples) -> Tuple[TensorType[...... method get_normals (line 75) | def get_normals(self) -> TensorType[..., 3]: method get_outputs (line 94) | def get_outputs( method forward (line 104) | def forward(self, ray_samples: RaySamples, compute_normals: bool = Fal... FILE: AutoReconForDens3R/nerfstudio/fields/density_fields.py class HashMLPDensityField (line 41) | class HashMLPDensityField(Field): method __init__ (line 54) | def __init__( method _get_normalized_positions (line 104) | def _get_normalized_positions(self, positions): method get_density (line 112) | def get_density(self, ray_samples: RaySamples): method get_outputs (line 134) | def get_outputs(self, ray_samples: RaySamples, density_embedding: Opti... FILE: AutoReconForDens3R/nerfstudio/fields/feature_field.py class FeatureFieldConfig (line 33) | class FeatureFieldConfig(FieldConfig): class FeatureField (line 56) | class FeatureField(Field): method __init__ (line 61) | def __init__(self, config: FeatureFieldConfig, spatial_distortion: Opt... method build_ngp (line 81) | def build_ngp(self): method build (line 142) | def build(self): method query_features (line 145) | def query_features( method get_outputs (line 158) | def get_outputs(self, ray_samples: RaySamples) -> Dict[str, torch.Tens... method forward (line 173) | def forward(self, ray_samples: RaySamples): method _get_input_positions (line 178) | def _get_input_positions(self, positions: torch.Tensor): class FeatureSegFieldConfig (line 191) | class FeatureSegFieldConfig(FieldConfig): class FeatureSegField (line 208) | class FeatureSegField(Field): method __init__ (line 213) | def __init__(self, config: FeatureFieldConfig, ptcd_data: BasicPointCl... method compute_seg_prob (line 233) | def compute_seg_prob( method search_knn_fg (line 277) | def search_knn_fg(self, feats: TensorType["n", "c"]): method search_knn_bg (line 280) | def search_knn_bg(self, feats: TensorType["n", "c"]): method search_knn_plane (line 283) | def search_knn_plane(self, feats: TensorType["n", "c"]): method _search_knn (line 286) | def _search_knn( method compute_cos_sim (line 299) | def compute_cos_sim( method _register_all_tensors_as_buffers (line 317) | def _register_all_tensors_as_buffers(self): method forward (line 329) | def forward( FILE: AutoReconForDens3R/nerfstudio/fields/instant_ngp_field.py function get_normalized_directions (line 41) | def get_normalized_directions(directions: TensorType["bs":..., 3]): class TCNNInstantNGPField (line 50) | class TCNNInstantNGPField(Field): method __init__ (line 66) | def __init__( method get_density (line 137) | def get_density(self, ray_samples: RaySamples): method get_outputs (line 151) | def get_outputs(self, ray_samples: RaySamples, density_embedding: Opti... method get_opacity (line 177) | def get_opacity(self, positions: TensorType["bs":..., 3], step_size) -... FILE: AutoReconForDens3R/nerfstudio/fields/nerfacto_field.py function get_normalized_directions (line 59) | def get_normalized_directions(directions: TensorType["bs":..., 3]): class TCNNNerfactoField (line 68) | class TCNNNerfactoField(Field): method __init__ (line 88) | def __init__( method get_density (line 229) | def get_density(self, ray_samples: RaySamples): method get_outputs (line 251) | def get_outputs(self, ray_samples: RaySamples, density_embedding: Opti... method _get_appeanreance_embedding (line 313) | def _get_appeanreance_embedding(self, camera_indices: TensorType, dire... method _get_normalized_positions (line 335) | def _get_normalized_positions(self, positions): class TorchNerfactoField (line 346) | class TorchNerfactoField(Field): method __init__ (line 351) | def __init__( method get_density (line 397) | def get_density(self, ray_samples: RaySamples): method get_outputs (line 408) | def get_outputs( FILE: AutoReconForDens3R/nerfstudio/fields/nerfw_field.py class VanillaNerfWField (line 38) | class VanillaNerfWField(Field): method __init__ (line 54) | def __init__( method get_density (line 112) | def get_density(self, ray_samples: RaySamples): method get_outputs (line 120) | def get_outputs( FILE: AutoReconForDens3R/nerfstudio/fields/sdf_field.py class LaplaceDensity (line 54) | class LaplaceDensity(nn.Module): # alpha * Laplace(loc=0, scale=beta).c... method __init__ (line 57) | def __init__(self, init_val, beta_min=0.0001): method forward (line 62) | def forward( method get_beta (line 73) | def get_beta(self): class SigmoidDensity (line 79) | class SigmoidDensity(nn.Module): # alpha * Laplace(loc=0, scale=beta).c... method __init__ (line 82) | def __init__(self, init_val, beta_min=0.0001): method forward (line 87) | def forward( method get_beta (line 100) | def get_beta(self): class SingleVarianceNetwork (line 106) | class SingleVarianceNetwork(nn.Module): method __init__ (line 116) | def __init__(self, init_val): method forward (line 120) | def forward(self, x): method get_variance (line 124) | def get_variance(self): class SDFFieldConfig (line 130) | class SDFFieldConfig(FieldConfig): class SDFField (line 202) | class SDFField(Field): method __init__ (line 211) | def __init__( method _build_geo_encodings (line 249) | def _build_geo_encodings(self) -> None: method _build_geo_networks (line 307) | def _build_geo_networks(self): method _build_color_encodings (line 348) | def _build_color_encodings(self, use_average_appearance_embedding): method _build_color_networks (line 358) | def _build_color_networks(self): method set_cos_anneal_ratio (line 383) | def set_cos_anneal_ratio(self, anneal: float) -> None: method forward_geonetwork (line 387) | def forward_geonetwork(self, inputs): method get_sdf (line 414) | def get_sdf(self, ray_samples: RaySamples): method get_gradient (line 422) | def get_gradient(self, ray_samples: RaySamples): method gradient (line 427) | def gradient(self, x): method get_density (line 437) | def get_density(self, ray_samples: RaySamples): method get_alpha (line 446) | def get_alpha(self, ray_samples: RaySamples, sdf=None, gradients=None): method get_occupancy (line 498) | def get_occupancy(self, sdf): method get_colors (line 503) | def get_colors(self, points, directions, normals, geo_features, camera... method get_outputs (line 531) | def get_outputs(self, ray_samples: RaySamples, return_alphas=False, re... method forward (line 594) | def forward(self, ray_samples: RaySamples, return_alphas=False, return... method _get_appeanreance_embedding (line 605) | def _get_appeanreance_embedding(self, camera_indices: TensorType, dire... method _get_direction_encoding (line 628) | def _get_direction_encoding(self): method _get_input_dims (line 639) | def _get_input_dims(self): method _get_geometric_init_radius (line 648) | def _get_geometric_init_radius(self): method _get_input_positions (line 661) | def _get_input_positions(self, inputs: torch.Tensor): method _get_normalized_positions (line 671) | def _get_normalized_positions(self, positions): method _compute_curvature_loss (line 681) | def _compute_curvature_loss(self, inputs: TensorType["bs", 3], normals... FILE: AutoReconForDens3R/nerfstudio/fields/semantic_nerf_field.py class SemanticNerfField (line 36) | class SemanticNerfField(Field): method __init__ (line 50) | def __init__( method get_density (line 91) | def get_density(self, ray_samples: RaySamples): method get_outputs (line 97) | def get_outputs( FILE: AutoReconForDens3R/nerfstudio/fields/tensorf_field.py class TensoRFField (line 33) | class TensoRFField(Field): method __init__ (line 36) | def __init__( method get_density (line 86) | def get_density(self, ray_samples: RaySamples): method get_outputs (line 95) | def get_outputs(self, ray_samples: RaySamples, density_embedding: Opti... method forward (line 115) | def forward( FILE: AutoReconForDens3R/nerfstudio/fields/vanilla_nerf_field.py class NeRFField (line 38) | class NeRFField(Field): method __init__ (line 54) | def __init__( method get_density (line 97) | def get_density(self, ray_samples: RaySamples): method get_outputs (line 113) | def get_outputs( method _get_normalized_positions (line 123) | def _get_normalized_positions(self, positions): FILE: AutoReconForDens3R/nerfstudio/model_components/losses.py function outer (line 35) | def outer( function lossfun_outer (line 67) | def lossfun_outer( function ray_samples_to_sdist (line 87) | def ray_samples_to_sdist(ray_samples): function interlevel_loss (line 95) | def interlevel_loss(weights_list, ray_samples_list): function lossfun_distortion (line 113) | def lossfun_distortion(t, w): function distortion_loss (line 127) | def distortion_loss(weights_list, ray_samples_list): function nerfstudio_distortion_loss (line 135) | def nerfstudio_distortion_loss( function orientation_loss (line 177) | def orientation_loss( function pred_normal_loss (line 192) | def pred_normal_loss( function monosdf_normal_loss (line 201) | def monosdf_normal_loss(normal_pred: torch.Tensor, normal_gt: torch.Tens... function compute_scale_and_shift (line 216) | def compute_scale_and_shift(prediction, target, mask): function reduction_batch_based (line 239) | def reduction_batch_based(image_loss, M): function reduction_image_based (line 251) | def reduction_image_based(image_loss, M): function mse_loss (line 262) | def mse_loss(prediction, target, mask, reduction=reduction_batch_based): function gradient_loss (line 271) | def gradient_loss(prediction, target, mask, reduction=reduction_batch_ba... class MiDaSMSELoss (line 291) | class MiDaSMSELoss(nn.Module): method __init__ (line 292) | def __init__(self, reduction="batch-based"): method forward (line 300) | def forward(self, prediction, target, mask): class GradientLoss (line 304) | class GradientLoss(nn.Module): method __init__ (line 305) | def __init__(self, scales=4, reduction="batch-based"): method forward (line 315) | def forward(self, prediction, target, mask): class ScaleAndShiftInvariantLoss (line 331) | class ScaleAndShiftInvariantLoss(nn.Module): method __init__ (line 332) | def __init__(self, alpha=0.5, scales=4, reduction="batch-based"): method forward (line 341) | def forward(self, prediction, target, mask): method __get_prediction_ssi (line 352) | def __get_prediction_ssi(self): class SSIM (line 362) | class SSIM(nn.Module): method __init__ (line 365) | def __init__(self, patch_size): method forward (line 378) | def forward(self, x, y): class NCC (line 396) | class NCC(nn.Module): method __init__ (line 399) | def __init__(self, patch_size: int = 11, min_patch_variance: float = 0... method forward (line 404) | def forward(self, x, y): class MultiViewLoss (line 432) | class MultiViewLoss(nn.Module): method __init__ (line 435) | def __init__(self, patch_size: int = 11, topk: int = 4, min_patch_vari... method forward (line 447) | def forward(self, patches: torch.Tensor, valid: torch.Tensor): class SensorDepthLoss (line 569) | class SensorDepthLoss(nn.Module): method __init__ (line 572) | def __init__(self, truncation: float): method forward (line 576) | def forward(self, batch, outputs): FILE: AutoReconForDens3R/nerfstudio/model_components/patch_warping.py function get_intersection_points (line 27) | def get_intersection_points( function get_homography (line 93) | def get_homography( class PatchWarping (line 140) | class PatchWarping(nn.Module): method __init__ (line 143) | def __init__(self, patch_size: int = 31, pixel_offset: float = 0.5, va... method forward (line 158) | def forward( FILE: AutoReconForDens3R/nerfstudio/model_components/ray_generators.py class RayGenerator (line 26) | class RayGenerator(nn.Module): method __init__ (line 35) | def __init__(self, cameras: Cameras, pose_optimizer: CameraOptimizer) ... method forward (line 41) | def forward(self, ray_indices: TensorType["num_rays", 3]) -> RayBundle: FILE: AutoReconForDens3R/nerfstudio/model_components/ray_samplers.py class Sampler (line 41) | class Sampler(nn.Module): method __init__ (line 48) | def __init__( method generate_ray_samples (line 56) | def generate_ray_samples(self) -> RaySamples: method forward (line 59) | def forward(self, *args, **kwargs) -> RaySamples: class SpacedSampler (line 64) | class SpacedSampler(Sampler): method __init__ (line 75) | def __init__( method generate_ray_samples (line 89) | def generate_ray_samples( class UniformSampler (line 145) | class UniformSampler(SpacedSampler): method __init__ (line 154) | def __init__( class LinearDisparitySampler (line 169) | class LinearDisparitySampler(SpacedSampler): method __init__ (line 178) | def __init__( class SqrtSampler (line 193) | class SqrtSampler(SpacedSampler): method __init__ (line 201) | def __init__( class LogSampler (line 216) | class LogSampler(SpacedSampler): method __init__ (line 224) | def __init__( class UniformLinDispPiecewiseSampler (line 239) | class UniformLinDispPiecewiseSampler(SpacedSampler): method __init__ (line 250) | def __init__( class PDFSampler (line 265) | class PDFSampler(Sampler): method __init__ (line 276) | def __init__( method generate_ray_samples (line 290) | def generate_ray_samples( class VolumetricSampler (line 397) | class VolumetricSampler(Sampler): method __init__ (line 408) | def __init__( method get_sigma_fn (line 423) | def get_sigma_fn(self, origins, directions) -> Optional[Callable]: method generate_ray_samples (line 446) | def generate_ray_samples(self) -> RaySamples: method forward (line 452) | def forward( class ProposalNetworkSampler (line 525) | class ProposalNetworkSampler(Sampler): method __init__ (line 528) | def __init__( method set_anneal (line 556) | def set_anneal(self, anneal: float) -> None: method step_cb (line 560) | def step_cb(self, step): method generate_ray_samples (line 565) | def generate_ray_samples( class ErrorBoundedSampler (line 609) | class ErrorBoundedSampler(Sampler): method __init__ (line 612) | def __init__( method generate_ray_samples (line 641) | def generate_ray_samples( method get_dstar (line 732) | def get_dstar(self, sdf, ray_samples: RaySamples): method get_updated_beta (line 756) | def get_updated_beta(self, beta0, beta, density_fn, sdf, d_star, ray_s... method get_error_bound (line 768) | def get_error_bound(self, beta, density_fn, sdf, d_star, ray_samples): method merge_ray_samples (line 786) | def merge_ray_samples(self, ray_bundle: RayBundle, ray_samples_1: RayS... function save_points (line 825) | def save_points(path_save, pts, colors=None, normals=None, BRG2RGB=False): class NeuSSampler (line 849) | class NeuSSampler(Sampler): method __init__ (line 852) | def __init__( method generate_ray_samples (line 881) | def generate_ray_samples( method rendering_sdf_with_fixed_inv_s (line 943) | def rendering_sdf_with_fixed_inv_s(self, ray_samples: RaySamples, sdf:... class UniSurfSampler (line 981) | class UniSurfSampler(Sampler): method __init__ (line 984) | def __init__( method step_cb (line 1022) | def step_cb(self, step): method generate_ray_samples (line 1027) | def generate_ray_samples( method merge_ray_samples_in_eculidean (line 1129) | def merge_ray_samples_in_eculidean( method secant_method (line 1166) | def secant_method( class NeuralReconWSampler (line 1175) | class NeuralReconWSampler(Sampler): method __init__ (line 1178) | def __init__( method init_grid_coordinate (line 1230) | def init_grid_coordinate(self): method update_binary_grid (line 1245) | def update_binary_grid(self, step, sdf_fn=None): method generate_ray_samples (line 1283) | def generate_ray_samples( class NeuSAccSampler (line 1349) | class NeuSAccSampler(Sampler): method __init__ (line 1352) | def __init__( method init_grid_coordinate (line 1395) | def init_grid_coordinate(self): method update_step_size (line 1412) | def update_step_size(self, step, inv_s=None): method update_binary_grid (line 1418) | def update_binary_grid(self, step, sdf_fn=None, inv_s=None): method create_ray_samples_from_ray_indices (line 1469) | def create_ray_samples_from_ray_indices(self, ray_bundle: RayBundle, r... method generate_ray_samples (line 1491) | def generate_ray_samples( class RayTracingSampler (line 1540) | class RayTracingSampler(Sampler): method __init__ (line 1555) | def __init__( method generate_ray_samples (line 1591) | def generate_ray_samples( method adjust_aabb_intersection (line 1619) | def adjust_aabb_intersection(self, ray_bundle: RayBundle) -> RayBundle: method sphere_tracing (line 1653) | def sphere_tracing(self, ray_bundle: RayBundle, sdf_fn: Callable): method _sphere_tracing_idr (line 1715) | def _sphere_tracing_idr(self, ray_bundle: RayBundle, sdf_fn: Callable): class DummySampler (line 1797) | class DummySampler(Sampler): method __init__ (line 1799) | def __init__(self): method generate_ray_samples (line 1802) | def generate_ray_samples(self, ray_bundle: RayBundle) -> RaySamples: FILE: AutoReconForDens3R/nerfstudio/model_components/renderers.py class RGBRenderer (line 46) | class RGBRenderer(nn.Module): method __init__ (line 52) | def __init__( method combine_rgb (line 60) | def combine_rgb( method forward (line 102) | def forward( class SHRenderer (line 133) | class SHRenderer(nn.Module): method __init__ (line 140) | def __init__( method forward (line 149) | def forward( class AccumulationRenderer (line 182) | class AccumulationRenderer(nn.Module): method forward (line 185) | def forward( class DepthRenderer (line 210) | class DepthRenderer(nn.Module): method __init__ (line 220) | def __init__(self, method: Literal["median", "expected"] = "median") -... method forward (line 224) | def forward( class UncertaintyRenderer (line 275) | class UncertaintyRenderer(nn.Module): method forward (line 278) | def forward( class SemanticRenderer (line 295) | class SemanticRenderer(nn.Module): method forward (line 298) | def forward( class NormalsRenderer (line 308) | class NormalsRenderer(nn.Module): method forward (line 311) | def forward( class FeatureRenderer (line 321) | class FeatureRenderer(nn.Module): method forward (line 324) | def forward( FILE: AutoReconForDens3R/nerfstudio/model_components/scene_colliders.py class SceneCollider (line 29) | class SceneCollider(nn.Module): method __init__ (line 32) | def __init__(self, **kwargs) -> None: method set_nears_and_fars (line 36) | def set_nears_and_fars(self, ray_bundle) -> RayBundle: method forward (line 40) | def forward(self, ray_bundle: RayBundle) -> RayBundle: class AABBBoxCollider (line 47) | class AABBBoxCollider(SceneCollider): method __init__ (line 54) | def __init__(self, scene_box: SceneBox, near_plane: float = 0.0, **kwa... method _intersect_with_aabb (line 59) | def _intersect_with_aabb( method set_nears_and_fars (line 97) | def set_nears_and_fars(self, ray_bundle: RayBundle) -> RayBundle: class NearFarCollider (line 111) | class NearFarCollider(SceneCollider): method __init__ (line 119) | def __init__(self, near_plane: float, far_plane: float, **kwargs) -> N... method set_nears_and_fars (line 124) | def set_nears_and_fars(self, ray_bundle: RayBundle) -> RayBundle: class SphereCollider (line 132) | class SphereCollider(SceneCollider): method __init__ (line 136) | def __init__(self, radius: float = 1., near_plane: float = 0.1, far_pl... method forward (line 142) | def forward(self, ray_bundle: RayBundle) -> RayBundle: class AABBBoxNearFarCollider (line 165) | class AABBBoxNearFarCollider(SceneCollider): method __init__ (line 171) | def __init__(self, scene_box: SceneBox, near_plane: float = 0.1, far_p... method _intersect_with_aabb (line 177) | def _intersect_with_aabb( method set_nears_and_fars (line 211) | def set_nears_and_fars(self, ray_bundle: RayBundle) -> RayBundle: FILE: AutoReconForDens3R/nerfstudio/models/base_model.py class ModelConfig (line 43) | class ModelConfig(InstantiateConfig): class Model (line 58) | class Model(nn.Module): method __init__ (line 70) | def __init__( method device (line 94) | def device(self): method get_training_callbacks (line 98) | def get_training_callbacks( # pylint:disable=no-self-use method populate_modules (line 104) | def populate_modules(self): method get_param_groups (line 115) | def get_param_groups(self) -> Dict[str, List[Parameter]]: method get_outputs (line 123) | def get_outputs(self, ray_bundle: RayBundle) -> Dict[str, torch.Tensor]: method forward (line 134) | def forward(self, ray_bundle: RayBundle) -> Dict[str, torch.Tensor]: method get_metrics_dict (line 147) | def get_metrics_dict(self, outputs, batch) -> Dict[str, torch.Tensor]: method get_loss_dict (line 159) | def get_loss_dict(self, outputs, batch, metrics_dict=None, **kwargs) -... method get_outputs_for_camera_ray_bundle (line 169) | def get_outputs_for_camera_ray_bundle(self, camera_ray_bundle: RayBund... method get_image_metrics_and_images (line 200) | def get_image_metrics_and_images( method load_model (line 216) | def load_model(self, loaded_state: Dict[str, Any]) -> None: FILE: AutoReconForDens3R/nerfstudio/models/base_surface_model.py function map_range_val (line 90) | def map_range_val(input_val, input_start, input_end, output_start, outpu... class SurfaceModelConfig (line 104) | class SurfaceModelConfig(ModelConfig): class SurfaceModel (line 211) | class SurfaceModel(Model): method __init__ (line 220) | def __init__(self, *args, **kwargs): method populate_modules (line 238) | def populate_modules(self): method step_cb (line 279) | def step_cb(self, step): method _build_scene_contraction (line 282) | def _build_scene_contraction(self): method get_param_groups (line 298) | def get_param_groups(self) -> Dict[str, List[Parameter]]: method get_training_callbacks (line 309) | def get_training_callbacks( method _build_collider (line 351) | def _build_collider(self) -> None: method _build_bg_model_and_sampler (line 370) | def _build_bg_model_and_sampler(self) -> None: method _build_renderers (line 445) | def _build_renderers(self) -> None: method sample_and_forward_field (line 457) | def sample_and_forward_field(self, ray_bundle: RayBundle) -> Dict: method get_outputs (line 467) | def get_outputs(self, ray_bundle: RayBundle) -> Dict: method forward (line 484) | def forward(self, ray_bundle: RayBundle) -> Dict[str, torch.Tensor]: method fg_fields_query_and_render (line 503) | def fg_fields_query_and_render(self, ray_bundle: RayBundle) -> Dict[st... method bg_fields_query_and_render (line 542) | def bg_fields_query_and_render( method sample_and_forward_field_bg (line 580) | def sample_and_forward_field_bg( method _sample_and_forward_field_bg_proposal (line 593) | def _sample_and_forward_field_bg_proposal( method _update_outputs_for_loss (line 621) | def _update_outputs_for_loss(self, ray_bundle: RayBundle, outputs: Dic... method _update_outptus_for_vis (line 643) | def _update_outptus_for_vis( method _compute_prop_net_depths (line 653) | def _compute_prop_net_depths( method _handle_no_intersect_rays_fg (line 668) | def _handle_no_intersect_rays_fg(self, ray_bundle: RayBundle, samples_... method _handle_bottom_intersect_rays_bg (line 697) | def _handle_bottom_intersect_rays_bg(self, ray_bundle: RayBundle, samp... method _reset_near_far_for_background (line 706) | def _reset_near_far_for_background(self, ray_bundle: RayBundle) -> Ray... method get_outputs_flexible (line 721) | def get_outputs_flexible(self, ray_bundle: RayBundle, additional_input... method get_loss_dict (line 754) | def get_loss_dict(self, outputs, batch, metrics_dict=None, step=None) ... method get_metrics_dict (line 858) | def get_metrics_dict(self, outputs, batch) -> Dict: method get_image_metrics_and_images (line 864) | def get_image_metrics_and_images( method _get_no_intersection_ray_mask (line 939) | def _get_no_intersection_ray_mask(self, outputs: Dict[str, torch.Tenso... method _compute_curvature_loss (line 944) | def _compute_curvature_loss(self, outputs: Dict[str, torch.Tensor], lo... method _compute_interlevel_loss_bg (line 965) | def _compute_interlevel_loss_bg(self, outputs: Dict[str, torch.Tensor]... method _get_interlevel_loss_inputs_bg (line 977) | def _get_interlevel_loss_inputs_bg(self, outputs: Dict[str, torch.Tens... method _build_anneal_fn (line 992) | def _build_anneal_fn(self, anneal_fn_name, max_step): method _compute_ptcd_reg_fg (line 1008) | def _compute_ptcd_reg_fg(self, outputs: Dict[str, torch.Tensor], loss_... method _compute_ptcd_reg_plane (line 1023) | def _compute_ptcd_reg_plane(self, outputs: Dict[str, torch.Tensor], lo... method _compute_mask_beta_prior (line 1042) | def _compute_mask_beta_prior(self, outputs: Dict[str, torch.Tensor], l... FILE: AutoReconForDens3R/nerfstudio/models/distilled_neus_facto.py class DistilledNeuSFactoModelConfig (line 32) | class DistilledNeuSFactoModelConfig(NeuSFactoModelConfig): class DistilledNeuSFactoModel (line 49) | class DistilledNeuSFactoModel(NeuSFactoModel): method populate_modules (line 53) | def populate_modules(self): method sample_and_forward_field (line 77) | def sample_and_forward_field(self, ray_bundle: RayBundle): method sample_and_forward_field_bg (line 100) | def sample_and_forward_field_bg( method forward_field (line 127) | def forward_field(self, ray_samples): method _forward_field_sdf (line 135) | def _forward_field_sdf(self, ray_samples): method _forward_field_nerfacto (line 144) | def _forward_field_nerfacto(self, ray_samples): method _mask_ray_samples_with_aabb (line 153) | def _mask_ray_samples_with_aabb(self, ray_bundle: RayBundle, ray_sampl... method _compute_aabb_intersections (line 184) | def _compute_aabb_intersections(self, ray_bundle: RayBundle): FILE: AutoReconForDens3R/nerfstudio/models/dto.py class DtoOModelConfig (line 58) | class DtoOModelConfig(NerfactoModelConfig): class DtoOModel (line 68) | class DtoOModel(NerfactoModel): method populate_modules (line 77) | def populate_modules(self): method get_training_callbacks (line 141) | def get_training_callbacks( method get_param_groups (line 166) | def get_param_groups(self) -> Dict[str, List[Parameter]]: method get_outputs (line 171) | def get_outputs(self, ray_bundle: RayBundle): method get_metrics_dict (line 437) | def get_metrics_dict(self, outputs, batch): method get_loss_dict (line 457) | def get_loss_dict(self, outputs, batch, metrics_dict=None): method get_image_metrics_and_images (line 512) | def get_image_metrics_and_images( FILE: AutoReconForDens3R/nerfstudio/models/instant_ngp.py class InstantNGPModelConfig (line 52) | class InstantNGPModelConfig(ModelConfig): class NGPModel (line 85) | class NGPModel(Model): method __init__ (line 95) | def __init__(self, config: InstantNGPModelConfig, **kwargs) -> None: method populate_modules (line 98) | def populate_modules(self): method get_training_callbacks (line 140) | def get_training_callbacks( method get_param_groups (line 159) | def get_param_groups(self) -> Dict[str, List[Parameter]]: method get_outputs (line 166) | def get_outputs(self, ray_bundle: RayBundle): method get_metrics_dict (line 211) | def get_metrics_dict(self, outputs, batch): method get_loss_dict (line 218) | def get_loss_dict(self, outputs, batch, metrics_dict=None): method get_image_metrics_and_images (line 225) | def get_image_metrics_and_images( FILE: AutoReconForDens3R/nerfstudio/models/mipnerf.py class MipNerfModel (line 43) | class MipNerfModel(Model): method __init__ (line 50) | def __init__( method populate_modules (line 58) | def populate_modules(self): method get_param_groups (line 91) | def get_param_groups(self) -> Dict[str, List[Parameter]]: method get_outputs (line 98) | def get_outputs(self, ray_bundle: RayBundle): method get_loss_dict (line 139) | def get_loss_dict(self, outputs, batch, metrics_dict=None): method get_image_metrics_and_images (line 147) | def get_image_metrics_and_images( FILE: AutoReconForDens3R/nerfstudio/models/monosdf.py class MonoSDFModelConfig (line 28) | class MonoSDFModelConfig(VolSDFModelConfig): class MonoSDFModel (line 38) | class MonoSDFModel(VolSDFModel): FILE: AutoReconForDens3R/nerfstudio/models/nerfacto.py class NerfactoModelConfig (line 63) | class NerfactoModelConfig(ModelConfig): class NerfactoModel (line 122) | class NerfactoModel(Model): method populate_modules (line 131) | def populate_modules(self): method get_param_groups (line 207) | def get_param_groups(self) -> Dict[str, List[Parameter]]: method get_training_callbacks (line 213) | def get_training_callbacks( method get_outputs (line 244) | def get_outputs(self, ray_bundle: RayBundle): method get_metrics_dict (line 286) | def get_metrics_dict(self, outputs, batch): method get_loss_dict (line 294) | def get_loss_dict(self, outputs, batch, metrics_dict=None): method get_image_metrics_and_images (line 316) | def get_image_metrics_and_images( FILE: AutoReconForDens3R/nerfstudio/models/neuralreconW.py class NeuralReconWModelConfig (line 35) | class NeuralReconWModelConfig(NeuSModelConfig): class NeuralReconWModel (line 41) | class NeuralReconWModel(NeuSModel): method populate_modules (line 50) | def populate_modules(self): method get_training_callbacks (line 61) | def get_training_callbacks( FILE: AutoReconForDens3R/nerfstudio/models/neus.py class NeuSModelConfig (line 37) | class NeuSModelConfig(SurfaceModelConfig): class NeuSModel (line 55) | class NeuSModel(SurfaceModel): method populate_modules (line 64) | def populate_modules(self): method get_training_callbacks (line 78) | def get_training_callbacks( method sample_and_forward_field (line 100) | def sample_and_forward_field(self, ray_bundle: RayBundle) -> Dict: method get_metrics_dict (line 117) | def get_metrics_dict(self, outputs, batch) -> Dict: FILE: AutoReconForDens3R/nerfstudio/models/neus_acc.py class NeuSAccModelConfig (line 39) | class NeuSAccModelConfig(NeuSModelConfig): class NeuSAccModel (line 47) | class NeuSAccModel(NeuSModel): method populate_modules (line 56) | def populate_modules(self): method get_training_callbacks (line 63) | def get_training_callbacks( method get_outputs (line 91) | def get_outputs(self, ray_bundle: RayBundle): method get_metrics_dict (line 140) | def get_metrics_dict(self, outputs, batch): FILE: AutoReconForDens3R/nerfstudio/models/neus_facto.py class NeuSFactoModelConfig (line 45) | class NeuSFactoModelConfig(NeuSModelConfig): class NeuSFactoModel (line 93) | class NeuSFactoModel(NeuSModel): method populate_modules (line 102) | def populate_modules(self): method get_param_groups (line 155) | def get_param_groups(self) -> Dict[str, List[Parameter]]: method get_training_callbacks (line 160) | def get_training_callbacks( method sample_and_forward_field (line 193) | def sample_and_forward_field(self, ray_bundle: RayBundle): method get_loss_dict (line 217) | def get_loss_dict(self, outputs, batch, metrics_dict=None): method get_image_metrics_and_images (line 226) | def get_image_metrics_and_images( method _compute_interlevel_loss (line 241) | def _compute_interlevel_loss(self, outputs: Dict[str, torch.Tensor], l... method _get_interlevel_loss_inputs (line 246) | def _get_interlevel_loss_inputs(self, outputs): method _compute_proposal_eikonal_loss (line 260) | def _compute_proposal_eikonal_loss( FILE: AutoReconForDens3R/nerfstudio/models/neus_facto_dff.py class NeuSFactoDFFModelConfig (line 37) | class NeuSFactoDFFModelConfig(NeuSFactoModelConfig): class NeuSFactoDFFModel (line 74) | class NeuSFactoDFFModel(NeuSFactoModel): method populate_modules (line 83) | def populate_modules(self): method get_param_groups (line 106) | def get_param_groups(self) -> Dict[str, List[Parameter]]: method sample_and_forward_field (line 117) | def sample_and_forward_field(self, ray_bundle: RayBundle): method _sample_and_forward_field_ray_tracing (line 152) | def _sample_and_forward_field_ray_tracing( method get_outputs (line 206) | def get_outputs(self, ray_bundle: RayBundle) -> Dict: method get_loss_dict (line 246) | def get_loss_dict(self, outputs, batch, metrics_dict=None): method get_image_metrics_and_images (line 257) | def get_image_metrics_and_images( method query_seg_field (line 269) | def query_seg_field(self, inputs: TensorType["bs": ..., 3]) -> TensorT... method seg_aware_sdf (line 275) | def seg_aware_sdf(self, inputs: TensorType["bs": ..., 3]) -> TensorTyp... method _rectify_sdf (line 284) | def _rectify_sdf(self, sdf: TensorType["bs": ..., 1], seg: TensorType[... method _get_pca_feature_rendering (line 293) | def _get_pca_feature_rendering( method _get_fg_seg_rendering (line 321) | def _get_fg_seg_rendering( method _get_pseudo_gt_seg_mask (line 344) | def _get_pseudo_gt_seg_mask(self, method _get_ray_tracing_mask (line 364) | def _get_ray_tracing_mask( method _get_ray_tracing_depth (line 389) | def _get_ray_tracing_depth( method get_outputs_for_mesh_culling (line 405) | def get_outputs_for_mesh_culling( method render_fg_seg_for_mesh_culling (line 438) | def render_fg_seg_for_mesh_culling( FILE: AutoReconForDens3R/nerfstudio/models/neus_facto_reg.py class NeuSFactoRegModelConfig (line 29) | class NeuSFactoRegModelConfig(NeuSFactoModelConfig): class NeuSFactoRegModel (line 65) | class NeuSFactoRegModel(NeuSFactoModel): method populate_modules (line 70) | def populate_modules(self): method get_param_groups (line 117) | def get_param_groups(self) -> Dict[str, List[Parameter]]: method sample_and_forward_field (line 122) | def sample_and_forward_field(self, ray_bundle: RayBundle): method fg_fields_query_and_render (line 172) | def fg_fields_query_and_render(self, ray_bundle: RayBundle) -> Dict[st... method get_loss_dict (line 230) | def get_loss_dict(self, outputs, batch, metrics_dict=None, step=None): method get_image_metrics_and_images (line 247) | def get_image_metrics_and_images( method _compute_ptcd_reg_plane_field (line 264) | def _compute_ptcd_reg_plane_field(self, outputs: Dict[str, torch.Tenso... method get_metrics_dict (line 279) | def get_metrics_dict(self, outputs, batch) -> Dict: FILE: AutoReconForDens3R/nerfstudio/models/semantic_nerfw.py class SemanticNerfWModelConfig (line 58) | class SemanticNerfWModelConfig(NerfactoModelConfig): class SemanticNerfWModel (line 71) | class SemanticNerfWModel(Model): method __init__ (line 80) | def __init__(self, config: SemanticNerfWModelConfig, metadata: Dict, *... method populate_modules (line 85) | def populate_modules(self): method get_param_groups (line 148) | def get_param_groups(self) -> Dict[str, List[Parameter]]: method get_training_callbacks (line 154) | def get_training_callbacks( method get_outputs (line 178) | def get_outputs(self, ray_bundle: RayBundle): method get_metrics_dict (line 228) | def get_metrics_dict(self, outputs, batch): method get_loss_dict (line 236) | def get_loss_dict(self, outputs, batch, metrics_dict=None): method get_image_metrics_and_images (line 256) | def get_image_metrics_and_images( FILE: AutoReconForDens3R/nerfstudio/models/tensorf.py class TensoRFModelConfig (line 54) | class TensoRFModelConfig(ModelConfig): class TensoRFModel (line 77) | class TensoRFModel(Model): method __init__ (line 84) | def __init__( method get_training_callbacks (line 109) | def get_training_callbacks( method populate_modules (line 146) | def populate_modules(self): method get_param_groups (line 196) | def get_param_groups(self) -> Dict[str, List[Parameter]]: method get_outputs (line 210) | def get_outputs(self, ray_bundle: RayBundle): method get_loss_dict (line 242) | def get_loss_dict(self, outputs, batch, metrics_dict=None) -> Dict[str... method get_image_metrics_and_images (line 253) | def get_image_metrics_and_images( FILE: AutoReconForDens3R/nerfstudio/models/unisurf.py class UniSurfModelConfig (line 38) | class UniSurfModelConfig(SurfaceModelConfig): class UniSurfModel (line 56) | class UniSurfModel(SurfaceModel): method populate_modules (line 65) | def populate_modules(self): method get_training_callbacks (line 79) | def get_training_callbacks( method sample_and_forward_field (line 92) | def sample_and_forward_field(self, ray_bundle: RayBundle) -> Dict: method get_metrics_dict (line 111) | def get_metrics_dict(self, outputs, batch) -> Dict: method get_loss_dict (line 119) | def get_loss_dict(self, outputs, batch, metrics_dict=None) -> Dict: FILE: AutoReconForDens3R/nerfstudio/models/vanilla_nerf.py class VanillaModelConfig (line 48) | class VanillaModelConfig(ModelConfig): class NeRFModel (line 63) | class NeRFModel(Model): method __init__ (line 70) | def __init__( method populate_modules (line 84) | def populate_modules(self): method get_param_groups (line 128) | def get_param_groups(self) -> Dict[str, List[Parameter]]: method get_outputs (line 137) | def get_outputs(self, ray_bundle: RayBundle): method get_loss_dict (line 184) | def get_loss_dict(self, outputs, batch, metrics_dict=None) -> Dict[str... method get_image_metrics_and_images (line 196) | def get_image_metrics_and_images( FILE: AutoReconForDens3R/nerfstudio/models/volsdf.py class VolSDFModelConfig (line 31) | class VolSDFModelConfig(SurfaceModelConfig): class VolSDFModel (line 43) | class VolSDFModel(SurfaceModel): method populate_modules (line 52) | def populate_modules(self): method sample_and_forward_field (line 62) | def sample_and_forward_field(self, ray_bundle: RayBundle) -> Dict: method get_metrics_dict (line 79) | def get_metrics_dict(self, outputs, batch) -> Dict: FILE: AutoReconForDens3R/nerfstudio/pipelines/base_pipeline.py function module_wrapper (line 55) | def module_wrapper(ddp_or_model: Union[DDP, Model]) -> Model: class Pipeline (line 64) | class Pipeline(nn.Module): method model (line 103) | def model(self): method device (line 108) | def device(self): method get_train_loss_dict (line 113) | def get_train_loss_dict(self, step: int): method get_eval_loss_dict (line 132) | def get_eval_loss_dict(self, step: int): method get_eval_image_metrics_and_images (line 152) | def get_eval_image_metrics_and_images(self, step: int): method get_average_eval_image_metrics (line 162) | def get_average_eval_image_metrics(self, step: Optional[int] = None): method load_pipeline (line 165) | def load_pipeline(self, loaded_state: Dict[str, Any]) -> None: method get_training_callbacks (line 172) | def get_training_callbacks( method get_param_groups (line 177) | def get_param_groups(self) -> Dict[str, List[Parameter]]: class VanillaPipelineConfig (line 186) | class VanillaPipelineConfig(cfg.InstantiateConfig): class VanillaPipeline (line 197) | class VanillaPipeline(Pipeline): method __init__ (line 214) | def __init__( method device (line 250) | def device(self): method get_train_loss_dict (line 255) | def get_train_loss_dict(self, step: int): method forward (line 281) | def forward(self): method get_eval_loss_dict (line 289) | def get_eval_loss_dict(self, step: int): method get_eval_image_metrics_and_images (line 305) | def get_eval_image_metrics_and_images(self, step: int): method get_average_eval_image_metrics (line 324) | def get_average_eval_image_metrics(self, step: Optional[int] = None): method load_pipeline (line 371) | def load_pipeline(self, loaded_state: Dict[str, Any], strict=True) -> ... method get_training_callbacks (line 392) | def get_training_callbacks( method get_param_groups (line 401) | def get_param_groups(self) -> Dict[str, List[Parameter]]: class FlexibleInputPipelineConfig (line 414) | class FlexibleInputPipelineConfig(VanillaPipelineConfig): class FlexibleInputPipeline (line 425) | class FlexibleInputPipeline(VanillaPipeline): method get_train_loss_dict (line 427) | def get_train_loss_dict(self, step: int): FILE: AutoReconForDens3R/nerfstudio/pipelines/dynamic_batch.py class DynamicBatchPipelineConfig (line 30) | class DynamicBatchPipelineConfig(VanillaPipelineConfig): class DynamicBatchPipeline (line 40) | class DynamicBatchPipeline(VanillaPipeline): method __init__ (line 49) | def __init__( method _update_pixel_samplers (line 65) | def _update_pixel_samplers(self): method _update_dynamic_num_rays_per_batch (line 70) | def _update_dynamic_num_rays_per_batch(self, num_samples_per_batch: int): method get_train_loss_dict (line 77) | def get_train_loss_dict(self, step: int): method get_eval_loss_dict (line 95) | def get_eval_loss_dict(self, step: int): FILE: AutoReconForDens3R/nerfstudio/process_data/colmap_utils.py class ColmapCameraModel (line 61) | class ColmapCameraModel: class Camera (line 73) | class Camera: class Image (line 89) | class Image: class Point3D (line 109) | class Point3D: function get_colmap_version (line 143) | def get_colmap_version(colmap_cmd: str, default_version=3.8) -> float: function read_next_bytes (line 162) | def read_next_bytes(fid: BufferedReader, num_bytes: int, format_char_seq... function read_cameras_text (line 176) | def read_cameras_text(path: Path) -> Dict[int, Camera]: function read_cameras_binary (line 202) | def read_cameras_binary(path_to_model_file: Path) -> Dict[int, Camera]: function read_images_text (line 229) | def read_images_text(path: Path) -> Dict[int, Image]: function read_images_binary (line 266) | def read_images_binary(path_to_model_file: Path) -> Dict[int, Image]: function read_points3d_text (line 304) | def read_points3d_text(path) -> Dict[int, Point3D]: function read_points3d_binary (line 333) | def read_points3d_binary(path_to_model_file: Path) -> Dict[int, Point3D]: function detect_model_format (line 360) | def detect_model_format(path: Path, ext: str) -> bool: function read_model (line 381) | def read_model(path: Path, ext: Optional[str] = None) -> Tuple[Dict[int,... function qvec2rotmat (line 410) | def qvec2rotmat(qvec) -> np.ndarray: function rotmat2qvec (line 439) | def rotmat2qvec(R): function get_vocab_tree (line 466) | def get_vocab_tree() -> Path: function run_colmap (line 491) | def run_colmap( function colmap_to_json (line 578) | def colmap_to_json(cameras_path: Path, images_path: Path, output_dir: Pa... function get_matching_summary (line 654) | def get_matching_summary(num_intial_frames: int, num_matched_frames: int... FILE: AutoReconForDens3R/nerfstudio/process_data/hloc_utils.py function run_hloc (line 47) | def run_hloc( FILE: AutoReconForDens3R/nerfstudio/process_data/insta360_utils.py function get_insta360_filenames (line 30) | def get_insta360_filenames(data: Path) -> Tuple[Path, Path]: function convert_insta360_to_images (line 54) | def convert_insta360_to_images( function convert_insta360_single_file_to_images (line 125) | def convert_insta360_single_file_to_images( FILE: AutoReconForDens3R/nerfstudio/process_data/metashape_utils.py function _find_distortion_param (line 30) | def _find_distortion_param(calib_xml: ET.Element, param_name: str): function metashape_to_json (line 37) | def metashape_to_json( # pylint: disable=too-many-statements FILE: AutoReconForDens3R/nerfstudio/process_data/polycam_utils.py function polycam_to_json (line 30) | def polycam_to_json( FILE: AutoReconForDens3R/nerfstudio/process_data/process_data_utils.py class CameraModel (line 32) | class CameraModel(Enum): function get_num_frames_in_video (line 45) | def get_num_frames_in_video(video: Path) -> int: function convert_video_to_images (line 62) | def convert_video_to_images( function copy_images_list (line 111) | def copy_images_list( function copy_images (line 156) | def copy_images(data: Path, image_dir: Path, verbose) -> int: function downscale_images (line 175) | def downscale_images(image_dir: Path, num_downscales: int, verbose: bool... function find_tool_feature_matcher_combination (line 215) | def find_tool_feature_matcher_combination( FILE: AutoReconForDens3R/nerfstudio/process_data/record3d_utils.py function record3d_to_json (line 31) | def record3d_to_json(images_paths: List[Path], metadata_path: Path, outp... FILE: AutoReconForDens3R/nerfstudio/utils/bilateral_solver.py function rgb2yuv (line 16) | def rgb2yuv(im): function yuv2rgb (line 20) | def yuv2rgb(im): function get_valid_idx (line 24) | def get_valid_idx(valid, candidates): class BilateralGrid (line 35) | class BilateralGrid(object): method __init__ (line 36) | def __init__(self, im, sigma_spatial=32, sigma_luma=8, sigma_chroma=8): method _compute_factorization (line 53) | def _compute_factorization(self, coords_flat): method _hash_coords (line 79) | def _hash_coords(self, coord): method splat (line 83) | def splat(self, x): method slice (line 86) | def slice(self, y): method blur (line 89) | def blur(self, x): method filter (line 97) | def filter(self, x): function bistochastize (line 103) | def bistochastize(grid, maxiter=10): class BilateralSolver (line 117) | class BilateralSolver(object): method __init__ (line 118) | def __init__(self, grid, params): method solve (line 123) | def solve(self, x, w): function bilateral_refine (line 154) | def bilateral_refine(image, mask, sigma_spatial=24, sigma_luma=4, sigma_... FILE: AutoReconForDens3R/nerfstudio/utils/colormaps.py function apply_colormap (line 26) | def apply_colormap(image: TensorType["bs":..., 1], cmap="viridis") -> Te... function apply_depth_colormap (line 48) | def apply_depth_colormap( function apply_boolean_colormap (line 83) | def apply_boolean_colormap( FILE: AutoReconForDens3R/nerfstudio/utils/colors.py function get_color (line 36) | def get_color(color: Union[str, list]) -> TensorType[3]: FILE: AutoReconForDens3R/nerfstudio/utils/comms.py function is_dist_avail_and_initialized (line 21) | def is_dist_avail_and_initialized() -> bool: function get_world_size (line 26) | def get_world_size() -> int: function get_rank (line 33) | def get_rank() -> int: function get_local_rank (line 40) | def get_local_rank() -> int: function get_local_size (line 50) | def get_local_size() -> int: function is_main_process (line 60) | def is_main_process() -> bool: function synchronize (line 65) | def synchronize(): FILE: AutoReconForDens3R/nerfstudio/utils/decorators.py function decorate_all (line 23) | def decorate_all(decorators: List[Callable]) -> Callable: function check_profiler_enabled (line 40) | def check_profiler_enabled(func: Callable) -> Callable: function check_viewer_enabled (line 52) | def check_viewer_enabled(func: Callable) -> Callable: function check_eval_enabled (line 64) | def check_eval_enabled(func: Callable) -> Callable: function check_main_thread (line 76) | def check_main_thread(func: Callable) -> Callable: FILE: AutoReconForDens3R/nerfstudio/utils/eval_utils.py function eval_load_checkpoint (line 36) | def eval_load_checkpoint(config: cfg.TrainerConfig, pipeline: Pipeline) ... function eval_setup (line 71) | def eval_setup( FILE: AutoReconForDens3R/nerfstudio/utils/func_utils.py function get_first_element (line 5) | def get_first_element(elems: Sequence): FILE: AutoReconForDens3R/nerfstudio/utils/images.py class BasicImages (line 26) | class BasicImages: method __init__ (line 39) | def __init__(self, images: List): method to (line 46) | def to(self, device): FILE: AutoReconForDens3R/nerfstudio/utils/install_checks.py function check_ffmpeg_installed (line 25) | def check_ffmpeg_installed(): function check_colmap_installed (line 35) | def check_colmap_installed(): FILE: AutoReconForDens3R/nerfstudio/utils/io.py function load_from_json (line 23) | def load_from_json(filename: Path): function write_to_json (line 34) | def write_to_json(filename: Path, content: dict): FILE: AutoReconForDens3R/nerfstudio/utils/marching_cubes.py function get_surface_sliding (line 17) | def get_surface_sliding( function get_surface_occupancy (line 177) | def get_surface_occupancy( FILE: AutoReconForDens3R/nerfstudio/utils/mask_utils.py function alpha_composite (line 8) | def alpha_composite(rgb0: torch.Tensor, rgb1: torch.Tensor, function overlay_mask_on_image (line 17) | def overlay_mask_on_image(mask_rgb: TensorType["h", "w", 3], function compute_mask_indices (line 25) | def compute_mask_indices(collated_batch: Dict[str, Any]) -> Dict[str, Any]: FILE: AutoReconForDens3R/nerfstudio/utils/math.py function components_from_spherical_harmonics (line 23) | def components_from_spherical_harmonics(levels: int, directions: TensorT... class Gaussians (line 88) | class Gaussians: function compute_3d_gaussian (line 100) | def compute_3d_gaussian( function cylinder_to_gaussian (line 128) | def cylinder_to_gaussian( function conical_frustum_to_gaussian (line 153) | def conical_frustum_to_gaussian( function expected_sin (line 182) | def expected_sin(x_means: torch.Tensor, x_vars: torch.Tensor) -> torch.T... FILE: AutoReconForDens3R/nerfstudio/utils/misc.py function get_dict_to_torch (line 25) | def get_dict_to_torch(stuff: Any, device: Union[torch.device, str] = "cp... function get_dict_to_cpu (line 45) | def get_dict_to_cpu(stuff: Any): function get_masked_dict (line 60) | def get_masked_dict(d, mask): class IterableWrapper (line 74) | class IterableWrapper: # pylint: disable=too-few-public-methods method __init__ (line 99) | def __init__(self, new_iter: Callable, new_next: Callable, length: int... method __next__ (line 104) | def __next__(self): method __iter__ (line 110) | def __iter__(self): function scale_dict (line 116) | def scale_dict(dictionary: Dict[Any, Any], coefficients: Dict[str, float... function step_check (line 132) | def step_check(step, step_size, run_at_zero=False) -> bool: function update_avg (line 139) | def update_avg(prev_avg: float, new_val: float, step: int) -> float: FILE: AutoReconForDens3R/nerfstudio/utils/plotly_utils.py function color_str (line 38) | def color_str(color): function get_line_segments_from_lines (line 51) | def get_line_segments_from_lines( function vis_dataset (line 102) | def vis_dataset(camera_origins: TensorType["num_cameras", 3], ray_bundle... function get_random_color (line 150) | def get_random_color(colormap: Optional[List[str]] = None, idx: Optional... function get_sphere (line 167) | def get_sphere( function get_cube (line 207) | def get_cube( function get_gaussian_ellipsiod (line 247) | def get_gaussian_ellipsiod( function get_gaussian_ellipsoids_list (line 306) | def get_gaussian_ellipsoids_list( function get_frustum_mesh (line 349) | def get_frustum_mesh( function get_frustums_mesh_list (line 400) | def get_frustums_mesh_list( function get_frustum_points (line 424) | def get_frustum_points( function get_ray_bundle_lines (line 456) | def get_ray_bundle_lines( function vis_camera_rays (line 487) | def vis_camera_rays(cameras: Cameras) -> go.Figure: # type: ignore function get_camera_frustums (line 541) | def get_camera_frustums(cameras: Cameras): FILE: AutoReconForDens3R/nerfstudio/utils/pointclouds.py class BasicPointClouds (line 19) | class BasicPointClouds(nn.Module): method __init__ (line 33) | def __init__( method build_nn_search_index (line 70) | def build_nn_search_index(self, separate_bg_plane: bool = False): method sample_fg_points (line 103) | def sample_fg_points(self, n_pts: int, replace: bool = True): method sample_plane_points (line 116) | def sample_plane_points(self, n_pts: int, replace: bool = True, rand_s... method device (line 135) | def device(self): method _register_all_tensors_as_buffers (line 138) | def _register_all_tensors_as_buffers(self): FILE: AutoReconForDens3R/nerfstudio/utils/poses.py function to4x4 (line 23) | def to4x4(pose: TensorType[..., 3, 4]) -> TensorType[..., 4, 4]: function inverse (line 37) | def inverse(pose: TensorType[..., 3, 4]) -> TensorType[..., 3, 4]: function multiply (line 53) | def multiply(pose_a: TensorType[..., 3, 4], pose_b: TensorType[..., 3, 4... function normalize (line 70) | def normalize(poses: TensorType[..., 3, 4]) -> TensorType[..., 3, 4]: FILE: AutoReconForDens3R/nerfstudio/utils/printing.py function print_tcnn_speed_warning (line 24) | def print_tcnn_speed_warning(method_name: str): function human_format (line 36) | def human_format(num): FILE: AutoReconForDens3R/nerfstudio/utils/profiler.py function time_function (line 39) | def time_function(func: Callable) -> Callable: function time_function_cuda (line 57) | def time_function_cuda(func: Callable) -> Callable: function flush_profiler (line 75) | def flush_profiler(config: cfg.LoggingConfig): function setup_profiler (line 81) | def setup_profiler(config: cfg.LoggingConfig): class Profiler (line 88) | class Profiler: method __init__ (line 91) | def __init__(self, config: cfg.LoggingConfig): method update_time (line 95) | def update_time(self, func_name: str, start_time: float, end_time: flo... method print_profile (line 109) | def print_profile(self): FILE: AutoReconForDens3R/nerfstudio/utils/rich_utils.py class ItersPerSecColumn (line 34) | class ItersPerSecColumn(ProgressColumn): method __init__ (line 37) | def __init__(self, suffix="it/s") -> None: method render (line 41) | def render(self, task: "Task") -> Text: function status (line 49) | def status(msg: str, spinner: str = "bouncingBall", verbose: bool = False): function get_progress (line 62) | def get_progress(description: str, suffix: Optional[str] = None): FILE: AutoReconForDens3R/nerfstudio/utils/scheduler.py function cosine_annealing (line 7) | def cosine_annealing(cur_step, max_step=None, min_val=0.0, max_val=1.0): function _compute_exp_anneal_gamma (line 14) | def _compute_exp_anneal_gamma(max_step, min_val, max_val): function exp_annealing (line 18) | def exp_annealing(cur_step, max_step=None, min_val=0.0, max_val=1.0): function constant (line 23) | def constant(cur_step, max_step=None, min_val=0.0, max_val=1.0): FILE: AutoReconForDens3R/nerfstudio/utils/scripts.py function run_command (line 25) | def run_command(cmd: str, verbose=False) -> Optional[str]: FILE: AutoReconForDens3R/nerfstudio/utils/tensor_dataclass.py class TensorDataclass (line 27) | class TensorDataclass: method __post_init__ (line 67) | def __post_init__(self) -> None: method _get_dict_batch_shapes (line 97) | def _get_dict_batch_shapes(self, dict_: Dict) -> List: method _broadcast_dict_fields (line 120) | def _broadcast_dict_fields(self, dict_: Dict, batch_shape) -> Dict: method __getitem__ (line 149) | def __getitem__(self: TensorDataclassT, indices) -> TensorDataclassT: method __setitem__ (line 164) | def __setitem__(self, indices, value) -> NoReturn: method __len__ (line 167) | def __len__(self) -> int: method __bool__ (line 172) | def __bool__(self) -> bool: method shape (line 181) | def shape(self) -> Tuple[int, ...]: method size (line 186) | def size(self) -> int: method ndim (line 193) | def ndim(self) -> int: method reshape (line 197) | def reshape(self: TensorDataclassT, shape: Tuple[int, ...]) -> TensorD... method flatten (line 219) | def flatten(self: TensorDataclassT) -> TensorDataclassT: method broadcast_to (line 227) | def broadcast_to(self: TensorDataclassT, shape: Union[torch.Size, Tupl... method to (line 248) | def to(self: TensorDataclassT, device) -> TensorDataclassT: method _apply_fn_to_fields (line 259) | def _apply_fn_to_fields( method _apply_fn_to_dict (line 293) | def _apply_fn_to_dict( FILE: AutoReconForDens3R/nerfstudio/utils/vis_utils.py function interpolate_trajectory (line 14) | def interpolate_trajectory(cameras: Cameras, num_views: int = 300): FILE: AutoReconForDens3R/nerfstudio/utils/writer.py class EventName (line 43) | class EventName(enum.Enum): class EventType (line 57) | class EventType(enum.Enum): function put_image (line 67) | def put_image(name, image: TensorType["H", "W", "C"], step: int): function put_scalar (line 81) | def put_scalar(name: str, scalar: Any, step: int): function put_dict (line 96) | def put_dict(name: str, scalar_dict: Dict[str, Any], step: int): function put_config (line 108) | def put_config(name: str, config_dict: Dict[str, Any], step: int): function put_time (line 120) | def put_time(name: str, duration: float, step: int, avg_over_steps: bool... function write_out_storage (line 156) | def write_out_storage(): function setup_local_writer (line 169) | def setup_local_writer(config: cfg.LoggingConfig, max_iter: int, banner_... function setup_event_writer (line 191) | def setup_event_writer(config: cfg.Config, log_dir: Path) -> None: class Writer (line 215) | class Writer: method write_image (line 219) | def write_image(self, name: str, image: TensorType["H", "W", "C"], ste... method write_scalar (line 230) | def write_scalar(self, name: str, scalar: Union[float, torch.Tensor], ... method write_scalar_dict (line 241) | def write_scalar_dict(self, name: str, scalar_dict: Dict[str, Any], st... class TimeWriter (line 252) | class TimeWriter: method __init__ (line 255) | def __init__(self, writer, name, step=None, write=True): method __enter__ (line 264) | def __enter__(self): method __exit__ (line 269) | def __exit__(self, *args): class WandbWriter (line 284) | class WandbWriter(Writer): method __init__ (line 287) | def __init__(self, log_dir: Path, experiment_name: str): method write_image (line 290) | def write_image(self, name: str, image: TensorType["H", "W", "C"], ste... method write_scalar (line 294) | def write_scalar(self, name: str, scalar: Union[float, torch.Tensor], ... method write_config (line 297) | def write_config(self, name: str, config_dict: Dict[str, Any], step: i... class TensorboardWriter (line 309) | class TensorboardWriter(Writer): method __init__ (line 312) | def __init__(self, log_dir: Path): method write_image (line 315) | def write_image(self, name: str, image: TensorType["H", "W", "C"], ste... method write_scalar (line 319) | def write_scalar(self, name: str, scalar: Union[float, torch.Tensor], ... method write_config (line 322) | def write_config(self, name: str, config_dict: Dict[str, Any], step: i... function _cursorup (line 331) | def _cursorup(x: int): function _format_time (line 340) | def _format_time(seconds): class LocalWriter (line 361) | class LocalWriter: method __init__ (line 370) | def __init__(self, config: cfg.LocalWriterConfig, banner_messages: Opt... method write_stats_log (line 381) | def write_stats_log(self, step: int) -> None: method write_config (line 399) | def write_config(self, name: str, config_dict: Dict[str, Any], step: i... method _consolidate_events (line 407) | def _consolidate_events(self): method _update_header (line 418) | def _update_header(self, latest_map, new_key): method _print_stats (line 438) | def _print_stats(self, latest_map, padding=" "): FILE: AutoReconForDens3R/nerfstudio/viewer/app/public/electron.js function createWindow (line 6) | function createWindow() { FILE: AutoReconForDens3R/nerfstudio/viewer/app/run_deploy.py function run_cmd (line 30) | def run_cmd(cmd: str): function main (line 37) | def main( FILE: AutoReconForDens3R/nerfstudio/viewer/app/src/App.jsx function App (line 13) | function App() { FILE: AutoReconForDens3R/nerfstudio/viewer/app/src/SceneNode.js function dispose (line 5) | function dispose(object) { class SceneNode (line 29) | class SceneNode { method constructor (line 30) | constructor(object, scene_state) { method get_scene_state (line 40) | get_scene_state() { method add_child (line 44) | add_child(object) { method create_child (line 50) | create_child(name) { method find (line 57) | find(path) { method find_object (line 69) | find_object(path) { method find_no_create (line 73) | find_no_create(path) { method find_object_no_create (line 85) | find_object_no_create(path) { method set_property (line 89) | set_property(property, value) { method set_transform (line 102) | set_transform(matrix) { method set_object_from_path (line 112) | set_object_from_path(path, object) { method set_object (line 116) | set_object(object) { method dispose_recursive (line 124) | dispose_recursive() { method delete (line 131) | delete(path) { FILE: AutoReconForDens3R/nerfstudio/viewer/app/src/modules/Banner/Banner.jsx function getParam (line 10) | function getParam(param_name) { function Banner (line 21) | function Banner() { FILE: AutoReconForDens3R/nerfstudio/viewer/app/src/modules/ConfigPanel/ConfigPanel.jsx function RenderControls (line 9) | function RenderControls() { FILE: AutoReconForDens3R/nerfstudio/viewer/app/src/modules/LandingModal/LandingModal.jsx function TabPanel (line 28) | function TabPanel(props: TabPanelProps) { function a11yProps (line 47) | function a11yProps(index: number) { function LandingModel (line 54) | function LandingModel(props: LandingModalProps) { FILE: AutoReconForDens3R/nerfstudio/viewer/app/src/modules/LogPanel/LogPanel.jsx function LogPanel (line 8) | function LogPanel() { FILE: AutoReconForDens3R/nerfstudio/viewer/app/src/modules/RenderModal/RenderModal.jsx function RenderModal (line 13) | function RenderModal(props: RenderModalProps) { FILE: AutoReconForDens3R/nerfstudio/viewer/app/src/modules/Scene/Scene.jsx constant SCENE_BOX_NAME (line 20) | const SCENE_BOX_NAME = 'Scene Box'; constant CAMERAS_NAME (line 21) | const CAMERAS_NAME = 'Training Cameras'; function get_scene_tree (line 23) | function get_scene_tree() { function SceneTreeWebSocketListener (line 388) | function SceneTreeWebSocketListener() { FILE: AutoReconForDens3R/nerfstudio/viewer/app/src/modules/Scene/drawing.js function getCameraWireframe (line 48) | function getCameraWireframe( function drawCameraImagePlane (line 103) | function drawCameraImagePlane(width, height, imageString, name) { function transpose (line 116) | function transpose(matrix) { FILE: AutoReconForDens3R/nerfstudio/viewer/app/src/modules/SidePanel/CameraPanel/CameraHelper.js function setPoint (line 9) | function setPoint(point, pointMap, geometry, camera, x, y, z) { class CameraHelper (line 27) | class CameraHelper extends THREE.Mesh { method constructor (line 28) | constructor(camera, color = 0x000000) { method update (line 99) | update() { method set_visibility (line 138) | set_visibility(visible) { method dispose (line 143) | dispose() { FILE: AutoReconForDens3R/nerfstudio/viewer/app/src/modules/SidePanel/CameraPanel/CameraPanel.jsx constant FOV_LABELS (line 52) | const FOV_LABELS = { function set_camera_position (line 57) | function set_camera_position(camera, matrix) { function FovSelector (line 63) | function FovSelector(props) { function CameraList (line 198) | function CameraList(props) { function CameraPanel (line 422) | function CameraPanel(props) { FILE: AutoReconForDens3R/nerfstudio/viewer/app/src/modules/SidePanel/CameraPanel/curve.js function get_catmull_rom_curve (line 5) | function get_catmull_rom_curve(list_of_3d_vectors, is_cycle, smoothness_... function get_curve_object_from_cameras (line 17) | function get_curve_object_from_cameras( function get_transform_matrix (line 69) | function get_transform_matrix(position, lookat, up) { FILE: AutoReconForDens3R/nerfstudio/viewer/app/src/modules/SidePanel/ExportPanel/ExportPanel.jsx function TabPanel (line 20) | function TabPanel(props: TabPanelProps) { function a11yProps (line 40) | function a11yProps(index: number) { constant CLIPPING_BOX_NAME (line 47) | const CLIPPING_BOX_NAME = 'Clipping Box'; function ExportPanel (line 49) | function ExportPanel(props) { FILE: AutoReconForDens3R/nerfstudio/viewer/app/src/modules/SidePanel/ExportPanel/MeshSubPanel.jsx function get_normal_outputs (line 9) | function get_normal_outputs(output_options) { function MeshSubPanel (line 27) | function MeshSubPanel(props) { FILE: AutoReconForDens3R/nerfstudio/viewer/app/src/modules/SidePanel/ExportPanel/PointcloudSubPanel.jsx function PointcloudSubPanel (line 9) | function PointcloudSubPanel(props) { FILE: AutoReconForDens3R/nerfstudio/viewer/app/src/modules/SidePanel/ScenePanel/ScenePanel.jsx function MenuItems (line 47) | function MenuItems(props: ListItemProps) { function ClickableList (line 191) | function ClickableList(props: ClickableListProps) { function ScenePanel (line 207) | function ScenePanel(props) { FILE: AutoReconForDens3R/nerfstudio/viewer/app/src/modules/SidePanel/SidePanel.jsx function TabPanel (line 47) | function TabPanel(props: TabPanelProps) { function a11yProps (line 66) | function a11yProps(index: number) { function BasicTabs (line 77) | function BasicTabs(props: BasicTabsProps) { FILE: AutoReconForDens3R/nerfstudio/viewer/app/src/modules/SidePanel/StatusPanel/StatusPanel.jsx function StatusPanel (line 18) | function StatusPanel(props: StatusPanelProps) { FILE: AutoReconForDens3R/nerfstudio/viewer/app/src/modules/ViewerWindow/ViewerWindow.jsx function CameraToggle (line 17) | function CameraToggle() { function TransformIcons (line 47) | function TransformIcons(props) { function ViewerWindow (line 93) | function ViewerWindow(props) { FILE: AutoReconForDens3R/nerfstudio/viewer/app/src/modules/ViewportControlsModal/ViewportControlsModal.jsx function ControlsModal (line 7) | function ControlsModal() { FILE: AutoReconForDens3R/nerfstudio/viewer/app/src/modules/WebRtcWindow/WebRtcWindow.jsx function WebRtcWindow (line 11) | function WebRtcWindow() { FILE: AutoReconForDens3R/nerfstudio/viewer/app/src/modules/WebSocket/WebSocket.jsx function WebSocketContextFunction (line 13) | function WebSocketContextFunction({ children }) { FILE: AutoReconForDens3R/nerfstudio/viewer/app/src/modules/WebSocketUrlField.jsx function WebSocketUrlField (line 6) | function WebSocketUrlField() { FILE: AutoReconForDens3R/nerfstudio/viewer/app/src/reducer.js function setData (line 54) | function setData(newState, state, path, data) { function rootReducer (line 68) | function rootReducer(state = initialState, action) { FILE: AutoReconForDens3R/nerfstudio/viewer/app/src/subscriber.js function subscribe_to_changes (line 4) | function subscribe_to_changes(selector_fn, fn) { FILE: AutoReconForDens3R/nerfstudio/viewer/app/src/utils.js function split_path (line 1) | function split_path(path_str) { FILE: AutoReconForDens3R/nerfstudio/viewer/server/path.py class Path (line 24) | class Path: method __init__ (line 33) | def __init__(self, entries: Tuple = tuple()): method append (line 36) | def append(self, other: str) -> "Path": method lower (line 50) | def lower(self): method __hash__ (line 54) | def __hash__(self): method __eq__ (line 57) | def __eq__(self, other): FILE: AutoReconForDens3R/nerfstudio/viewer/server/server.py class WebSocketHandler (line 35) | class WebSocketHandler(tornado.websocket.WebSocketHandler): # pylint: d... method __init__ (line 38) | def __init__(self, *args, **kwargs): method check_origin (line 42) | def check_origin(self, origin): method open (line 46) | def open(self, *args: str, **kwargs: str): method on_message (line 52) | async def on_message(self, message: bytearray): # pylint: disable=inv... method on_close (line 88) | def on_close(self): class ZMQWebSocketBridge (line 93) | class ZMQWebSocketBridge: method __init__ (line 103) | def __init__(self, zmq_port: int, websocket_port: int, ip_address: str): method __str__ (line 122) | def __str__(self) -> str: method make_app (line 126) | def make_app(self): method handle_zmq (line 130) | def handle_zmq(self, frames: List[bytes]): method forward_to_websockets (line 157) | def forward_to_websockets( method setup_zmq (line 172) | def setup_zmq(self, url: str): method send_scene (line 184) | def send_scene(self, websocket: WebSocketHandler): method run (line 196) | def run(self): function run_viewer_bridge_server (line 201) | def run_viewer_bridge_server( function entrypoint (line 228) | def entrypoint(): FILE: AutoReconForDens3R/nerfstudio/viewer/server/state/node.py class Node (line 23) | class Node(defaultdict): method __init__ (line 28) | def __init__(self, *args, **kwargs): function get_tree (line 32) | def get_tree(node_class: Callable) -> Callable: function find_node (line 44) | def find_node(tree, path): function set_node_value (line 51) | def set_node_value(tree, path, value): function walk (line 58) | def walk(path, tree): FILE: AutoReconForDens3R/nerfstudio/viewer/server/state/state_node.py class StateNode (line 20) | class StateNode(Node): method __init__ (line 25) | def __init__(self, *args, **kwargs): FILE: AutoReconForDens3R/nerfstudio/viewer/server/subprocess.py function run_viewer_bridge_server_as_subprocess (line 36) | def run_viewer_bridge_server_as_subprocess( FILE: AutoReconForDens3R/nerfstudio/viewer/server/utils.py function get_chunks (line 28) | def get_chunks( function three_js_perspective_camera_focal_length (line 50) | def three_js_perspective_camera_focal_length(fov: float, image_height: i... function get_intrinsics_matrix_and_camera_to_world_h (line 65) | def get_intrinsics_matrix_and_camera_to_world_h( function find_available_port (line 98) | def find_available_port(func: Callable, default_port: int, max_attempts:... function force_codec (line 120) | def force_codec(pc: RTCPeerConnection, sender: RTCRtpSender, forced_code... FILE: AutoReconForDens3R/nerfstudio/viewer/server/video_stream.py class SingleFrameStreamTrack (line 22) | class SingleFrameStreamTrack(VideoStreamTrack): method __init__ (line 25) | def __init__(self): method put_frame (line 31) | def put_frame(self, frame: np.ndarray) -> None: method recv (line 39) | async def recv(self): FILE: AutoReconForDens3R/nerfstudio/viewer/server/viewer_utils.py function get_viewer_version (line 64) | def get_viewer_version() -> str: function setup_viewer (line 72) | def setup_viewer(config: cfg.ViewerConfig, log_filename: Path): class OutputTypes (line 83) | class OutputTypes(str, enum.Enum): class ColormapTypes (line 93) | class ColormapTypes(str, enum.Enum): class IOChangeException (line 104) | class IOChangeException(Exception): class SetTrace (line 108) | class SetTrace: method __init__ (line 111) | def __init__(self, func): method __enter__ (line 114) | def __enter__(self): method __exit__ (line 118) | def __exit__(self, ext_type, exc_value, traceback): class RenderThread (line 122) | class RenderThread(threading.Thread): method __init__ (line 131) | def __init__(self, state: "ViewerState", graph: Model, camera_ray_bund... method run (line 139) | def run(self): method join (line 159) | def join(self, timeout=None): class CheckThread (line 165) | class CheckThread(threading.Thread): method __init__ (line 172) | def __init__(self, state): method run (line 176) | def run(self): class ViewerState (line 222) | class ViewerState: method __init__ (line 229) | def __init__(self, config: cfg.ViewerConfig, log_filename: Path): method _pick_drawn_image_idxs (line 283) | def _pick_drawn_image_idxs(self, total_num: int) -> list[int]: method init_scene (line 299) | def init_scene(self, dataset: InputDataset, start_train=True) -> None: method _check_camera_path_payload (line 341) | def _check_camera_path_payload(self, trainer, step: int): method _check_webrtc_offer (line 354) | def _check_webrtc_offer(self): method update_scene (line 374) | def update_scene(self, trainer, step: int, graph: Model, num_rays_per_... method check_interrupt (line 438) | def check_interrupt(self, frame, event, arg): # pylint: disable=unuse... method _get_camera_object (line 447) | def _get_camera_object(self): method _apply_colormap (line 463) | def _apply_colormap(self, outputs: Dict[str, Any], colors: torch.Tenso... method send_webrtc_answer (line 514) | async def send_webrtc_answer(self, data): method set_image (line 565) | def set_image(self, image): method _send_output_to_viewer (line 570) | def _send_output_to_viewer(self, outputs: Dict[str, Any], colors: torc... method _update_viewer_stats (line 606) | def _update_viewer_stats(self, render_time: float, num_rays: int, imag... method _calculate_image_res (line 638) | def _calculate_image_res(self, camera_object, is_training: bool) -> Op... method _process_invalid_output (line 691) | def _process_invalid_output(self, output_type: str) -> str: method _render_image_in_viewer (line 714) | def _render_image_in_viewer(self, camera_object, graph: Model, is_trai... FILE: AutoReconForDens3R/nerfstudio/viewer/server/visualizer.py class ViewerWindow (line 31) | class ViewerWindow: method __init__ (line 41) | def __init__(self, zmq_port, ip_address="127.0.0.1"): method send (line 48) | def send(self, command): method send_ping (line 59) | def send_ping(self): method timeout_ping (line 73) | def timeout_ping(self, timeout_in_sec: int = 15): method assert_connected (line 94) | def assert_connected(self, timeout_in_sec: int = 15): class Viewer (line 110) | class Viewer: method __init__ (line 119) | def __init__( method view_into (line 131) | def view_into(window: ViewerWindow, path: Path): method __getitem__ (line 137) | def __getitem__(self, path): method __repr__ (line 140) | def __repr__(self): method write (line 143) | def write(self, data: Union[Dict, str, None] = None): method read (line 148) | def read(self): method delete (line 153) | def delete(self): FILE: AutoReconForDens3R/scripts/blender/render_mesh_blender.py function load_K_Rt_from_P (line 20) | def load_K_Rt_from_P(filename, P=None): function load_all_poses (line 44) | def load_all_poses(cam_path, img_dir): function parse_images_from_camera_dict (line 59) | def parse_images_from_camera_dict(camera_path, image_dir): function load_object_aabb (line 73) | def load_object_aabb(object_anno_path, camera_anno_path) -> np.ndarray: function set_background_color (line 85) | def set_background_color(color): function parse_args (line 98) | def parse_args(): function compute_ground_poision (line 120) | def compute_ground_poision(obj_aabb): function scale_aabb_to_unit_sphere (line 127) | def scale_aabb_to_unit_sphere(all_poses, obj_aabb): function main (line 135) | def main(): FILE: AutoReconForDens3R/scripts/completions/install.py function _check_tyro_cli (line 38) | def _check_tyro_cli(script_path: pathlib.Path) -> bool: function _generate_completion (line 73) | def _generate_completion( function _exclamation (line 128) | def _exclamation() -> str: function _update_rc (line 132) | def _update_rc( function main (line 194) | def main(mode: ConfigureMode = "install") -> None: function entrypoint (line 281) | def entrypoint(): FILE: AutoReconForDens3R/scripts/datasets/extract_monocular_cues.py function standardize_depth_map (line 122) | def standardize_depth_map(img, mask_valid=None, trunc_value=0.1): function save_outputs (line 142) | def save_outputs(img_path, output_file_name): FILE: AutoReconForDens3R/scripts/datasets/process_nerfstudio_to_sdfstudio.py function main (line 15) | def main(): FILE: AutoReconForDens3R/scripts/datasets/process_neuralrgbd_to_sdfstudio.py function alphanum_key (line 31) | def alphanum_key(s): function load_poses (line 38) | def load_poses(posefile): FILE: AutoReconForDens3R/scripts/datasets/process_nuscenes_masks.py class ProcessNuScenesMasks (line 17) | class ProcessNuScenesMasks: method main (line 41) | def main(self) -> None: function entrypoint (line 124) | def entrypoint(): FILE: AutoReconForDens3R/scripts/docs/add_nb_tags.py function main (line 13) | def main(check: bool = False): FILE: AutoReconForDens3R/scripts/docs/build_docs.py function run_command (line 15) | def run_command(command: str) -> None: function main (line 27) | def main(clean_cache: bool = False): FILE: AutoReconForDens3R/scripts/eval.py class ComputePSNR (line 20) | class ComputePSNR: method main (line 28) | def main(self) -> None: function entrypoint (line 46) | def entrypoint(): FILE: AutoReconForDens3R/scripts/eval_mask.py function evenly_sample_elems (line 20) | def evenly_sample_elems(elems, n_samples): function read_mask (line 31) | def read_mask(mask_path): function resize_mask (line 40) | def resize_mask(gt_mask, pred_mask, max_side_length=None): function mask_to_boundary (line 69) | def mask_to_boundary(mask, dilation_ratio=0.02): function boundary_iou (line 91) | def boundary_iou(gt, dt, dilation_ratio=0.02, vis_path=None, img_path=No... function mask_iou (line 107) | def mask_iou(gt, dt, vis_path=None, img_path=None): class EvalMask (line 136) | class EvalMask: method main (line 147) | def main(self) -> None: method parse_filenames (line 180) | def parse_filenames(self, gt_filenames, pred_filenames): method eval_frame (line 194) | def eval_frame(self, fn): method gather_results (line 229) | def gather_results(self, all_metrics): function entrypoint (line 255) | def entrypoint(): FILE: AutoReconForDens3R/scripts/exporter.py class Exporter (line 32) | class Exporter: class ExportPointCloud (line 42) | class ExportPointCloud(Exporter): method main (line 66) | def main(self) -> None: class ExportTSDFMesh (line 100) | class ExportTSDFMesh(Exporter): method main (line 141) | def main(self) -> None: method _config_override_fn (line 194) | def _config_override_fn(self, config): class ExportPoissonMesh (line 207) | class ExportPoissonMesh(Exporter): method validate_pipeline (line 252) | def validate_pipeline(self, pipeline: Pipeline) -> None: method main (line 277) | def main(self) -> None: method _config_override_fn (line 354) | def _config_override_fn(self, config): class ExportMarchingCubesMesh (line 361) | class ExportMarchingCubesMesh(Exporter): method main (line 367) | def main(self) -> None: function entrypoint (line 380) | def entrypoint(): FILE: AutoReconForDens3R/scripts/extract_mesh.py class ExtractMesh (line 36) | class ExtractMesh: method main (line 100) | def main(self) -> None: method extract_sdf (line 159) | def extract_sdf(self, pipeline, x): method _remove_internal_geometry (line 166) | def _remove_internal_geometry(self, output_path: Path) -> Path: method _remove_internal_geometry_igl (line 176) | def _remove_internal_geometry_igl( method _remove_internal_geometry_meshlab (line 192) | def _remove_internal_geometry_meshlab(self, output_path: Path) -> Path: method _remove_isolated_components (line 206) | def _remove_isolated_components(self, output_path: Path) -> Path: method _fix_mesh (line 230) | def _fix_mesh(self, output_path: Path) -> Path: method _extract_texture (line 253) | def _extract_texture(self, pipeline, mesh_path: Path) -> Path: method _config_override_fn (line 268) | def _config_override_fn(self, config): # TODO: make optional function entrypoint (line 288) | def entrypoint(): FILE: AutoReconForDens3R/scripts/extract_volume.py class ExtractVolume (line 24) | class ExtractVolume: method main (line 46) | def main(self) -> None: method _config_override_fn (line 78) | def _config_override_fn(self, config): method extract_seg_volume (line 87) | def extract_seg_volume(self, pipeline, volume_extractor): method extract_sdf_volume (line 91) | def extract_sdf_volume(self, pipeline, volume_extractor): function extract_volume (line 103) | def extract_volume( function postprocess_volume_for_mrc (line 156) | def postprocess_volume_for_mrc(volume, volume_type): function compute_volume_and_pad_val (line 167) | def compute_volume_and_pad_val(volume: np.ndarray, volume_type: str): function entrypoint (line 182) | def entrypoint(): FILE: AutoReconForDens3R/scripts/github/run_actions.py function run_command (line 16) | def run_command(command: str, continue_on_fail: bool = False) -> bool: function run_github_actions_file (line 31) | def run_github_actions_file(filename: str, continue_on_fail: bool = False): function run_code_checks (line 77) | def run_code_checks(continue_on_fail: bool = False): function entrypoint (line 89) | def entrypoint(): FILE: AutoReconForDens3R/scripts/heritage_to_nerfstudio.py class CameraModel (line 24) | class CameraModel(Enum): function show_result (line 42) | def show_result(seg): class Renderer (line 52) | class Renderer: method __init__ (line 53) | def __init__(self, height=480, width=640): method __call__ (line 58) | def __call__(self, height, width, intrinsics, pose, mesh): method fix_pose (line 70) | def fix_pose(self, pose): method mesh_opengl (line 80) | def mesh_opengl(self, mesh): method delete (line 83) | def delete(self): function colmap_to_json (line 87) | def colmap_to_json( FILE: AutoReconForDens3R/scripts/preprocess/preprocess_neus_pose.py class PreprocessNeusPose (line 22) | class PreprocessNeusPose: method main (line 28) | def main(self): method load_camera_dict (line 46) | def load_camera_dict(self): method renormalize_scene (line 55) | def renormalize_scene(self, camera_dict): method _parse_images_from_camera_dict (line 63) | def _parse_images_from_camera_dict(self): FILE: AutoReconForDens3R/scripts/process_data.py class ProcessImages (line 33) | class ProcessImages: method main (line 83) | def main(self) -> None: class ProcessVideo (line 156) | class ProcessVideo: method main (line 209) | def main(self) -> None: class ProcessInsta360 (line 281) | class ProcessInsta360: method main (line 316) | def main(self) -> None: class ProcessRecord3D (line 400) | class ProcessRecord3D: method main (line 423) | def main(self) -> None: class ProcessPolycam (line 477) | class ProcessPolycam: method main (line 511) | def main(self) -> None: class ProcessMetashape (line 616) | class ProcessMetashape: method main (line 643) | def main(self) -> None: function entrypoint (line 723) | def entrypoint(): FILE: AutoReconForDens3R/scripts/render.py function _save_gt_video (line 44) | def _save_gt_video( function _render_trajectory_video (line 78) | def _render_trajectory_video( function save_frame (line 157) | def save_frame( function save_video (line 183) | def save_video( class RenderTrajectory (line 213) | class RenderTrajectory: method main (line 252) | def main(self) -> None: method _config_override_fn (line 305) | def _config_override_fn(self, config): function entrypoint (line 319) | def entrypoint(): FILE: AutoReconForDens3R/scripts/render_mesh.py function _render_trajectory_video (line 33) | def _render_trajectory_video( class RenderTrajectory (line 150) | class RenderTrajectory: method main (line 177) | def main(self) -> None: function entrypoint (line 228) | def entrypoint(): FILE: AutoReconForDens3R/scripts/texture.py class TextureMesh (line 23) | class TextureMesh: method main (line 43) | def main(self) -> None: function entrypoint (line 68) | def entrypoint(): FILE: AutoReconForDens3R/scripts/train.py function _find_free_port (line 66) | def _find_free_port() -> str: function _set_random_seed (line 75) | def _set_random_seed(seed) -> None: function train_loop (line 82) | def train_loop(local_rank: int, world_size: int, config: cfg.Config, glo... function _distributed_worker (line 96) | def _distributed_worker( function launch (line 152) | def launch( function main (line 223) | def main(config: cfg.Config) -> None: function entrypoint (line 250) | def entrypoint(): FILE: AutoReconForDens3R/scripts/viewer/view_dataset.py function main (line 27) | def main( FILE: AutoReconForDens3R/tests/cameras/test_cameras.py function test_pinhole_camera (line 110) | def test_pinhole_camera(): function test_equirectangular_camera (line 124) | def test_equirectangular_camera(): function test_camera_as_tensordataclass (line 162) | def test_camera_as_tensordataclass(): function check_generate_rays_shape (line 204) | def check_generate_rays_shape(): function _check_dataclass_allclose (line 281) | def _check_dataclass_allclose(ipt, other): function _check_cam_shapes (line 288) | def _check_cam_shapes(cam: Cameras, _batch_size): FILE: AutoReconForDens3R/tests/cameras/test_rays.py function test_frustum_get_position (line 11) | def test_frustum_get_position(): function test_frustum_get_gaussian_blob (line 33) | def test_frustum_get_gaussian_blob(): function test_frustum_apply_masks (line 49) | def test_frustum_apply_masks(): function test_get_mock_frustum (line 69) | def test_get_mock_frustum(): FILE: AutoReconForDens3R/tests/field_components/test_embedding.py function test_indexing (line 7) | def test_indexing(): FILE: AutoReconForDens3R/tests/field_components/test_encodings.py function test_scaling_and_offset (line 10) | def test_scaling_and_offset(): function test_nerf_encoder (line 28) | def test_nerf_encoder(): function test_rff_encoder (line 57) | def test_rff_encoder(): function test_tensor_vm_encoder (line 77) | def test_tensor_vm_encoder(): function test_tensor_cp_encoder (line 101) | def test_tensor_cp_encoder(): function test_tensor_sh_encoder (line 124) | def test_tensor_sh_encoder(): function test_tensor_hash_encoder (line 142) | def test_tensor_hash_encoder(): FILE: AutoReconForDens3R/tests/field_components/test_field_outputs.py function test_field_output (line 17) | def test_field_output(): function test_density_output (line 36) | def test_density_output(): function test_rgb_output (line 46) | def test_rgb_output(): function test_sh_output (line 56) | def test_sh_output(): FILE: AutoReconForDens3R/tests/field_components/test_fields.py function test_tcnn_instant_ngp_field (line 10) | def test_tcnn_instant_ngp_field(): FILE: AutoReconForDens3R/tests/field_components/test_mlp.py function test_mlp (line 10) | def test_mlp(): FILE: AutoReconForDens3R/tests/field_components/test_temporal_distortions.py function test_dnerf_distortion (line 9) | def test_dnerf_distortion(): FILE: AutoReconForDens3R/tests/model_components/test_ray_sampler.py function test_uniform_sampler (line 18) | def test_uniform_sampler(): function test_lin_disp_sampler (line 37) | def test_lin_disp_sampler(): function test_sqrt_sampler (line 54) | def test_sqrt_sampler(): function test_log_sampler (line 71) | def test_log_sampler(): function test_pdf_sampler (line 88) | def test_pdf_sampler(): FILE: AutoReconForDens3R/tests/model_components/test_renderers.py function test_rgb_renderer (line 11) | def test_rgb_renderer(): function test_sh_renderer (line 28) | def test_sh_renderer(): function test_acc_renderer (line 46) | def test_acc_renderer(): function test_depth_renderer (line 59) | def test_depth_renderer(): FILE: AutoReconForDens3R/tests/test_train.py function set_reduced_config (line 20) | def set_reduced_config(config: Config): function test_train (line 49) | def test_train(): FILE: AutoReconForDens3R/tests/utils/test_poses.py function test_to4x4 (line 10) | def test_to4x4(): function test_multiply (line 24) | def test_multiply(): function test_inverse (line 60) | def test_inverse(): function test_normalize (line 84) | def test_normalize(): FILE: AutoReconForDens3R/tests/utils/test_tensor_dataclass.py class DummyNestedClass (line 14) | class DummyNestedClass(TensorDataclass): class DummyTensorDataclass (line 21) | class DummyTensorDataclass(TensorDataclass): function test_init (line 30) | def test_init(): function test_broadcasting (line 44) | def test_broadcasting(): function test_tensor_ops (line 64) | def test_tensor_ops(): # pylint: disable=(too-many-statements) function test_nested_class (line 136) | def test_nested_class(): function test_iter (line 169) | def test_iter(): FILE: AutoReconForDens3R/tests/utils/test_visualization.py function test_apply_colormap (line 9) | def test_apply_colormap(): function test_apply_depth_colormap (line 19) | def test_apply_depth_colormap(): function test_apply_boolean_colormap (line 31) | def test_apply_boolean_colormap(): FILE: AutoReconForDens3R/third_party/AutoDecomp/auto_decomp/cli/inference_transformer.py class Config (line 34) | class Config: function _update_config (line 61) | def _update_config(cfg): function main (line 68) | def main(config: Config): FILE: AutoReconForDens3R/third_party/AutoDecomp/auto_decomp/decomp/preprocess.py class SfMPreprocessConfig (line 33) | class SfMPreprocessConfig: class SfMPreprocessWorker (line 76) | class SfMPreprocessWorker: method __init__ (line 77) | def __init__(self, cfg: DictConfig, inst_rel_dir: str, device: str = "... method _init_paths (line 91) | def _init_paths(self, inst_rel_dir: str): method _handle_resume (line 124) | def _handle_resume(self) -> bool: method _sanity_check (line 133) | def _sanity_check(self) -> None: method update_thresholds (line 154) | def update_thresholds(self): method __call__ (line 178) | def __call__(self) -> Optional[Dict[str, Any]]: method parse_poses (line 205) | def parse_poses(self, data): method build_neural_ptcd (line 208) | def build_neural_ptcd(self, data): method stat_outlier_removal (line 226) | def stat_outlier_removal(self, data): method voxel_downsample (line 232) | def voxel_downsample(self, data): method filter_pts_behind_cameras (line 235) | def filter_pts_behind_cameras(self, data): method filter_ground_plane_pts (line 238) | def filter_ground_plane_pts(self, data): method save_transformer_inference (line 241) | def save_transformer_inference(self): method save_ncut_inference (line 247) | def save_ncut_inference(self): function ray_wrapper (line 255) | def ray_wrapper(cfg, task_id=None, **kwargs) -> Tuple[bool, str, Optiona... class SfMPreprocess (line 272) | class SfMPreprocess: method __init__ (line 273) | def __init__(self, cfg: DictConfig): method __call__ (line 280) | def __call__(self) -> Optional[Dict[str, Any]]: method setup_logger (line 312) | def setup_logger(self): FILE: AutoReconForDens3R/third_party/AutoDecomp/auto_decomp/decomp/transformer/dataset/utils.py function normalize_pointcloud (line 15) | def normalize_pointcloud(data): function to_numpy (line 43) | def to_numpy(obj): function move_to (line 67) | def move_to(obj, device): function build_inference_batch (line 86) | def build_inference_batch( function compute_average_camera_position (line 124) | def compute_average_camera_position(poses: Dict[str, Dict[str, np.ndarra... FILE: AutoReconForDens3R/third_party/AutoDecomp/auto_decomp/decomp/transformer/lightning/module.py function if_none_then_true (line 29) | def if_none_then_true(x): class SfMTransformerConfig (line 34) | class SfMTransformerConfig: class SfMTransformer (line 63) | class SfMTransformer(pl.LightningModule): method __init__ (line 64) | def __init__(self, cfg: SfMTransformerConfig): method forward (line 73) | def forward(self, data): method inference (line 79) | def inference(self, batch): method test_val_step (line 82) | def test_val_step(self, batch, batch_idx, mode="test"): method validation_step (line 106) | def validation_step(self, batch, batch_idx): method test_step (line 109) | def test_step(self, batch, batch_idx): method build_aux_attrs (line 112) | def build_aux_attrs(self, batch, mode: Literal["train", "test", "val"]): FILE: AutoReconForDens3R/third_party/AutoDecomp/auto_decomp/decomp/transformer/modeling/encoding.py function expected_sin (line 27) | def expected_sin(x_means: torch.Tensor, x_vars: torch.Tensor) -> torch.T... class FieldComponent (line 39) | class FieldComponent(nn.Module): method __init__ (line 46) | def __init__(self, in_dim: Optional[int] = None, out_dim: Optional[int... method build_nn_modules (line 51) | def build_nn_modules(self) -> None: method set_in_dim (line 55) | def set_in_dim(self, in_dim: int) -> None: method get_out_dim (line 64) | def get_out_dim(self) -> int: method forward (line 71) | def forward(self, in_tensor: TensorType["bs":..., "input_dim"]) -> Ten... class Encoding (line 80) | class Encoding(FieldComponent): method __init__ (line 87) | def __init__(self, in_dim: int) -> None: method forward (line 93) | def forward(self, in_tensor: TensorType["bs":..., "input_dim"]) -> Ten... class NeRFEncoding (line 102) | class NeRFEncoding(Encoding): method __init__ (line 114) | def __init__( method get_out_dim (line 124) | def get_out_dim(self) -> int: method forward (line 132) | def forward( class RFFEncoding (line 165) | class RFFEncoding(Encoding): method __init__ (line 175) | def __init__(self, in_dim: int, num_frequencies: int, scale: float, in... method get_out_dim (line 188) | def get_out_dim(self) -> int: method forward (line 191) | def forward( FILE: AutoReconForDens3R/third_party/AutoDecomp/auto_decomp/decomp/transformer/modeling/point_transformer.py class PointTransformer (line 14) | class PointTransformer(nn.Module): method __init__ (line 15) | def __init__( method _init_weights (line 124) | def _init_weights(self, m): method interpolate_pos_encoding (line 133) | def interpolate_pos_encoding(self, x: Tensor): method prepare_tokens (line 157) | def prepare_tokens(self, xyz: Tensor, feat: Tensor, cls_token: Optiona... method forward (line 190) | def forward(self, xyz: Tensor, feat: Tensor, cls_token: Optional[Tenso... class VoxelEmbedding (line 198) | class VoxelEmbedding(nn.Module): method __init__ (line 199) | def __init__(self, voxel_resolution=128, in_chans=384, embed_dim=384): method forward (line 205) | def forward(self, xyz: Tensor, feat: Tensor): class PosFeatMerger (line 211) | class PosFeatMerger(nn.Module): method __init__ (line 212) | def __init__(self, pos_dim, feat_dim, hidden_dim, out_dim, act_layer=n... method forward (line 218) | def forward(self, pos, feat): class TokenSegHead (line 222) | class TokenSegHead(nn.Module): method __init__ (line 223) | def __init__(self, use_softmax=False, temperature=1.0, num_classes=2): method forward (line 229) | def forward(self, tokens: Tensor, cls_tokens: Tensor) -> Tensor: FILE: AutoReconForDens3R/third_party/AutoDecomp/auto_decomp/decomp/transformer/modeling/utils.py function _no_grad_trunc_normal_ (line 9) | def _no_grad_trunc_normal_(tensor, mean, std, a, b): function trunc_normal_ (line 46) | def trunc_normal_(tensor, mean=0., std=1., a=-2., b=2.): function elu_feature_map (line 51) | def elu_feature_map(x): class LinearAttention (line 55) | class LinearAttention(nn.Module): method __init__ (line 56) | def __init__(self, eps=1e-6): method forward (line 61) | def forward(self, queries, keys, values): FILE: AutoReconForDens3R/third_party/AutoDecomp/auto_decomp/decomp/transformer/modeling/vision_transformer.py function drop_path (line 27) | def drop_path(x, drop_prob: float = 0.0, training: bool = False): class DropPath (line 38) | class DropPath(nn.Module): method __init__ (line 41) | def __init__(self, drop_prob=None): method forward (line 45) | def forward(self, x): class Mlp (line 49) | class Mlp(nn.Module): method __init__ (line 50) | def __init__(self, in_features, hidden_features=None, out_features=Non... method forward (line 59) | def forward(self, x): class Attention (line 68) | class Attention(nn.Module): method __init__ (line 69) | def __init__( method forward (line 86) | def forward(self, x): class Block (line 105) | class Block(nn.Module): method __init__ (line 106) | def __init__( method forward (line 136) | def forward(self, x, return_attention=False): class PatchEmbed (line 145) | class PatchEmbed(nn.Module): method __init__ (line 148) | def __init__(self, img_size=224, patch_size=16, in_chans=3, embed_dim=... method forward (line 157) | def forward(self, x): class VisionTransformer (line 163) | class VisionTransformer(nn.Module): method __init__ (line 166) | def __init__( method _init_weights (line 222) | def _init_weights(self, m): method interpolate_pos_encoding (line 231) | def interpolate_pos_encoding(self, x, w, h): method prepare_tokens (line 253) | def prepare_tokens(self, x): method forward (line 266) | def forward(self, x): method get_last_selfattention (line 273) | def get_last_selfattention(self, x): method get_intermediate_layers (line 282) | def get_intermediate_layers(self, x, n=1): function vit_tiny (line 293) | def vit_tiny(patch_size=16, **kwargs): function vit_small (line 307) | def vit_small(patch_size=16, **kwargs): function vit_base (line 321) | def vit_base(patch_size=16, **kwargs): class DINOHead (line 335) | class DINOHead(nn.Module): method __init__ (line 336) | def __init__( method _init_weights (line 361) | def _init_weights(self, m): method forward (line 367) | def forward(self, x): FILE: AutoReconForDens3R/third_party/AutoDecomp/auto_decomp/decomp/transformer/utils/postprocess.py function _to_numpy (line 35) | def _to_numpy(x: Union[np.ndarray, Tensor]) -> Tuple[np.ndarray, Any]: function _to_tensor (line 45) | def _to_tensor(x: Union[np.ndarray, Tensor], device: str) -> Tensor: function extract_max_cluster (line 49) | def extract_max_cluster( function remove_outliers (line 73) | def remove_outliers( function segment_fg_supporting_plane (line 97) | def segment_fg_supporting_plane( function extract_object_centric_fg_aabb (line 220) | def extract_object_centric_fg_aabb( class PostprocessConfig (line 264) | class PostprocessConfig: class Postprocess (line 318) | class Postprocess: method __init__ (line 319) | def __init__(self, cfg: PostprocessConfig): method __call__ (line 325) | def __call__( method run_euclidean_clustering (line 401) | def run_euclidean_clustering(self): method run_outlier_removal (line 424) | def run_outlier_removal(self): method run_plane_alignment (line 447) | def run_plane_alignment(self): method _segment_support_plane (line 460) | def _segment_support_plane(self): method _align_world_with_plane (line 474) | def _align_world_with_plane(self): method compute_object_centric_transformation (line 492) | def compute_object_centric_transformation(self): method build_fg_aabb (line 517) | def build_fg_aabb(self): method _extract_object_centric_fg_aabb (line 540) | def _extract_object_centric_fg_aabb(self) -> Float[Tensor, "batch 8 3"]: method extract_bbox_no_postprocess (line 557) | def extract_bbox_no_postprocess( method compute_full_decomposition (line 571) | def compute_full_decomposition(self) -> Dict[str, Dict[str, Any]]: method set_bg_in_bbox_to_fg (line 607) | def set_bg_in_bbox_to_fg(self, fg_pts, bg_pts, fg_aabb_min_max_info): method _build_decomposed_pts (line 629) | def _build_decomposed_pts(self, b_id: int, decomp_results: Dict[str, A... method _build_decomposed_aux_attrs (line 687) | def _build_decomposed_aux_attrs(self, b_id: int, decomp_results: Dict[... FILE: AutoReconForDens3R/third_party/AutoDecomp/auto_decomp/decomp/transformer/utils/saving.py class SfMTransformerSaveConfig (line 17) | class SfMTransformerSaveConfig: class SaveResults (line 41) | class SaveResults: method __init__ (line 42) | def __init__(self, cfg: SfMTransformerSaveConfig): method _sanity_check (line 46) | def _sanity_check(self): method save_annotations (line 58) | def save_annotations(self, batch: Dict[str, Any], all_decomp_results: ... method _compute_transform_new_object_frame (line 106) | def _compute_transform_new_object_frame( method _build_object_anno (line 128) | def _build_object_anno( method _build_pose_anno (line 160) | def _build_pose_anno( method visualize_object_anno (line 197) | def visualize_object_anno(self, object_anno: Dict[str, Any], save_path... method save_decomp_visualization (line 230) | def save_decomp_visualization(self): FILE: AutoReconForDens3R/third_party/AutoDecomp/auto_decomp/feature_extraction/dino_vit/extract_features.py class DINOExtractionConfig (line 26) | class DINOExtractionConfig: method __post_init__ (line 50) | def __post_init__(self): function resize_max_area (line 69) | def resize_max_area(h, w, max_area): function get_sfm_dir (line 77) | def get_sfm_dir(args, inst_rel_dir): function _save_feat_dirname_to_cache (line 91) | def _save_feat_dirname_to_cache(args, inst_rel_dir, feat_dirname): function extract_single (line 103) | def extract_single(args, inst_rel_dir): function ray_wrapper (line 168) | def ray_wrapper(*args, worker=None, task_id=None, **kwargs): function main (line 181) | def main(dino_extraction: DictConfig): FILE: AutoReconForDens3R/third_party/AutoDecomp/auto_decomp/feature_extraction/dino_vit/vit_extractor.py class ViTExtractor (line 25) | class ViTExtractor: method __init__ (line 37) | def __init__( method create_model (line 74) | def create_model(model_type: str) -> nn.Module: method _fix_pos_enc (line 99) | def _fix_pos_enc(patch_size: int, stride_hw: Tuple[int, int]): method patch_vit_resolution (line 139) | def patch_vit_resolution(model: nn.Module, stride: int) -> nn.Module: method preprocess (line 161) | def preprocess( method _get_hook (line 179) | def _get_hook(self, facet: str): method _register_hooks (line 206) | def _register_hooks(self, layers: List[int], facet: str) -> None: method _unregister_hooks (line 223) | def _unregister_hooks(self) -> None: method _extract_features (line 231) | def _extract_features( method _log_bin (line 259) | def _log_bin(self, x: torch.Tensor, hierarchy: int = 2) -> torch.Tensor: method extract_descriptors (line 308) | def extract_descriptors( method extract_saliency_maps (line 369) | def extract_saliency_maps(self, batch: torch.Tensor) -> torch.Tensor: method reshape_descriptor (line 390) | def reshape_descriptor(self, descriptor: torch.Tensor, bin: bool = Fal... function extract_single (line 402) | def extract_single(args): function str2bool (line 454) | def str2bool(v): function parse_args (line 465) | def parse_args(): function main (line 512) | def main(): FILE: AutoReconForDens3R/third_party/AutoDecomp/auto_decomp/sfm/colmap_from_idr.py class IDR2COLMAPConfig (line 20) | class IDR2COLMAPConfig: function load_K_Rt_from_P (line 34) | def load_K_Rt_from_P(filename, P=None): function parse_idr_poses (line 58) | def parse_idr_poses(pose_path, image_dir) -> Dict[str, Dict[str, np.ndar... function process_instance (line 78) | def process_instance(cfg, inst_dirname): function main (line 115) | def main(idr2colmap: IDR2COLMAPConfig): FILE: AutoReconForDens3R/third_party/AutoDecomp/auto_decomp/sfm/pairs_from_sequential.py function read_image_pairs (line 19) | def read_image_pairs(path): function pairs_from_file (line 25) | def pairs_from_file(image_list: List[Path], pairs_path: Path) -> List[Tu... function build_loop_detection_pairs (line 38) | def build_loop_detection_pairs( function main (line 74) | def main( FILE: AutoReconForDens3R/third_party/AutoDecomp/auto_decomp/sfm/sfm.py class SfMConfig (line 37) | class SfMConfig: class SparseReconActor (line 160) | class SparseReconActor: method __init__ (line 161) | def __init__(self, task_queue, return_queue, share_intrinsics=True, ma... method run (line 168) | def run(self): method _run (line 176) | def _run(self, task_inst_id): method sparse_recon (line 207) | def sparse_recon(self, task): method triangulate (line 225) | def triangulate(self, task): method manhattan_alignment (line 244) | def manhattan_alignment(self, task): function extract_and_match (line 277) | def extract_and_match(task): # FeatureTask class FeatureActor (line 288) | class FeatureActor: method __init__ (line 289) | def __init__(self, feature_queue: Queue, recon_queue: Queue): method run (line 293) | def run(self): method _run_task (line 305) | def _run_task(self, task): method _extract_and_match (line 315) | def _extract_and_match(self, task): # FeatureTask method _run_loftr (line 322) | def _run_loftr(self, task): # FeatureTask function read_write_cache (line 340) | def read_write_cache( function evenly_sample_images (line 374) | def evenly_sample_images(images, n_samples): function parse_images_from_reference_model (line 386) | def parse_images_from_reference_model(sfm_ref_dir): function plot_reconstruction (line 393) | def plot_reconstruction(rec, outputs_dir, show_axes=False, suffix=None): function manhattan_alignment (line 405) | def manhattan_alignment(image_dir, sfm_dir, output_dir): function reconstruct_instance (line 423) | def reconstruct_instance( function reconstruct_instance_wrapper (line 552) | def reconstruct_instance_wrapper(cli_args, *args, **kwargs): function main (line 571) | def main(sfm: SfMConfig): function dens3r_recon (line 636) | def dens3r_recon(sfm: SfMConfig): function postprocess_dens3r (line 694) | def postprocess_dens3r(sfm: SfMConfig, sfm_dir_name: str): FILE: AutoReconForDens3R/third_party/AutoDecomp/auto_decomp/utils/aggregation.py function multiview_feature_extraction (line 13) | def multiview_feature_extraction( function multiview_feature_aggregation (line 97) | def multiview_feature_aggregation(p3Ds_xyz, p3Ds_mv_feats, mode="mean", ... function gini (line 130) | def gini(array, eps=1e-7): function gini_feature_aggregation (line 153) | def gini_feature_aggregation(mv_feats, mode="max"): function affinity_feature_aggregation (line 166) | def affinity_feature_aggregation(mv_feats, mode): FILE: AutoReconForDens3R/third_party/AutoDecomp/auto_decomp/utils/cli.py function str2bool (line 4) | def str2bool(v): FILE: AutoReconForDens3R/third_party/AutoDecomp/auto_decomp/utils/colmap.py function read_extrinsics (line 5) | def read_extrinsics(rec: pycolmap.Reconstruction): function parse_poses (line 16) | def parse_poses(rec): FILE: AutoReconForDens3R/third_party/AutoDecomp/auto_decomp/utils/geometry/box3d.py function min_max_to_vertices_torch (line 33) | def min_max_to_vertices_torch(box: Tensor) -> Tensor: function min_max_to_vertices (line 48) | def min_max_to_vertices(box): function _vertices_to_min_max (line 61) | def _vertices_to_min_max(box): function _center_extent_to_vertices (line 66) | def _center_extent_to_vertices(box): # extent:=half_extent function _vertices_to_center_extent (line 71) | def _vertices_to_center_extent(box): function _center_extent_to_min_max (line 78) | def _center_extent_to_min_max(box): function convert_box_format (line 95) | def convert_box_format(box, src_format, tgt_format): function compute_box3d_iou (line 109) | def compute_box3d_iou( function normalize (line 131) | def normalize(array: np.ndarray, axis: int = -1): function extend_aabb (line 137) | def extend_aabb( FILE: AutoReconForDens3R/third_party/AutoDecomp/auto_decomp/utils/geometry/pointcloud/convert.py function open3d_ptcd_from_numpy (line 7) | def open3d_ptcd_from_numpy( FILE: AutoReconForDens3R/third_party/AutoDecomp/auto_decomp/utils/geometry/pointcloud/misc.py function compute_knn_distance_threshold (line 13) | def compute_knn_distance_threshold( function compute_world_to_object_transformation (line 35) | def compute_world_to_object_transformation( FILE: AutoReconForDens3R/third_party/AutoDecomp/auto_decomp/utils/geometry/pointcloud/plane.py function fit_plane (line 18) | def fit_plane(points: np.ndarray, method: Literal["OLS", "TLS"] = "OLS",... function fit_plane_ols (line 30) | def fit_plane_ols(points: np.ndarray): function fit_plane_tls (line 46) | def fit_plane_tls(points: np.ndarray): function fit_plane_tls_torch (line 59) | def fit_plane_tls_torch(points: np.ndarray, device="cuda", max_n_points=... function point_to_plane_dist (line 86) | def point_to_plane_dist(points: Float[np.ndarray, "N 3"], plane: Float[n... function ground_plane_alignment (line 94) | def ground_plane_alignment( function ground_plane_alignment_simple (line 152) | def ground_plane_alignment_simple(plane: Plane, source_vector: np.ndarra... FILE: AutoReconForDens3R/third_party/AutoDecomp/auto_decomp/utils/geometry/pointcloud/pointcloud.py function build_color_ptcd_from_colmap (line 26) | def build_color_ptcd_from_colmap( function build_feature_ptcd (line 53) | def build_feature_ptcd( class NeuralPointCloud (line 128) | class NeuralPointCloud: method __init__ (line 131) | def __init__( method from_pycolmap_rec (line 158) | def from_pycolmap_rec( method statistical_outlier_removal (line 194) | def statistical_outlier_removal(self, n_neighbors, std_ratio): method to_open3d_ptcd (line 199) | def to_open3d_ptcd(self): # points only, no attributes method update_ptcd_with_indices (line 202) | def update_ptcd_with_indices(self, inds): method update_ptcd_with_mask (line 208) | def update_ptcd_with_mask(self): method visualize_ptcd (line 211) | def visualize_ptcd(self, base_name): method _draw_color_ptcd (line 216) | def _draw_color_ptcd(self, base_name): method _draw_feature_ptcd (line 225) | def _draw_feature_ptcd(self, base_name): FILE: AutoReconForDens3R/third_party/AutoDecomp/auto_decomp/utils/geometry/pointcloud/segmentation.py function euclidean_clustering (line 17) | def euclidean_clustering( function segment_planes (line 65) | def segment_planes( FILE: AutoReconForDens3R/third_party/AutoDecomp/auto_decomp/utils/geometry/transform.py function rotmat (line 4) | def rotmat(a, b): FILE: AutoReconForDens3R/third_party/AutoDecomp/auto_decomp/utils/misc.py class EarlyTermination (line 2) | class EarlyTermination(Exception): FILE: AutoReconForDens3R/third_party/AutoDecomp/auto_decomp/utils/ray.py function is_ray_environment (line 4) | def is_ray_environment(): FILE: AutoReconForDens3R/third_party/AutoDecomp/auto_decomp/utils/tqdm.py function tqdm (line 7) | def tqdm(*args, **kwargs): FILE: AutoReconForDens3R/third_party/AutoDecomp/auto_decomp/utils/viz_3d.py function to_homogeneous (line 19) | def to_homogeneous(points): function init_figure (line 24) | def init_figure( function plot_points (line 65) | def plot_points( function plot_mesh (line 98) | def plot_mesh( function plot_cube (line 124) | def plot_cube( function plot_sphere (line 185) | def plot_sphere( function plot_camera (line 216) | def plot_camera( function plot_camera_colmap (line 281) | def plot_camera_colmap( function plot_cameras (line 295) | def plot_cameras(fig: go.Figure, reconstruction: pycolmap.Reconstruction... function plot_coordinate_frames (line 301) | def plot_coordinate_frames( function plot_reconstruction (line 344) | def plot_reconstruction( function save_fig (line 382) | def save_fig(fig, save_dir: Path, fig_name: str, mode: Union[str, Plotly... FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/colmap_from_nvm.py function recover_database_images_and_ids (line 13) | def recover_database_images_and_ids(database_path): function quaternion_to_rotation_matrix (line 27) | def quaternion_to_rotation_matrix(qvec): function camera_center_to_translation (line 37) | def camera_center_to_translation(c, qvec): function read_nvm_model (line 42) | def read_nvm_model( function main (line 165) | def main(nvm, intrinsics, database, output, skip_points=False): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/extract_features.py function resize_image (line 145) | def resize_image(image, size, interp): class ImageDataset (line 163) | class ImageDataset(torch.utils.data.Dataset): method __init__ (line 172) | def __init__(self, root, conf, paths=None): method __getitem__ (line 199) | def __getitem__(self, idx): method __len__ (line 223) | def __len__(self): function main (line 228) | def main(conf: Dict, FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/extractors/cosplace.py class CosPlace (line 23) | class CosPlace(BaseModel): method _init (line 29) | def _init(self, conf): method _forward (line 41) | def _forward(self, data): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/extractors/d2net.py class D2Net (line 14) | class D2Net(BaseModel): method _init (line 23) | def _init(self, conf): method _forward (line 36) | def _forward(self, data): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/extractors/dir.py class DIR (line 25) | class DIR(BaseModel): method _init (line 43) | def _init(self, conf): method _forward (line 59) | def _forward(self, data): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/extractors/disk.py class DISK (line 14) | class DISK(BaseModel): method _init (line 24) | def _init(self, conf): method _forward (line 51) | def _forward(self, data): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/extractors/dog.py function sift_to_rootsift (line 14) | def sift_to_rootsift(x): class DoG (line 21) | class DoG(BaseModel): method _init (line 36) | def _init(self, conf): method to (line 47) | def to(self, *args, **kwargs): method _forward (line 57) | def _forward(self, data): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/extractors/netvlad.py class NetVLADLayer (line 18) | class NetVLADLayer(nn.Module): method __init__ (line 19) | def __init__(self, input_dim=512, K=64, score_bias=False, intranorm=Tr... method forward (line 29) | def forward(self, x): class NetVLAD (line 43) | class NetVLAD(BaseModel): method _init (line 57) | def _init(self, conf): method _forward (line 129) | def _forward(self, data): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/extractors/openibl.py class OpenIBL (line 7) | class OpenIBL(BaseModel): method _init (line 13) | def _init(self, conf): method _forward (line 20) | def _forward(self, data): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/extractors/r2d2.py class R2D2 (line 12) | class R2D2(BaseModel): method _init (line 26) | def _init(self, conf): method _forward (line 36) | def _forward(self, data): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/extractors/superpoint.py function sample_descriptors_fix_sampling (line 13) | def sample_descriptors_fix_sampling(keypoints, descriptors, s: int = 8): class SuperPoint (line 26) | class SuperPoint(BaseModel): method _init (line 37) | def _init(self, conf): method _forward (line 42) | def _forward(self, data): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/localize_inloc.py function interpolate_scan (line 16) | def interpolate_scan(scan, kp): function get_scan_pose (line 38) | def get_scan_pose(dataset_dir, rpath): function pose_from_cluster (line 61) | def pose_from_cluster(dataset_dir, q, retrieved, feature_file, match_file, function main (line 114) | def main(dataset_dir, retrieval, features, matches, results, FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/localize_sfm.py function do_covisibility_clustering (line 15) | def do_covisibility_clustering(frame_ids: List[int], class QueryLocalizer (line 51) | class QueryLocalizer: method __init__ (line 52) | def __init__(self, reconstruction, config=None): method localize (line 56) | def localize(self, points2D_all, points2D_idxs, points3D_id, query_cam... function pose_from_cluster (line 67) | def pose_from_cluster( function main (line 126) | def main(reference_sfm: Union[Path, pycolmap.Reconstruction], FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/match_dense.py function to_cpts (line 88) | def to_cpts(kpts, ps): function assign_keypoints (line 94) | def assign_keypoints(kpts: np.ndarray, function get_grouped_ids (line 132) | def get_grouped_ids(array): function get_unique_matches (line 143) | def get_unique_matches(match_ids, scores): function matches_to_matches0 (line 155) | def matches_to_matches0(matches, scores): function kpids_to_matches0 (line 166) | def kpids_to_matches0(kpt_ids0, kpt_ids1, scores): function scale_keypoints (line 177) | def scale_keypoints(kpts, scale): class ImagePairDataset (line 183) | class ImagePairDataset(torch.utils.data.Dataset): method __init__ (line 191) | def __init__(self, image_dir, conf, pairs): method preprocess (line 205) | def preprocess(self, image: np.ndarray): method __len__ (line 232) | def __len__(self): method __getitem__ (line 235) | def __getitem__(self, idx): function match_dense (line 249) | def match_dense(conf: Dict, function load_keypoints (line 305) | def load_keypoints(conf: Dict, function aggregate_matches (line 341) | def aggregate_matches( function assign_matches (line 435) | def assign_matches( function match_and_assign (line 465) | def match_and_assign(conf: Dict, function main (line 524) | def main(conf: Dict, FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/match_features.py class WorkQueue (line 72) | class WorkQueue(): method __init__ (line 73) | def __init__(self, work_fn, num_threads=1): method join (line 82) | def join(self): method thread_fn (line 88) | def thread_fn(self, work_fn): method put (line 94) | def put(self, data): class FeaturePairsDataset (line 98) | class FeaturePairsDataset(torch.utils.data.Dataset): method __init__ (line 99) | def __init__(self, pairs, feature_path_q, feature_path_r): method __getitem__ (line 104) | def __getitem__(self, idx): method __len__ (line 120) | def __len__(self): function writer_fn (line 124) | def writer_fn(inp, match_path): function main (line 137) | def main(conf: Dict, function find_unique_new_pairs (line 165) | def find_unique_new_pairs(pairs_all: List[Tuple[str]], match_path: Path ... function match_from_paths (line 187) | def match_from_paths(conf: Dict, FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/matchers/__init__.py function get_matcher (line 1) | def get_matcher(matcher): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/matchers/adalam.py class AdaLAM (line 9) | class AdaLAM(BaseModel): method _init (line 31) | def _init(self, conf): method _forward (line 34) | def _forward(self, data): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/matchers/loftr.py class LoFTR (line 9) | class LoFTR(BaseModel): method _init (line 20) | def _init(self, conf): method _forward (line 25) | def _forward(self, data): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/matchers/nearest_neighbor.py function find_nn (line 6) | def find_nn(sim, ratio_thresh, distance_thresh): function mutual_check (line 19) | def mutual_check(m0, m1): class NearestNeighbor (line 27) | class NearestNeighbor(BaseModel): method _init (line 35) | def _init(self, conf): method _forward (line 38) | def _forward(self, data): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/matchers/superglue.py class SuperGlue (line 10) | class SuperGlue(BaseModel): method _init (line 21) | def _init(self, conf): method _forward (line 24) | def _forward(self, data): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/pairs_from_covisibility.py function main (line 11) | def main(model, output, num_matched): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/pairs_from_exhaustive.py function main (line 11) | def main( FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/pairs_from_poses.py function get_pairwise_distances (line 13) | def get_pairwise_distances(images): function main (line 42) | def main(model, output, num_matched, rotation_threshold=DEFAULT_ROT_THRE... FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/pairs_from_retrieval.py function parse_names (line 15) | def parse_names(prefix, names, names_all): function get_descriptors (line 36) | def get_descriptors(names, path, name2idx=None, key='global_descriptor'): function pairs_from_score_matrix (line 48) | def pairs_from_score_matrix(scores: torch.Tensor, function main (line 70) | def main(descriptors, output, num_matched, FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/pipelines/4Seasons/utils.py function get_timestamps (line 13) | def get_timestamps(files, idx): function delete_unused_images (line 29) | def delete_unused_images(root, timestamps): function camera_from_calibration_file (line 41) | def camera_from_calibration_file(id_, path): function parse_poses (line 56) | def parse_poses(path, colmap=False): function parse_relocalization (line 76) | def parse_relocalization(path, has_poses=False): function build_empty_colmap_model (line 96) | def build_empty_colmap_model(root, sfm_dir): function generate_query_lists (line 131) | def generate_query_lists(timestamps, seq_dir, out_path): function generate_localization_pairs (line 142) | def generate_localization_pairs(sequence, reloc, num, ref_pairs, out_path): function prepare_submission (line 174) | def prepare_submission(results, relocs, poses_path, out_dir): function evaluate_submission (line 204) | def evaluate_submission(submission_dir, relocs, ths=[0.1, 0.2, 0.5]): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/pipelines/7Scenes/create_gt_sfm.py function scene_coordinates (line 11) | def scene_coordinates(p2D, R_w2c, t_w2c, depth, camera): function interpolate_depth (line 21) | def interpolate_depth(depth, kp): function image_path_to_rendered_depth_path (line 43) | def image_path_to_rendered_depth_path(image_name): function project_to_image (line 51) | def project_to_image(p3D, R, t, camera, eps: float = 1e-4, pad: int = 1): function correct_sfm_with_gt_depth (line 63) | def correct_sfm_with_gt_depth(sfm_path, depth_folder_path, output_path): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/pipelines/7Scenes/pipeline.py function run_scene (line 14) | def run_scene(images, gt_dir, retrieval, outputs, results, num_covis, FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/pipelines/7Scenes/utils.py function create_reference_sfm (line 9) | def create_reference_sfm(full_model, ref_model, blacklist=None, ext='.bi... FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/pipelines/CMU/pipeline.py function generate_query_list (line 10) | def generate_query_list(dataset, path, slice_): function run_slice (line 29) | def run_slice(slice_, root, outputs, num_covis, num_loc): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/pipelines/Cambridge/pipeline.py function run_scene (line 13) | def run_scene(images, gt_dir, outputs, results, num_covis, num_loc): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/pipelines/Cambridge/utils.py function scale_sfm_images (line 12) | def scale_sfm_images(full_model, scaled_model, image_dir): function create_query_list_with_intrinsics (line 44) | def create_query_list_with_intrinsics(model, out, list_file=None, ext='.... function evaluate (line 84) | def evaluate(model, results, list_file=None, ext='.bin', only_localized=... FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/pipelines/RobotCar/colmap_from_nvm.py function read_nvm_model (line 17) | def read_nvm_model( function main (line 137) | def main(nvm, database, output, skip_points=False): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/pipelines/RobotCar/pipeline.py function generate_query_list (line 13) | def generate_query_list(dataset, image_dir, path): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/reconstruction.py function create_empty_db (line 15) | def create_empty_db(database_path: Path): function import_images (line 26) | def import_images(image_dir: Path, function get_image_ids (line 43) | def get_image_ids(database_path: Path) -> Dict[str, int]: function run_reconstruction (line 52) | def run_reconstruction(sfm_dir: Path, function main (line 93) | def main(sfm_dir: Path, FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/triangulation.py class OutputCapture (line 18) | class OutputCapture: method __init__ (line 19) | def __init__(self, verbose: bool): method __enter__ (line 22) | def __enter__(self): method __exit__ (line 27) | def __exit__(self, exc_type, *args): function create_db_from_model (line 35) | def create_db_from_model(reconstruction: pycolmap.Reconstruction, function import_features (line 57) | def import_features(image_ids: Dict[str, int], function import_matches (line 72) | def import_matches(image_ids: Dict[str, int], function estimation_and_geometric_verification (line 103) | def estimation_and_geometric_verification(database_path: Path, function geometric_verification (line 114) | def geometric_verification(image_ids: Dict[str, int], function run_triangulation (line 181) | def run_triangulation(model_path: Path, function main (line 200) | def main(sfm_dir: Path, function parse_option_args (line 239) | def parse_option_args(args: List[str], default_options) -> Dict[str, Any]: FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/utils/base_model.py class BaseModel (line 8) | class BaseModel(nn.Module, metaclass=ABCMeta): method __init__ (line 12) | def __init__(self, conf): method forward (line 20) | def forward(self, data): method _init (line 27) | def _init(self, conf): method _forward (line 32) | def _forward(self, data): function dynamic_load (line 37) | def dynamic_load(root, model): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/utils/database.py function image_ids_to_pair_id (line 115) | def image_ids_to_pair_id(image_id1, image_id2): function pair_id_to_image_ids (line 121) | def pair_id_to_image_ids(pair_id): function array_to_blob (line 127) | def array_to_blob(array): function blob_to_array (line 134) | def blob_to_array(blob, dtype, shape=(-1,)): class COLMAPDatabase (line 141) | class COLMAPDatabase(sqlite3.Connection): method connect (line 144) | def connect(database_path): method __init__ (line 148) | def __init__(self, *args, **kwargs): method add_camera (line 166) | def add_camera(self, model, width, height, params, method add_image (line 175) | def add_image(self, name, camera_id, method add_keypoints (line 184) | def add_keypoints(self, image_id, keypoints): method add_descriptors (line 193) | def add_descriptors(self, image_id, descriptors): method add_matches (line 199) | def add_matches(self, image_id1, image_id2, matches): method add_two_view_geometry (line 212) | def add_two_view_geometry(self, image_id1, image_id2, matches, function example_usage (line 236) | def example_usage(): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/utils/geometry.py function to_homogeneous (line 5) | def to_homogeneous(p): function vector_to_cross_product_matrix (line 9) | def vector_to_cross_product_matrix(v): function compute_epipolar_errors (line 17) | def compute_epipolar_errors(qvec_r2t, tvec_r2t, p2d_r, p2d_t): function pose_matrix_from_qvec_tvec (line 32) | def pose_matrix_from_qvec_tvec(qvec, tvec): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/utils/io.py function read_image (line 10) | def read_image(path, grayscale=False): function list_h5_names (line 23) | def list_h5_names(path): function get_keypoints (line 33) | def get_keypoints(path: Path, name: str, function find_pair (line 44) | def find_pair(hfile: h5py.File, name0: str, name1: str): function get_matches (line 63) | def get_matches(path: Path, name0: str, name1: str) -> Tuple[np.ndarray]: FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/utils/parsers.py function parse_image_list (line 10) | def parse_image_list(path, with_intrinsics=False): function parse_image_lists (line 31) | def parse_image_lists(paths, with_intrinsics=False): function parse_retrieval (line 40) | def parse_retrieval(path): function names_to_pair (line 51) | def names_to_pair(name0, name1, separator='/'): function names_to_pair_old (line 55) | def names_to_pair_old(name0, name1): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/utils/read_write_model.py class Image (line 52) | class Image(BaseImage): method qvec2rotmat (line 53) | def qvec2rotmat(self): function read_next_bytes (line 76) | def read_next_bytes(fid, num_bytes, format_char_sequence, endian_charact... function write_next_bytes (line 88) | def write_next_bytes(fid, data, format_char_sequence, endian_character="... function read_cameras_text (line 104) | def read_cameras_text(path): function read_cameras_binary (line 130) | def read_cameras_binary(path_to_model_file): function write_cameras_text (line 159) | def write_cameras_text(cameras, path): function write_cameras_binary (line 176) | def write_cameras_binary(cameras, path_to_model_file): function read_images_text (line 196) | def read_images_text(path): function read_images_binary (line 227) | def read_images_binary(path_to_model_file): function write_images_text (line 262) | def write_images_text(images, path): function write_images_binary (line 290) | def write_images_binary(images, path_to_model_file): function read_points3D_text (line 311) | def read_points3D_text(path): function read_points3D_binary (line 338) | def read_points3D_binary(path_to_model_file): function write_points3D_text (line 368) | def write_points3D_text(points3D, path): function write_points3D_binary (line 393) | def write_points3D_binary(points3D, path_to_model_file): function detect_model_format (line 412) | def detect_model_format(path, ext): function read_model (line 421) | def read_model(path, ext=""): function write_model (line 449) | def write_model(cameras, images, points3D, path, ext=".bin"): function qvec2rotmat (line 461) | def qvec2rotmat(qvec): function rotmat2qvec (line 474) | def rotmat2qvec(R): function main (line 488) | def main(): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/utils/viz.py function cm_RdGn (line 15) | def cm_RdGn(x): function plot_images (line 22) | def plot_images(imgs, titles=None, cmaps='gray', dpi=100, pad=.5, function plot_keypoints (line 56) | def plot_keypoints(kpts, colors='lime', ps=4): function plot_matches (line 70) | def plot_matches(kpts0, kpts1, color=None, lw=1.5, ps=4, indices=(0, 1),... function add_text (line 112) | def add_text(idx, text, pos=(0.01, 0.99), fs=15, color='w', function save_plot (line 123) | def save_plot(path, **kw): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/utils/viz_3d.py function to_homogeneous (line 17) | def to_homogeneous(points): function init_figure (line 22) | def init_figure(height: int = 800) -> go.Figure: function plot_points (line 59) | def plot_points( function plot_camera (line 75) | def plot_camera( function plot_camera_colmap (line 124) | def plot_camera_colmap( function plot_cameras (line 140) | def plot_cameras( function plot_reconstruction (line 150) | def plot_reconstruction( FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/hloc/visualization.py function visualize_sfm_2d (line 12) | def visualize_sfm_2d(reconstruction, image_dir, color_by='visibility', function visualize_loc (line 56) | def visualize_loc(results, image_dir, reconstruction=None, db_image_dir=... function visualize_loc_from_log (line 79) | def visualize_loc_from_log(image_dir, query_name, loc, reconstruction=None, FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/SuperGluePretrainedNetwork/models/matching.py class Matching (line 49) | class Matching(torch.nn.Module): method __init__ (line 51) | def __init__(self, config={}): method forward (line 56) | def forward(self, data): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/SuperGluePretrainedNetwork/models/superglue.py function MLP (line 51) | def MLP(channels: List[int], do_bn: bool = True) -> nn.Module: function normalize_keypoints (line 65) | def normalize_keypoints(kpts, image_shape): class KeypointEncoder (line 75) | class KeypointEncoder(nn.Module): method __init__ (line 77) | def __init__(self, feature_dim: int, layers: List[int]) -> None: method forward (line 82) | def forward(self, kpts, scores): function attention (line 87) | def attention(query: torch.Tensor, key: torch.Tensor, value: torch.Tenso... class MultiHeadedAttention (line 94) | class MultiHeadedAttention(nn.Module): method __init__ (line 96) | def __init__(self, num_heads: int, d_model: int): method forward (line 104) | def forward(self, query: torch.Tensor, key: torch.Tensor, value: torch... class AttentionalPropagation (line 112) | class AttentionalPropagation(nn.Module): method __init__ (line 113) | def __init__(self, feature_dim: int, num_heads: int): method forward (line 119) | def forward(self, x: torch.Tensor, source: torch.Tensor) -> torch.Tensor: class AttentionalGNN (line 124) | class AttentionalGNN(nn.Module): method __init__ (line 125) | def __init__(self, feature_dim: int, layer_names: List[str]) -> None: method forward (line 132) | def forward(self, desc0: torch.Tensor, desc1: torch.Tensor) -> Tuple[t... function log_sinkhorn_iterations (line 143) | def log_sinkhorn_iterations(Z: torch.Tensor, log_mu: torch.Tensor, log_n... function log_optimal_transport (line 152) | def log_optimal_transport(scores: torch.Tensor, alpha: torch.Tensor, ite... function arange_like (line 175) | def arange_like(x, dim: int): class SuperGlue (line 179) | class SuperGlue(nn.Module): method __init__ (line 206) | def __init__(self, config): method forward (line 230) | def forward(self, data): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/SuperGluePretrainedNetwork/models/superpoint.py function simple_nms (line 47) | def simple_nms(scores, nms_radius: int): function remove_borders (line 65) | def remove_borders(keypoints, scores, border: int, height: int, width: i... function top_k_keypoints (line 73) | def top_k_keypoints(keypoints, scores, k: int): function sample_descriptors (line 80) | def sample_descriptors(keypoints, descriptors, s: int = 8): class SuperPoint (line 95) | class SuperPoint(nn.Module): method __init__ (line 111) | def __init__(self, config): method forward (line 145) | def forward(self, data): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/SuperGluePretrainedNetwork/models/utils.py class AverageTimer (line 57) | class AverageTimer: method __init__ (line 60) | def __init__(self, smoothing=0.3, newline=False): method reset (line 67) | def reset(self): method update (line 74) | def update(self, name='default'): method print (line 83) | def print(self, text='Timer'): class VideoStreamer (line 99) | class VideoStreamer: method __init__ (line 106) | def __init__(self, basedir, resize, skip, image_glob, max_length=10000... method load_image (line 159) | def load_image(self, impath): method next_frame (line 175) | def next_frame(self): method start_ip_camera_thread (line 213) | def start_ip_camera_thread(self): method update_ip_camera (line 220) | def update_ip_camera(self): method cleanup (line 235) | def cleanup(self): function process_resize (line 240) | def process_resize(w, h, resize): function frame2tensor (line 259) | def frame2tensor(frame, device): function read_image (line 263) | def read_image(path, device, resize, rotation, resize_float): function estimate_pose (line 288) | def estimate_pose(kpts0, kpts1, K0, K1, thresh, conf=0.99999): function rotate_intrinsics (line 315) | def rotate_intrinsics(K, image_shape, rot): function rotate_pose_inplane (line 335) | def rotate_pose_inplane(i_T_w, rot): function scale_intrinsics (line 346) | def scale_intrinsics(K, scales): function to_homogeneous (line 351) | def to_homogeneous(points): function compute_epipolar_error (line 355) | def compute_epipolar_error(kpts0, kpts1, T_0to1, K0, K1): function angle_error_mat (line 377) | def angle_error_mat(R1, R2): function angle_error_vec (line 383) | def angle_error_vec(v1, v2): function compute_pose_error (line 388) | def compute_pose_error(T_0to1, R, t): function pose_auc (line 397) | def pose_auc(errors, thresholds): function plot_image_pair (line 415) | def plot_image_pair(imgs, dpi=100, size=6, pad=.5): function plot_keypoints (line 429) | def plot_keypoints(kpts0, kpts1, color='w', ps=2): function plot_matches (line 435) | def plot_matches(kpts0, kpts1, color, lw=1.5, ps=4): function make_matching_plot (line 452) | def make_matching_plot(image0, image1, kpts0, kpts1, mkpts0, mkpts1, function make_matching_plot_fast (line 484) | def make_matching_plot_fast(image0, image1, kpts0, kpts1, mkpts0, function error_colormap (line 553) | def error_colormap(x): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/d2net/train.py function process_epoch (line 156) | def process_epoch( FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/deep-image-retrieval/dirtorch/datasets/__main__.py function viz_dataset (line 7) | def viz_dataset(db, nr=6, nc=6): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/deep-image-retrieval/dirtorch/datasets/create.py class DatasetCreator (line 5) | class DatasetCreator: method __init__ (line 15) | def __init__(self, globs): method __call__ (line 19) | def __call__(self, dataset_cmd ): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/deep-image-retrieval/dirtorch/datasets/dataset.py class Dataset (line 8) | class Dataset(object): method __len__ (line 36) | def __len__(self): method get_filename (line 39) | def get_filename(self, img_idx, root=None): method get_key (line 42) | def get_key(self, img_idx): method key_to_index (line 45) | def key_to_index(self, key): method get_image (line 50) | def get_image(self, img_idx, resize=None): method get_image_size (line 57) | def get_image_size(self, img_idx): method get_label (line 60) | def get_label(self, img_idx, toint=False): method has_label (line 63) | def has_label(self): method get_query_db (line 67) | def get_query_db(self): method get_query_groundtruth (line 70) | def get_query_groundtruth(self, query_idx, what='AP'): method eval_query_AP (line 83) | def eval_query_AP(self, query_idx, scores): method eval_query_top (line 94) | def eval_query_top(self, query_idx, scores, k=(1,5,10,20,50,100)): method original (line 103) | def original(self): method __repr__ (line 106) | def __repr__(self): function split (line 122) | def split( dataset, train_prop, val_prop=0, method='balanced' ): class SubDataset (line 201) | class SubDataset(Dataset): method __init__ (line 204) | def __init__(self, dataset, indices): method get_key (line 213) | def get_key(self, i): method get_label (line 216) | def get_label(self, i, **kw): method get_bbox (line 219) | def get_bbox(self, i, **kw): method __repr__ (line 225) | def __repr__(self): method viz_distr (line 231) | def viz_distr(self): class CatDataset (line 241) | class CatDataset(Dataset): method __init__ (line 244) | def __init__(self, *datasets): method which (line 275) | def which(self, i): method get (line 280) | def get(self, i, attr): method __getattr__ (line 284) | def __getattr__(self, name): method call (line 292) | def call(self, i, func, *args, **kwargs): method get_key (line 296) | def get_key(self, i): method get_label (line 301) | def get_label(self, i, toint=False): method get_bbox (line 305) | def get_bbox(self,i): method get_polygons (line 309) | def get_polygons(self,i,**kw): function deploy (line 316) | def deploy( dataset, target_dir, transforms=None, redo=False, ext=None, ... class DeployedDataset (line 404) | class DeployedDataset(Dataset): method __init__ (line 407) | def __init__(self, dataset, root, imsizes=None, trfs=None, ext=None): method __repr__ (line 442) | def __repr__(self): method __len__ (line 447) | def __len__(self): method get_key (line 450) | def get_key(self, i): method get_something (line 455) | def get_something(self, what, i, *args, **fmt): method get_bbox (line 464) | def get_bbox(self, i, **kw): method get_polygons (line 467) | def get_polygons(self, i, *args, **kw): method get_label_map (line 470) | def get_label_map(self, i, *args, **kw): method get_instance_map (line 473) | def get_instance_map(self, i, *args, **kw): method get_angle_map (line 476) | def get_angle_map(self, i, *args, **kw): method original (line 480) | def original(self): function deploy_and_split (line 485) | def deploy_and_split( trainset, deploy_trf=None, deploy_dir='/dev/shm', class CropDataset (line 514) | class CropDataset(Dataset): method __init__ (line 517) | def __init__(self, dataset, list_of_imgs_and_crops): method get_image (line 524) | def get_image(self, img_idx): method get_filename (line 531) | def get_filename(self, img_idx): method get_key (line 534) | def get_key(self, img_idx): method crop_image (line 537) | def crop_image(self, img, polygons): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/deep-image-retrieval/dirtorch/datasets/downloader.py function download_dataset (line 6) | def download_dataset(dataset): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/deep-image-retrieval/dirtorch/datasets/generic.py class ImageList (line 13) | class ImageList(Dataset): method __init__ (line 18) | def __init__(self, img_list_path, root='', imgs=None): method get_key (line 29) | def get_key(self, i): class LabelledDataset (line 33) | class LabelledDataset (Dataset): method find_classes (line 37) | def find_classes(self, *arg, **cls_idx): class ImageListLabels (line 44) | class ImageListLabels(LabelledDataset): method __init__ (line 49) | def __init__(self, img_list_path, root=None): method get_key (line 67) | def get_key(self, i): method get_label (line 70) | def get_label(self, i, toint=False): method get_query_db (line 76) | def get_query_db(self): class ImageListLabelsQ (line 80) | class ImageListLabelsQ(ImageListLabels): method __init__ (line 85) | def __init__(self, img_list_path, query_list_path, root=None): method find_classes (line 98) | def find_classes(self, *arg, **cls_idx): method get_query_db (line 104) | def get_query_db(self): class ImagesAndLabels (line 108) | class ImagesAndLabels(ImageListLabels): method __init__ (line 113) | def __init__(self, imgs, labels, cls_idx, root=None): class ImageListRelevants (line 124) | class ImageListRelevants(Dataset): method __init__ (line 130) | def __init__(self, gt_file, root=None, img_dir='jpg', ext='.jpg'): method get_relevants (line 150) | def get_relevants(self, qimg_idx, mode='classic'): method get_junk (line 161) | def get_junk(self, qimg_idx, mode='classic'): method get_query_filename (line 172) | def get_query_filename(self, qimg_idx, root=None): method get_query_roi (line 175) | def get_query_roi(self, qimg_idx): method get_key (line 178) | def get_key(self, i): method get_query_key (line 181) | def get_query_key(self, i): method get_query_db (line 184) | def get_query_db(self): method get_query_groundtruth (line 187) | def get_query_groundtruth(self, query_idx, what='AP', mode='classic'): method eval_query_AP (line 196) | def eval_query_AP(self, query_idx, scores): class ImageListROIs (line 227) | class ImageListROIs(Dataset): method __init__ (line 228) | def __init__(self, root, img_dir, imgs, rois): method get_key (line 238) | def get_key(self, i): method get_roi (line 241) | def get_roi(self, i): method get_image (line 244) | def get_image(self, img_idx, resize=None): function not_none (line 253) | def not_none(label): class ImageClusters (line 257) | class ImageClusters(LabelledDataset): method __init__ (line 262) | def __init__(self, json_path, root=None, filter=not_none): method get_key (line 285) | def get_key(self, i): method get_label (line 288) | def get_label(self, i, toint=False): class NullCluster (line 295) | class NullCluster(ImageClusters): method __init__ (line 298) | def __init__(self, json_path, root=None): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/deep-image-retrieval/dirtorch/datasets/generic_func.py function find_and_list_classes (line 8) | def find_and_list_classes(labels, cls_idx=None ): function find_relevants (line 46) | def find_relevants(labels): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/deep-image-retrieval/dirtorch/datasets/landmarks.py class Landmarks_clean (line 6) | class Landmarks_clean(ImageListLabels): method __init__ (line 7) | def __init__(self): class Landmarks_clean_val (line 11) | class Landmarks_clean_val(ImageListLabels): method __init__ (line 12) | def __init__(self): class Landmarks_lite (line 16) | class Landmarks_lite(ImageListLabels): method __init__ (line 17) | def __init__(self): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/deep-image-retrieval/dirtorch/datasets/landmarks18.py class Landmarks18_train (line 6) | class Landmarks18_train(ImageListLabels): method __init__ (line 7) | def __init__(self): class Landmarks18 (line 11) | class Landmarks18(ImageListLabels): method __init__ (line 12) | def __init__(self): class Landmarks18_lite (line 16) | class Landmarks18_lite(ImageListLabels): method __init__ (line 17) | def __init__(self): class Landmarks18_mid (line 21) | class Landmarks18_mid(ImageListLabels): method __init__ (line 22) | def __init__(self): class Landmarks18_5K (line 26) | class Landmarks18_5K(ImageListLabels): method __init__ (line 27) | def __init__(self): class Landmarks18_val (line 31) | class Landmarks18_val(ImageListLabels): method __init__ (line 32) | def __init__(self): class Landmarks18_valdstr (line 36) | class Landmarks18_valdstr(ImageListLabels): method __init__ (line 37) | def __init__(self): class Landmarks18_index (line 41) | class Landmarks18_index(ImageList): method __init__ (line 42) | def __init__(self): class Landmarks18_new_index (line 46) | class Landmarks18_new_index(ImageList): method __init__ (line 47) | def __init__(self): class Landmarks18_test (line 51) | class Landmarks18_test(ImageList): method __init__ (line 52) | def __init__(self): class Landmarks18_pca (line 56) | class Landmarks18_pca(ImageList): method __init__ (line 57) | def __init__(self): class Landmarks18_missing_index (line 61) | class Landmarks18_missing_index(ImageList): method __init__ (line 62) | def __init__(self): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/deep-image-retrieval/dirtorch/datasets/oxford.py class Oxford5K (line 6) | class Oxford5K(ImageListRelevants): method __init__ (line 7) | def __init__(self): class ROxford5K (line 11) | class ROxford5K(ImageListRelevants): method __init__ (line 12) | def __init__(self): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/deep-image-retrieval/dirtorch/datasets/paris.py class Paris6K (line 6) | class Paris6K(ImageListRelevants): method __init__ (line 7) | def __init__(self): class RParis6K (line 11) | class RParis6K(ImageListRelevants): method __init__ (line 12) | def __init__(self): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/deep-image-retrieval/dirtorch/extract_features.py function extract_features (line 26) | def extract_features(db, net, trfs, pooling='mean', gemp=3, detailed=Fal... function load_model (line 71) | def load_model(path, iscuda): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/deep-image-retrieval/dirtorch/extract_kapture.py function extract_kapture_global_features (line 20) | def extract_kapture_global_features(kapture_root_path: str, net, trfs, p... FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/deep-image-retrieval/dirtorch/loss.py class APLoss (line 7) | class APLoss (nn.Module): method __init__ (line 20) | def __init__(self, nq=25, min=0, max=1): method forward (line 43) | def forward(self, x, label, qw=None, ret='1-mAP'): method measures (line 67) | def measures(self, x, gt, loss=None): class TAPLoss (line 73) | class TAPLoss (APLoss): method __init__ (line 86) | def __init__(self, nq=25, min=0, max=1, simplified=False): method forward (line 90) | def forward(self, x, label, qw=None, ret='1-mAP'): method measures (line 133) | def measures(self, x, gt, loss=None): class TripletMarginLoss (line 139) | class TripletMarginLoss(nn.TripletMarginLoss): method eval_func (line 144) | def eval_func(self, dp, dn): class TripletLogExpLoss (line 148) | class TripletLogExpLoss(nn.Module): method __init__ (line 183) | def __init__(self, p=2, eps=1e-6, swap=False): method forward (line 189) | def forward(self, anchor, positive, negative): method eval_func (line 206) | def eval_func(self, dp, dn): function sim_to_dist (line 210) | def sim_to_dist(scores): class APLoss_dist (line 214) | class APLoss_dist (APLoss): method forward (line 215) | def forward(self, x, label, **kw): class TAPLoss_dist (line 220) | class TAPLoss_dist (TAPLoss): method forward (line 221) | def forward(self, x, label, **kw): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/deep-image-retrieval/dirtorch/nets/__init__.py function list_archs (line 18) | def list_archs(): function create_model (line 26) | def create_model(arch, pretrained='', delete_fc=False, *args, **kwargs): function load_pretrained_weights (line 69) | def load_pretrained_weights(net, state_dict, delete_fc=False): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/deep-image-retrieval/dirtorch/nets/backbones/__init__.py function load_pretrained_weights (line 4) | def load_pretrained_weights(net, state_dict): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/deep-image-retrieval/dirtorch/nets/backbones/resnet.py function conv3x3 (line 9) | def conv3x3(in_planes, out_planes, stride=1): class BasicBlock (line 15) | class BasicBlock(nn.Module): method __init__ (line 18) | def __init__(self, inplanes, planes, stride=1, downsample=None): method forward (line 28) | def forward(self, x): class Bottleneck (line 46) | class Bottleneck(nn.Module): method __init__ (line 54) | def __init__(self, inplanes, planes, stride=1, dilation=1, downsample=... method forward (line 67) | def forward(self, x): function reset_weights (line 92) | def reset_weights(net): class ResNet (line 102) | class ResNet(nn.Module): method __init__ (line 105) | def __init__(self, block, layers, fc_out, model_name, self_similarity_... method _make_layer (line 134) | def _make_layer(self, block, planes, blocks, stride=1, self_similarity... method forward (line 157) | def forward(self, x, out_layer=0): method load_pretrained_weights (line 176) | def load_pretrained_weights(self, pretrain_code): function resnet18 (line 205) | def resnet18(out_dim=2048): function resnet50 (line 211) | def resnet50(out_dim=2048): function resnet101 (line 217) | def resnet101(out_dim=2048): function resnet152 (line 223) | def resnet152(out_dim=2048): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/deep-image-retrieval/dirtorch/nets/backbones/resnext101_features.py class LambdaBase (line 7) | class LambdaBase(nn.Sequential): method __init__ (line 8) | def __init__(self, fn, *args): method forward_prepare (line 12) | def forward_prepare(self, input): class Lambda (line 18) | class Lambda(LambdaBase): method forward (line 19) | def forward(self, input): class LambdaMap (line 22) | class LambdaMap(LambdaBase): method forward (line 23) | def forward(self, input): class LambdaReduce (line 26) | class LambdaReduce(LambdaBase): method forward (line 27) | def forward(self, input): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/deep-image-retrieval/dirtorch/nets/layers/pooling.py class GeneralizedMeanPooling (line 12) | class GeneralizedMeanPooling(Module): method __init__ (line 31) | def __init__(self, norm, output_size=1, eps=1e-6): method forward (line 38) | def forward(self, x): method __repr__ (line 42) | def __repr__(self): class GeneralizedMeanPoolingP (line 49) | class GeneralizedMeanPoolingP(GeneralizedMeanPooling): method __init__ (line 52) | def __init__(self, norm=3, output_size=1, eps=1e-6): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/deep-image-retrieval/dirtorch/nets/rmac_resnet.py function l2_normalize (line 7) | def l2_normalize(x, axis=-1): class ResNet_RMAC (line 12) | class ResNet_RMAC(ResNet): method __init__ (line 15) | def __init__(self, block, layers, model_name, out_dim=2048, norm_featu... method forward (line 39) | def forward(self, x): function resnet18_rmac (line 74) | def resnet18_rmac(backbone=ResNet_RMAC, **kwargs): function resnet50_rmac (line 78) | def resnet50_rmac(backbone=ResNet_RMAC, **kwargs): function resnet101_rmac (line 82) | def resnet101_rmac(backbone=ResNet_RMAC, **kwargs): function resnet152_rmac (line 86) | def resnet152_rmac(backbone=ResNet_RMAC, **kwargs): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/deep-image-retrieval/dirtorch/nets/rmac_resnet_fpn.py function l2_normalize (line 6) | def l2_normalize(x, axis=-1): class ResNet_RMAC_FPN (line 11) | class ResNet_RMAC_FPN(ResNet): method __init__ (line 14) | def __init__(self, block, layers, model_name, out_dim=None, norm_featu... method forward (line 52) | def forward(self, x): function resnet18_fpn_rmac (line 95) | def resnet18_fpn_rmac(backbone=ResNet_RMAC_FPN, **kwargs): function resnet50_fpn_rmac (line 99) | def resnet50_fpn_rmac(backbone=ResNet_RMAC_FPN, **kwargs): function resnet101_fpn_rmac (line 103) | def resnet101_fpn_rmac(backbone=ResNet_RMAC_FPN, **kwargs): function resnet101_fpn0_rmac (line 107) | def resnet101_fpn0_rmac(backbone=ResNet_RMAC_FPN, **kwargs): function resnet152_fpn_rmac (line 111) | def resnet152_fpn_rmac(backbone=ResNet_RMAC_FPN, **kwargs): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/deep-image-retrieval/dirtorch/nets/rmac_resnext.py function l2_normalize (line 5) | def l2_normalize(x, axis=-1): class ResNext_RMAC (line 10) | class ResNext_RMAC(nn.Module): method __init__ (line 13) | def __init__(self, backbone, out_dim=2048, norm_features=False, method forward (line 36) | def forward(self, x): function resnet18_rmac (line 69) | def resnet18_rmac(backbone=ResNet_RMAC, **kwargs): function resnet50_rmac (line 73) | def resnet50_rmac(backbone=ResNet_RMAC, **kwargs): function resnet101_rmac (line 77) | def resnet101_rmac(backbone=ResNet_RMAC, **kwargs): function resnet152_rmac (line 81) | def resnet152_rmac(backbone=ResNet_RMAC, **kwargs): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/deep-image-retrieval/dirtorch/test_dir.py function expand_descriptors (line 24) | def expand_descriptors(descs, db=None, alpha=0, k=0): function extract_image_features (line 47) | def extract_image_features(dataset, transforms, net, ret_imgs=False, sam... function eval_model (line 97) | def eval_model(db, net, trfs, pooling='mean', gemp=3, detailed=False, wh... function load_model (line 183) | def load_model(path, iscuda): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/deep-image-retrieval/dirtorch/utils/common.py function typename (line 19) | def typename(x): function tonumpy (line 23) | def tonumpy(x): function matmul (line 30) | def matmul(A, B): function pool (line 41) | def pool(x, pooling='mean', gemp=3): function torch_set_gpu (line 58) | def torch_set_gpu(gpus, seed=None, randomize=True): function torch_set_seed (line 84) | def torch_set_seed(seed, cuda, randomize=True): function save_checkpoint (line 102) | def save_checkpoint(state, is_best, filename): function load_checkpoint (line 117) | def load_checkpoint(filename, iscuda=False): function switch_model_to_cuda (line 150) | def switch_model_to_cuda(model, iscuda=True, checkpoint=None): function model_size (line 178) | def model_size(model): function freeze_batch_norm (line 187) | def freeze_batch_norm(model, freeze=True, only_running=False): function variables (line 205) | def variables(inputs, iscuda, not_on_gpu=[]): function transform (line 221) | def transform(pca, X, whitenp=0.5, whitenv=None, whitenm=1.0, use_sklear... function whiten_features (line 235) | def whiten_features(X, pca, l2norm=True, whitenp=0.5, whitenv=None, whit... FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/deep-image-retrieval/dirtorch/utils/convenient.py function mkdir (line 6) | def mkdir(d): function mkdir (line 11) | def mkdir( fname, isfile='auto' ): function touch (line 26) | def touch(filename): function assert_outpath (line 35) | def assert_outpath( path, ext='', mkdir=False ): class _BasePool (line 51) | class _BasePool (object): method __init__ (line 52) | def __init__(self, nt=0): method starmap (line 54) | def starmap(self, func, args): class ProcessPool (line 57) | class ProcessPool (_BasePool): method __init__ (line 58) | def __init__(self, nt=0): class ThreadPool (line 62) | class ThreadPool (_BasePool): method __init__ (line 63) | def __init__(self, nt=0): function is_iterable (line 71) | def is_iterable(val, exclude={str}): function listify (line 83) | def listify( val, exclude={str} ): function unlistify (line 88) | def unlistify( lis ): function sig_folder_ext (line 98) | def sig_folder_ext(f): function sig_folder (line 100) | def sig_folder(f): function sig_ext (line 102) | def sig_ext(f): function sig_3folder_ext (line 104) | def sig_3folder_ext(f): function sig_all (line 108) | def sig_all(f): function saferm (line 111) | def saferm(f, sig=sig_folder_ext ): function tic (line 139) | def tic(tag='tic'): function toc (line 144) | def toc(tag='', cum=False): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/deep-image-retrieval/dirtorch/utils/evaluation.py function accuracy_topk (line 8) | def accuracy_topk(output, target, topk=(1,)): function compute_AP (line 41) | def compute_AP(label, score): function compute_average_precision (line 46) | def compute_average_precision(positive_ranks): function compute_average_precision_quantized (line 85) | def compute_average_precision_quantized(labels, idx, step=0.01): function pixelwise_iou (line 101) | def pixelwise_iou(output, target): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/deep-image-retrieval/dirtorch/utils/funcs.py function sigmoid (line 7) | def sigmoid(x, a=1, b=0): function sigmoid_range (line 11) | def sigmoid_range(x, at5, at95): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/deep-image-retrieval/dirtorch/utils/pytorch_loader.py function get_loader (line 11) | def get_loader( dataset, trf_chain, iscuda, class PytorchLoader (line 78) | class PytorchLoader (data.Dataset): method __init__ (line 96) | def __init__(self, dataset, transform=None, method __getitem__ (line 106) | def __getitem__(self, index): method __len__ (line 164) | def __len__(self): method __repr__ (line 167) | def __repr__(self): class BalancedSampler (line 184) | class BalancedSampler (data.sampler.Sampler): method __init__ (line 202) | def __init__(self, dataset, size=1.0, balanced=1.0, use_all=False): method __iter__ (line 220) | def __iter__(self): method __len__ (line 248) | def __len__(self): function load_one_img (line 256) | def load_one_img( loader ): function tensor2img (line 270) | def tensor2img(tensor, model): function test_loader_speed (line 287) | def test_loader_speed(loader_): function try_to_get (line 298) | def try_to_get(func, *args, **kwargs): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/deep-image-retrieval/dirtorch/utils/transforms.py function create (line 11) | def create(cmd_line, to_tensor=False, **vars): class Identity (line 40) | class Identity (object): method __call__ (line 43) | def __call__(self, inp): class Pad (line 47) | class Pad(object): method __init__ (line 53) | def __init__(self, size, color=(127,127,127)): method __call__ (line 60) | def __call__(self, inp): class PadSquare (line 77) | class PadSquare (object): method __init__ (line 87) | def __init__(self, size=None, color=(127,127,127)): method __call__ (line 94) | def __call__(self, inp): class RandomBorder (line 108) | class RandomBorder (object): method __init__ (line 112) | def __init__(self, min_size, max_size, color=(127,127,127)): method __call__ (line 122) | def __call__(self, inp): class Scale (line 133) | class Scale (object): method __init__ (line 144) | def __init__(self, size, interpolation=Image.BILINEAR, largest=False, ... method get_params (line 154) | def get_params(self, imsize): method __call__ (line 174) | def __call__(self, inp): class RandomScale (line 189) | class RandomScale (Scale): method __init__ (line 203) | def __init__(self, min_size, max_size, ar=1, can_upscale=False, can_do... method get_params (line 214) | def get_params(self, imsize): class RandomCrop (line 249) | class RandomCrop (object): method __init__ (line 262) | def __init__(self, size, padding=0): method get_params (line 270) | def get_params(img, output_size): method __call__ (line 279) | def __call__(self, inp): class CenterCrop (line 301) | class CenterCrop (RandomCrop): method get_params (line 310) | def get_params(img, output_size): class CropToBbox (line 319) | class CropToBbox(object): method __init__ (line 328) | def __init__(self, margin=0.5, min_size=0): method __call__ (line 332) | def __call__(self, inp): class RandomRotation (line 361) | class RandomRotation(object): method __init__ (line 372) | def __init__(self, degrees, interpolation=Image.BILINEAR): method __call__ (line 376) | def __call__(self, inp): class RandomFlip (line 391) | class RandomFlip (object): method __call__ (line 394) | def __call__(self, inp): class RandomTilting (line 406) | class RandomTilting(object): method __init__ (line 417) | def __init__(self, magnitude, directions='all'): method __call__ (line 421) | def __call__(self, inp): class StillTransform (line 491) | class StillTransform (object): method _transform (line 494) | def _transform(self, img): method __call__ (line 497) | def __call__(self, inp): class ColorJitter (line 507) | class ColorJitter (StillTransform): method __init__ (line 519) | def __init__(self, brightness=0, contrast=0, saturation=0, hue=0): method get_params (line 526) | def get_params(brightness, contrast, saturation, hue): method _transform (line 555) | def _transform(self, img): class RandomErasing (line 560) | class RandomErasing (StillTransform): method __init__ (line 576) | def __init__(self, area): method _transform (line 579) | def _transform(self, image): class ToTensor (line 617) | class ToTensor (StillTransform, tvf.ToTensor): method _transform (line 618) | def _transform(self, img): class Normalize (line 621) | class Normalize (StillTransform, tvf.Normalize): method _transform (line 622) | def _transform(self, img): class BBoxToPixelLabel (line 626) | class BBoxToPixelLabel (object): method __init__ (line 629) | def __init__(self, nclass, downsize, mode): method __call__ (line 637) | def __call__(self, inp): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/deep-image-retrieval/dirtorch/utils/transforms_tools.py function is_pil_image (line 6) | def is_pil_image(img): class DummyImg (line 9) | class DummyImg: method __init__ (line 12) | def __init__(self, size): method resize (line 15) | def resize(self, size, *args, **kwargs): method expand (line 18) | def expand(self, border): method crop (line 27) | def crop(self, border): method rotate (line 35) | def rotate(self, angle): method transform (line 38) | def transform(self, size, *args, **kwargs): function grab_img (line 42) | def grab_img( img_and_label ): function update_img_and_labels (line 59) | def update_img_and_labels(img_and_label, img, aff=None, persp=None): function rand_log_uniform (line 100) | def rand_log_uniform(a, b): function int_tuple (line 104) | def int_tuple(*args): function aff_translate (line 107) | def aff_translate(tx, ty): function aff_rotate (line 111) | def aff_rotate(angle): function aff_mul (line 115) | def aff_mul(aff, aff2): function persp_mul (line 134) | def persp_mul(mat, mat2): function adjust_brightness (line 157) | def adjust_brightness(img, brightness_factor): function adjust_contrast (line 175) | def adjust_contrast(img, contrast_factor): function adjust_saturation (line 193) | def adjust_saturation(img, saturation_factor): function adjust_hue (line 211) | def adjust_hue(img, hue_factor): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/disk/colmap/colmap/database.py function image_ids_to_pair_id (line 113) | def image_ids_to_pair_id(image_id1, image_id2): function pair_id_to_image_ids (line 119) | def pair_id_to_image_ids(pair_id): function array_to_blob (line 125) | def array_to_blob(array): function blob_to_array (line 132) | def blob_to_array(blob, dtype, shape=(-1,)): class COLMAPDatabase (line 139) | class COLMAPDatabase(sqlite3.Connection): method connect (line 142) | def connect(database_path): method __init__ (line 146) | def __init__(self, *args, **kwargs): method add_camera (line 164) | def add_camera(self, model, width, height, params, method add_image (line 173) | def add_image(self, name, camera_id, method add_keypoints (line 181) | def add_keypoints(self, image_id, keypoints): method add_descriptors (line 190) | def add_descriptors(self, image_id, descriptors): method add_matches (line 196) | def add_matches(self, image_id1, image_id2, matches): method add_two_view_geometry (line 209) | def add_two_view_geometry(self, image_id1, image_id2, matches, function example_usage (line 228) | def example_usage(): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/disk/colmap/colmap/read_dense.py function read_array (line 39) | def read_array(path): function parse_args (line 57) | def parse_args(): function main (line 73) | def main(): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/disk/colmap/colmap/read_model.py class Image (line 48) | class Image(BaseImage): method qvec2rotmat (line 49) | def qvec2rotmat(self): function read_next_bytes (line 70) | def read_next_bytes(fid, num_bytes, format_char_sequence, endian_charact... function read_cameras_text (line 81) | def read_cameras_text(path): function read_cameras_binary (line 107) | def read_cameras_binary(path_to_model_file): function read_images_text (line 136) | def read_images_text(path): function read_images_binary (line 167) | def read_images_binary(path_to_model_file): function read_points3D_text (line 203) | def read_points3D_text(path): function read_points3d_binary (line 230) | def read_points3d_binary(path_to_model_file): function read_model (line 260) | def read_model(path, ext): function qvec2rotmat (line 272) | def qvec2rotmat(qvec): function rotmat2qvec (line 285) | def rotmat2qvec(R): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/disk/colmap/colmap2dataset.py function convert_depth (line 8) | def convert_depth(name, src_path, dst_path): function camera_to_K (line 22) | def camera_to_K(camera): function create_calibration (line 37) | def create_calibration(image, camera, prefix): function covisible_pairs (line 48) | def covisible_pairs(images, low=0.5, high=0.8): function encode_pairs (line 78) | def encode_pairs(pairs): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/disk/colmap/h5_to_db.py function get_focal (line 8) | def get_focal(image_path, err_on_default=False): function create_camera (line 36) | def create_camera(db, image_path): function add_keypoints (line 58) | def add_keypoints(db, h5_path, image_path): function add_matches (line 80) | def add_matches(db, h5_path, fname_to_id): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/disk/compute_validation_auc.py function read_deltas (line 8) | def read_deltas(path): function calculate_auc (line 26) | def calculate_auc(Δ_θ, Δ_T, length): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/disk/detect.py class Image (line 11) | class Image: method __init__ (line 12) | def __init__(self, bitmap: ['C', 'H', 'W'], fname: str, orig_shape=None): method resize_to (line 20) | def resize_to(self, shape): method to_image_coord (line 28) | def to_image_coord(self, xys: [2, 'N']) -> ([2, 'N'], ['N']): method _compute_interpolation_size (line 39) | def _compute_interpolation_size(self, shape): method _interpolate (line 53) | def _interpolate(self, image: ['C', 'H', 'W'], shape) -> ['C', 'h', 'w']: method _pad (line 63) | def _pad(self, image: ['C', 'H', 'W'], shape) -> ['C', 'h', 'w']: class SceneDataset (line 73) | class SceneDataset: method __init__ (line 74) | def __init__(self, image_path, crop_size=(None, None)): method __len__ (line 80) | def __len__(self): method __getitem__ (line 83) | def __getitem__(self, ix): method collate_fn (line 103) | def collate_fn(images): function extract (line 108) | def extract(dataset, save_path): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/disk/disk/common/errors.py class DataError (line 1) | class DataError(Exception): class EstimationFailedError (line 4) | class EstimationFailedError(Exception): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/disk/disk/common/image.py function _rescale (line 8) | def _rescale(tensor: ['C', 'H', 'W'], size) -> ['C', 'h', 'w']: function _pad (line 17) | def _pad(tensor: ['C', 'H', 'W'], size, value=0.): class Image (line 32) | class Image: method __init__ (line 34) | def __init__( method K_inv (line 54) | def K_inv(self): method hwc (line 58) | def hwc(self): method shape (line 62) | def shape(self): method scale (line 65) | def scale(self, size): method pad (line 95) | def pad(self, size): method to (line 104) | def to(self, *args, **kwargs): method unproject (line 119) | def unproject(self, xy: [2, 'N']) -> [3, 'N']: method project (line 133) | def project(self, xyw: [3, 'N']) -> [2, 'N']: method in_range_mask (line 139) | def in_range_mask(self, xy: [2, 'N']) -> ['N']: method fetch_depth (line 146) | def fetch_depth(self, xy: [2, 'N']) -> ['N']: FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/disk/disk/common/logger.py class Logger (line 3) | class Logger: method __init__ (line 4) | def __init__(self, path): method add_scalar (line 8) | def add_scalar(self, tag, value): method add_scalars (line 17) | def add_scalars(self, tag_to_value, prefix=''): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/disk/disk/common/structs.py class NpArray (line 8) | class NpArray: method __class_getitem__ (line 9) | def __class_getitem__(self, arg): class _NpArray (line 13) | class _NpArray: method __getitem__ (line 14) | def __getitem__(self, _idx): class Features (line 19) | class Features: method __init__ (line 21) | def __init__(self, kp: ['N', 2], desc: ['N', 'F'], kp_logp: ['N']): method n (line 30) | def n(self): method device (line 34) | def device(self): method detached_and_grad_ (line 37) | def detached_and_grad_(self): method requires_grad_ (line 44) | def requires_grad_(self, is_on): method grad_tensors (line 48) | def grad_tensors(self): method to (line 51) | def to(self, *args, **kwargs): class MatchDistribution (line 58) | class MatchDistribution(abc.ABC): method sample (line 60) | def sample(self) -> [2, 'K']: method mle (line 64) | def mle(self) -> [2, 'K']: method dense_logp (line 68) | def dense_logp(self): method dense_p (line 72) | def dense_p(self): method features_1 (line 76) | def features_1(self) -> Features: method features_2 (line 80) | def features_2(self) -> Features: method shape (line 84) | def shape(self): method matched_pairs (line 87) | def matched_pairs(self, mle=False): class MatchedPairs (line 97) | class MatchedPairs: method __init__ (line 99) | def __init__(self, kps1: ['N', 2], kps2: ['M', 2], matches: [2, 'K']): method to (line 104) | def to(self, *args, **kwargs): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/disk/disk/common/vis.py class MultiFigure (line 11) | class MultiFigure: method __init__ (line 13) | def __init__( method mark_xy (line 57) | def mark_xy( FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/disk/disk/geom/distance_matrix.py function distance_matrix (line 8) | def distance_matrix(fs1: ['N', 'F'], fs2: ['M', 'F']) -> ['N', 'M']: FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/disk/disk/geom/epi.py function cross_product_matrix (line 6) | def cross_product_matrix(v: [3]) -> [3, 3]: function xy_to_xyw (line 18) | def xy_to_xyw(xy: [2, 'N']) -> [3, 'N']: function ims2E (line 23) | def ims2E(im1, im2) -> [3, 3]: function ims2F (line 29) | def ims2F(im1, im2) -> [3, 3]: function symdimm (line 34) | def symdimm(x1: [2, 'N'], x2: [2, 'M'], im1, im2) -> ['N', 'M']: function asymmdist (line 53) | def asymmdist(x1: [2, 'N'], x2: [2, 'M'], F: [3, 3]) -> ['N', 'M']: function asymmdist_from_imgs (line 68) | def asymmdist_from_imgs(x1: [2, 'N'], x2: [2, 'M'], im1, im2) -> ['N', '... function p_asymmdist (line 73) | def p_asymmdist(x1: [2, 'N'], x2: [2, 'N'], F: [3, 3]) -> ['N']: function p_asymmdist_from_imgs (line 89) | def p_asymmdist_from_imgs(x1: [2, 'N'], x2: [2, 'N'], im1, im2) -> ['N']: FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/disk/disk/geom/pose.py class PoseError (line 9) | class PoseError(typing.NamedTuple): method __repr__ (line 13) | def __repr__(self): method to_dict (line 16) | def to_dict(self): class Pose (line 22) | class Pose(typing.NamedTuple): method __str__ (line 26) | def __str__(self): method to (line 29) | def to(self, *args, **kwargs): method R_4x4 (line 36) | def R_4x4(self): method T_4x4 (line 43) | def T_4x4(self): method TR_4x4 (line 49) | def TR_4x4(self): method from_poselike (line 53) | def from_poselike(cls, poselike): method relative (line 57) | def relative(p1, p2, normed=False): method error (line 70) | def error(p1, p2): function _normalized_cosine_error (line 86) | def _normalized_cosine_error(v1: ['N'], v2: ['N']): function matrix_to_quaternion (line 102) | def matrix_to_quaternion(M: [3, 3]) -> [4]: FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/disk/disk/loss/discrete_metric.py function classify_pairs (line 9) | def classify_pairs(kps1: ['N', 2], kps2: ['N', 2], img1: Image, img2: Im... class DiscreteMetric (line 20) | class DiscreteMetric(torch.nn.Module): method __init__ (line 21) | def __init__(self, th=2., lm_kp=0., lm_tp=1., lm_fp=-0.25): method forward (line 29) | def forward( method _loss_one_pair (line 61) | def _loss_one_pair(self, pairs: MatchedPairs, img1: Image, img2: Image): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/disk/disk/loss/pose_metric.py class PoseQualityResult (line 13) | class PoseQualityResult: method __init__ (line 14) | def __init__(self, error: PoseError, n_inliers: int, success: bool = T... method to_dict (line 19) | def to_dict(self): method __str__ (line 26) | def __str__(self): class Job (line 41) | class Job(typing.NamedTuple): method __call__ (line 49) | def __call__(self): method execute (line 74) | def execute(job): class DummyPool (line 77) | class DummyPool: method map (line 81) | def map(self, f, args): class PoseQuality (line 84) | class PoseQuality: method __init__ (line 85) | def __init__(self, ransac=Ransac(), dummy_pool=False, n_proc=6): method __call__ (line 92) | def __call__( FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/disk/disk/loss/ransac.py function _recover_pose (line 9) | def _recover_pose(E: [3, 3], i_coords: ['N', 2], j_coords: ['N', 2]): function _normalize_coords (line 29) | def _normalize_coords(coords: ['N', 2], K: [3, 3]) -> ['N', 2]: class Ransac (line 37) | class Ransac(typing.NamedTuple): method __call__ (line 44) | def __call__( FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/disk/disk/loss/reinforce.py class Reinforce (line 6) | class Reinforce: method __init__ (line 7) | def __init__(self, reward, lm_kp): method _loss_for_pair (line 11) | def _loss_for_pair(self, match_dist: MatchDistribution, img1: Image, i... method accumulate_grad (line 56) | def accumulate_grad( FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/disk/disk/loss/rewards.py class EpipolarReward (line 7) | class EpipolarReward: method __init__ (line 8) | def __init__(self, th=2., lm_tp=1., lm_fp=-0.25): method __call__ (line 14) | def __call__( method classify (line 29) | def classify( class DepthReward (line 48) | class DepthReward: method __init__ (line 49) | def __init__(self, th=2., lm_tp=1., lm_fp=-0.25): method __call__ (line 57) | def __call__( FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/disk/disk/model/consistent_matcher.py class ConsistentMatchDistribution (line 9) | class ConsistentMatchDistribution(MatchDistribution): method __init__ (line 10) | def __init__( method dense_p (line 33) | def dense_p(self) -> ['N', 'M']: method dense_logp (line 40) | def dense_logp(self) -> ['N', 'M']: method _select_cycle_consistent (line 47) | def _select_cycle_consistent(self, left: ['N'], right: ['M']) -> [2, '... method sample (line 59) | def sample(self) -> [2, 'K']: method mle (line 66) | def mle(self) -> [2, 'K']: method features_1 (line 76) | def features_1(self) -> Features: method features_2 (line 79) | def features_2(self) -> Features: class ConsistentMatcher (line 82) | class ConsistentMatcher(torch.nn.Module): method __init__ (line 83) | def __init__(self, inverse_T=1.): method extra_repr (line 87) | def extra_repr(self): method match_pair (line 90) | def match_pair(self, features_1: Features, features_2: Features): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/disk/disk/model/cycle_matcher.py class CycleMatcher (line 8) | class CycleMatcher: method match_features (line 10) | def match_features(self, feat_1: ['N', 'F'], feat_2: ['M', 'F']) -> [2... method match_pairwise (line 42) | def match_pairwise( FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/disk/disk/model/detector.py function select_on_last (line 12) | def select_on_last(values: [..., 'T'], indices: [...]) -> [...]: function point_distribution (line 28) | def point_distribution( class Keypoints (line 55) | class Keypoints: method __init__ (line 63) | def __init__(self, xys: ['N', 2], logp: ['N']): method merge_with_descriptors (line 68) | def merge_with_descriptors(self, descriptors: ['C', 'H', 'W']) -> Feat... class Detector (line 80) | class Detector: method __init__ (line 81) | def __init__(self, window=8): method _tile (line 85) | def _tile(self, heatmap: ['B', 'C', 'H', 'W']) -> ['B', 'C', 'h', 'w',... method sample (line 102) | def sample(self, heatmap: ['B', 1, 'H', 'W']) -> NpArray[Keypoints]: method nms (line 145) | def nms( FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/disk/disk/model/disk.py class DISK (line 12) | class DISK(torch.nn.Module): method __init__ (line 13) | def __init__( method _split (line 32) | def _split(self, unet_output: ['B', 'C', 'H', 'W']) \ method features (line 45) | def features( FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/disk/disk/model/nms.py function nms (line 6) | def nms(signal: ['B', 'H', 'W'], window_size=5, cutoff=0.) -> ['B', 'H',... FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/disk/match.py class H5Store (line 57) | class H5Store: method __init__ (line 58) | def __init__(self, path, dtype=torch.float32): method keys (line 63) | def keys(self): method __getitem__ (line 66) | def __getitem__(self, ix): method get_kp (line 75) | def get_kp(self, ix): function pair_key (line 78) | def pair_key(key_1, key_2): function _binary_to_index (line 87) | def _binary_to_index(binary_mask: ['N'], ix2: ['M']) -> [2, 'M']: function _ratio_one_way (line 94) | def _ratio_one_way(dist_m: ['N', 'M'], rt) -> [2, 'K']: function _match_chunkwise (line 103) | def _match_chunkwise(ds1: ['N', 'F'], ds2: ['M', 'F'], rt) -> [2, 'K']: function _match (line 119) | def _match(ds1: ['N', 'F'], ds2: ['M', 'F'], rt) -> [2, 'K']: function match (line 131) | def match(desc_1, desc_2, rt=1., u16=False): function brute_match (line 140) | def brute_match(descriptors, hdf): class MatcherWrapper (line 172) | class MatcherWrapper: class InnerWrapper (line 173) | class InnerWrapper: method __init__ (line 174) | def __init__(self): method raw_mle_match_pair (line 181) | def raw_mle_match_pair(self, ds1: ['N', 'F'], ds2: ['M', 'F']) -> [2... method __init__ (line 185) | def __init__(self): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/disk/view_h5.py function show_or_save (line 32) | def show_or_save(): function view_keypoints (line 45) | def view_keypoints(h5_path, image_path): function view_matches (line 66) | def view_matches(h5_path, image_path): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/r2d2/datasets/aachen.py class AachenImages (line 13) | class AachenImages (Dataset): method __init__ (line 16) | def __init__(self, select='db day night', root='data/aachen'): method get_key (line 33) | def get_key(self, idx): class AachenImages_DB (line 38) | class AachenImages_DB (AachenImages): method __init__ (line 41) | def __init__(self, **kw): method get_tag (line 45) | def get_tag(self, idx): class AachenPairs_StyleTransferDayNight (line 51) | class AachenPairs_StyleTransferDayNight (AachenImages_DB, StillPairDatas... method __init__ (line 55) | def __init__(self, root='data/aachen/style_transfer', **kw): class AachenPairs_OpticalFlow (line 77) | class AachenPairs_OpticalFlow (AachenImages_DB, PairDataset): method __init__ (line 80) | def __init__(self, root='data/aachen/optical_flow', **kw): method get_mask_filename (line 95) | def get_mask_filename(self, pair_idx): method get_mask (line 99) | def get_mask(self, pair_idx): method get_flow_filename (line 102) | def get_flow_filename(self, pair_idx): method get_flow (line 106) | def get_flow(self, pair_idx): method get_pair (line 118) | def get_pair(self, idx, output=()): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/r2d2/datasets/dataset.py class Dataset (line 11) | class Dataset(object): method __len__ (line 18) | def __len__(self): method get_key (line 21) | def get_key(self, img_idx): method get_filename (line 24) | def get_filename(self, img_idx, root=None): method get_image (line 27) | def get_image(self, img_idx): method __repr__ (line 35) | def __repr__(self): class CatDataset (line 43) | class CatDataset (Dataset): method __init__ (line 46) | def __init__(self, *datasets): method which (line 56) | def which(self, i): method get_key (line 61) | def get_key(self, i): method get_filename (line 65) | def get_filename(self, i): method __repr__ (line 69) | def __repr__(self): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/r2d2/datasets/imgfolder.py class ImgFolder (line 11) | class ImgFolder (Dataset): method __init__ (line 14) | def __init__(self, root, imgs=None, exts=('.jpg','.png','.ppm')): method get_key (line 20) | def get_key(self, idx): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/r2d2/datasets/pair_dataset.py class PairDataset (line 14) | class PairDataset (Dataset): method __init__ (line 17) | def __init__(self): method get_filename (line 21) | def get_filename(self, img_idx, root=None): method get_image (line 26) | def get_image(self, img_idx): method get_corres_filename (line 31) | def get_corres_filename(self, pair_idx): method get_homography_filename (line 34) | def get_homography_filename(self, pair_idx): method get_flow_filename (line 37) | def get_flow_filename(self, pair_idx): method get_mask_filename (line 40) | def get_mask_filename(self, pair_idx): method get_pair (line 43) | def get_pair(self, idx, output=()): method get_paired_images (line 55) | def get_paired_images(self): method __len__ (line 63) | def __len__(self): method __repr__ (line 66) | def __repr__(self): method _flow2png (line 74) | def _flow2png(flow, path): method _png2flow (line 81) | def _png2flow(path): class StillPairDataset (line 90) | class StillPairDataset (PairDataset): method get_pair (line 95) | def get_pair(self, pair_idx, output=()): class SyntheticPairDataset (line 119) | class SyntheticPairDataset (PairDataset): method __init__ (line 123) | def __init__(self, dataset, scale='', distort=''): method attach_dataset (line 128) | def attach_dataset(self, dataset): method make_pair (line 137) | def make_pair(self, img): method get_pair (line 140) | def get_pair(self, i, output=('aflow')): method __repr__ (line 170) | def __repr__(self): class TransformedPairs (line 180) | class TransformedPairs (PairDataset): method __init__ (line 185) | def __init__(self, dataset, trf=''): method attach_dataset (line 189) | def attach_dataset(self, dataset): method get_pair (line 199) | def get_pair(self, i, output=''): method __repr__ (line 228) | def __repr__(self): class CatPairDataset (line 237) | class CatPairDataset (CatDataset): method __init__ (line 240) | def __init__(self, *datasets): method __len__ (line 248) | def __len__(self): method __repr__ (line 251) | def __repr__(self): method pair_which (line 257) | def pair_which(self, i): method pair_call (line 262) | def pair_call(self, func, i, *args, **kwargs): method get_pair (line 266) | def get_pair(self, i, output=()): method get_flow_filename (line 270) | def get_flow_filename(self, pair_idx, *args, **kwargs): method get_mask_filename (line 273) | def get_mask_filename(self, pair_idx, *args, **kwargs): method get_corres_filename (line 276) | def get_corres_filename(self, pair_idx, *args, **kwargs): function is_pair (line 281) | def is_pair(x): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/r2d2/datasets/web_images.py class RandomWebImages (line 11) | class RandomWebImages (Dataset): method __init__ (line 15) | def __init__(self, start=0, end=1024, root="data/revisitop1m"): method get_key (line 49) | def get_key(self, i): function verify_img (line 54) | def verify_img(folder, f): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/r2d2/extract.py function load_network (line 16) | def load_network(model_fn): class NonMaxSuppression (line 29) | class NonMaxSuppression (torch.nn.Module): method __init__ (line 30) | def __init__(self, rel_thr=0.7, rep_thr=0.7): method forward (line 36) | def forward(self, reliability, repeatability, **kw): function extract_multiscale (line 50) | def extract_multiscale( net, img, detector, scale_f=2**0.25, function extract_keypoints (line 111) | def extract_keypoints(args): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/r2d2/extract_kapture.py function extract_kapture_keypoints (line 25) | def extract_kapture_keypoints(args): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/r2d2/nets/ap_loss.py class APLoss (line 11) | class APLoss (nn.Module): method __init__ (line 20) | def __init__(self, nq=25, min=0, max=1, euc=False): method compute_AP (line 43) | def compute_AP(self, x, label): method forward (line 60) | def forward(self, x, label): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/r2d2/nets/losses.py class MultiLoss (line 16) | class MultiLoss (nn.Module): method __init__ (line 23) | def __init__(self, *args, dbg=()): method forward (line 35) | def forward(self, select=None, **variables): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/r2d2/nets/patchnet.py class BaseNet (line 11) | class BaseNet (nn.Module): method softmax (line 16) | def softmax(self, ux): method normalize (line 23) | def normalize(self, x, ureliability, urepeatability): method forward_one (line 28) | def forward_one(self, x): method forward (line 31) | def forward(self, imgs, **kw): class PatchNet (line 39) | class PatchNet (BaseNet): method __init__ (line 43) | def __init__(self, inchan=3, dilated=True, dilation=1, bn=True, bn_aff... method _make_bn (line 53) | def _make_bn(self, outd): method _add_conv (line 56) | def _add_conv(self, outd, k=3, stride=1, dilation=1, bn=True, relu=True): method forward_one (line 68) | def forward_one(self, x): class L2_Net (line 75) | class L2_Net (PatchNet): method __init__ (line 79) | def __init__(self, dim=128, **kw ): class Quad_L2Net (line 92) | class Quad_L2Net (PatchNet): method __init__ (line 95) | def __init__(self, dim=128, mchan=4, relu22=False, **kw ): class Quad_L2Net_ConfCFS (line 111) | class Quad_L2Net_ConfCFS (Quad_L2Net): method __init__ (line 114) | def __init__(self, **kw ): method forward_one (line 122) | def forward_one(self, x): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/r2d2/nets/reliability_loss.py class PixelAPLoss (line 12) | class PixelAPLoss (nn.Module): method __init__ (line 20) | def __init__(self, sampler, nq=20): method loss_from_ap (line 26) | def loss_from_ap(self, ap, rel): method forward (line 29) | def forward(self, descriptors, aflow, **kw): class ReliabilityLoss (line 45) | class ReliabilityLoss (PixelAPLoss): method __init__ (line 49) | def __init__(self, sampler, base=0.5, **kw): method loss_from_ap (line 55) | def loss_from_ap(self, ap, rel): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/r2d2/nets/repeatability_loss.py class CosimLoss (line 13) | class CosimLoss (nn.Module): method __init__ (line 16) | def __init__(self, N=16): method extract_patches (line 21) | def extract_patches(self, sal): method forward (line 26) | def forward(self, repeatability, aflow, **kw): class PeakyLoss (line 41) | class PeakyLoss (nn.Module): method __init__ (line 47) | def __init__(self, N=16): method forward_one (line 55) | def forward_one(self, sali): method forward (line 59) | def forward(self, repeatability, **kw): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/r2d2/nets/sampler.py class FullSampler (line 17) | class FullSampler(nn.Module): method __init__ (line 22) | def __init__(self): method _aflow_to_grid (line 28) | def _aflow_to_grid(aflow): method _warp (line 37) | def _warp(self, feats, confs, aflow): method _warp_positions (line 56) | def _warp_positions(self, aflow): class SubSampler (line 71) | class SubSampler (FullSampler): method __init__ (line 74) | def __init__(self, border, subq, subd, perimage=False): method __repr__ (line 82) | def __repr__(self): method __call__ (line 86) | def __call__(self, feats, confs, aflow): class NghSampler (line 138) | class NghSampler (FullSampler): method __init__ (line 141) | def __init__(self, ngh, subq=1, subd=1, ignore=1, border=None): method __repr__ (line 153) | def __repr__(self): method trans (line 157) | def trans(self, arr, i, j): method __call__ (line 161) | def __call__(self, feats, confs, aflow): class FarNearSampler (line 193) | class FarNearSampler (FullSampler): method __init__ (line 212) | def __init__(self, subq, ngh, subd_ngh, subd_far, border=None, ignore=1, method __repr__ (line 222) | def __repr__(self): method __call__ (line 230) | def __call__(self, feats, confs, aflow): class NghSampler2 (line 252) | class NghSampler2 (nn.Module): method __init__ (line 261) | def __init__(self, ngh, subq=1, subd=1, pos_d=0, neg_d=2, border=None, method precompute_offsets (line 279) | def precompute_offsets(self): method gen_grid (line 297) | def gen_grid(self, step, aflow): method forward (line 321) | def forward(self, feats, confs, aflow, **kw): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/r2d2/tools/common.py function mkdir_for (line 10) | def mkdir_for(file_path): function model_size (line 14) | def model_size(model): function torch_set_gpu (line 23) | def torch_set_gpu(gpus): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/r2d2/tools/dataloader.py class PairLoader (line 22) | class PairLoader: method __init__ (line 35) | def __init__(self, dataset, crop='', scale='', distort='', norm = norm... method __len__ (line 48) | def __len__(self): method __repr__ (line 52) | def __repr__(self): method __getitem__ (line 62) | def __getitem__(self, i): function threaded_loader (line 257) | def threaded_loader( loader, iscuda, threads, batch_size=1, shuffle=True): function collate (line 287) | def collate(batch, _use_shared_memory=True): function tensor2img (line 333) | def tensor2img(tensor, model=None): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/r2d2/tools/trainer.py class Trainer (line 13) | class Trainer (nn.Module): method __init__ (line 22) | def __init__(self, net, loader, loss, optimizer): method iscuda (line 29) | def iscuda(self): method todevice (line 32) | def todevice(self, x): method __call__ (line 43) | def __call__(self): method forward_backward (line 71) | def forward_backward(self, inputs): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/r2d2/tools/transforms.py function instanciate_transformation (line 21) | def instanciate_transformation(cmd_line): class Scale (line 38) | class Scale (object): method __init__ (line 49) | def __init__(self, size, interpolation=Image.BILINEAR, largest=False, method __repr__ (line 58) | def __repr__(self): method get_params (line 65) | def get_params(self, imsize): method __call__ (line 81) | def __call__(self, inp): class RandomScale (line 96) | class RandomScale (Scale): method __init__ (line 111) | def __init__(self, min_size, max_size, ar=1, method get_params (line 123) | def get_params(self, imsize): class RandomCrop (line 157) | class RandomCrop (object): method __init__ (line 171) | def __init__(self, size, padding=0): method __repr__ (line 178) | def __repr__(self): method get_params (line 182) | def get_params(img, output_size): method __call__ (line 191) | def __call__(self, inp): class CenterCrop (line 212) | class CenterCrop (RandomCrop): method get_params (line 222) | def get_params(img, output_size): class RandomRotation (line 231) | class RandomRotation(object): method __init__ (line 243) | def __init__(self, degrees, interpolation=Image.BILINEAR): method __call__ (line 247) | def __call__(self, inp): class RandomTilting (line 263) | class RandomTilting(object): method __init__ (line 275) | def __init__(self, magnitude, directions='all'): method __repr__ (line 279) | def __repr__(self): method __call__ (line 282) | def __call__(self, inp): class Tilt (line 357) | class Tilt(object): method __init__ (line 360) | def __init__(self, *homography): method __call__ (line 364) | def __call__(self, inp): class StillTransform (line 376) | class StillTransform (object): method _transform (line 379) | def _transform(self, img): method __call__ (line 382) | def __call__(self, inp): class PixelNoise (line 395) | class PixelNoise (StillTransform): method __init__ (line 398) | def __init__(self, ampl=20): method __repr__ (line 403) | def __repr__(self): method _transform (line 406) | def _transform(self, img): class ColorJitter (line 413) | class ColorJitter (StillTransform): method __init__ (line 427) | def __init__(self, brightness=0, contrast=0, saturation=0, hue=0): method __repr__ (line 433) | def __repr__(self): method get_params (line 438) | def get_params(brightness, contrast, saturation, hue): method _transform (line 469) | def _transform(self, img): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/r2d2/tools/transforms_tools.py class DummyImg (line 10) | class DummyImg: method __init__ (line 13) | def __init__(self, size): method resize (line 16) | def resize(self, size, *args, **kwargs): method expand (line 19) | def expand(self, border): method crop (line 28) | def crop(self, border): method rotate (line 36) | def rotate(self, angle): method transform (line 39) | def transform(self, size, *args, **kwargs): function grab_img (line 43) | def grab_img( img_and_label ): function update_img_and_labels (line 60) | def update_img_and_labels(img_and_label, img, persp=None): function rand_log_uniform (line 79) | def rand_log_uniform(a, b): function translate (line 83) | def translate(tx, ty): function rotate (line 88) | def rotate(angle): function persp_mul (line 94) | def persp_mul(mat, mat2): function persp_apply (line 108) | def persp_apply(mat, pts): function is_pil_image (line 128) | def is_pil_image(img): function adjust_brightness (line 132) | def adjust_brightness(img, brightness_factor): function adjust_contrast (line 151) | def adjust_contrast(img, contrast_factor): function adjust_saturation (line 170) | def adjust_saturation(img, saturation_factor): function adjust_hue (line 189) | def adjust_hue(img, hue_factor): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/r2d2/tools/viz.py function make_colorwheel (line 10) | def make_colorwheel(): function flow_compute_color (line 59) | def flow_compute_color(u, v, convert_to_bgr=False): function flow_to_color (line 105) | def flow_to_color(flow_uv, clip_flow=None, convert_to_bgr=False): function show_flow (line 138) | def show_flow( img0, img1, flow, mask=None ): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/r2d2/train.py class MyTrainer (line 59) | class MyTrainer(trainer.Trainer): method forward_backward (line 63) | def forward_backward(self, inputs): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/Hierarchical-Localization/third_party/r2d2/viz_heatmaps.py function transparent (line 14) | def transparent(img, alpha, cmap, **kw): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/loftr/config/default.py function get_cfg_defaults (line 167) | def get_cfg_defaults(): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/loftr/datasets/megadepth.py class MegaDepthDataset (line 11) | class MegaDepthDataset(Dataset): method __init__ (line 12) | def __init__(self, method __len__ (line 64) | def __len__(self): method __getitem__ (line 67) | def __getitem__(self, idx): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/loftr/datasets/sampler.py class RandomConcatSampler (line 5) | class RandomConcatSampler(Sampler): method __init__ (line 21) | def __init__(self, method __len__ (line 41) | def __len__(self): method __iter__ (line 44) | def __iter__(self): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/loftr/datasets/scannet.py class ScanNetDataset (line 17) | class ScanNetDataset(utils.data.Dataset): method __init__ (line 18) | def __init__(self, method __len__ (line 53) | def __len__(self): method _read_abs_pose (line 56) | def _read_abs_pose(self, scene_name, name): method _compute_rel_pose (line 62) | def _compute_rel_pose(self, scene_name, name0, name1): method __getitem__ (line 68) | def __getitem__(self, idx): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/loftr/lightning/data.py class MultiSceneDataModule (line 31) | class MultiSceneDataModule(pl.LightningDataModule): method __init__ (line 36) | def __init__(self, args, config): method setup (line 106) | def setup(self, stage=None): method _setup_dataset (line 168) | def _setup_dataset(self, method _build_concat_dataset (line 192) | def _build_concat_dataset( method _build_concat_dataset_parallel (line 237) | def _build_concat_dataset_parallel( method train_dataloader (line 286) | def train_dataloader(self): method val_dataloader (line 300) | def val_dataloader(self): method test_dataloader (line 313) | def test_dataloader(self, *args, **kwargs): function _build_dataset (line 319) | def _build_dataset(dataset: Dataset, *args, **kwargs): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/loftr/lightning/lightning_loftr.py class PL_LoFTR (line 27) | class PL_LoFTR(pl.LightningModule): method __init__ (line 28) | def __init__(self, config, pretrained_ckpt=None, profiler=None, dump_d... method configure_optimizers (line 54) | def configure_optimizers(self): method optimizer_step (line 60) | def optimizer_step( method _trainval_inference (line 82) | def _trainval_inference(self, batch): method _compute_metrics (line 95) | def _compute_metrics(self, batch): method training_step (line 112) | def training_step(self, batch, batch_idx): method training_epoch_end (line 135) | def training_epoch_end(self, outputs): method validation_step (line 142) | def validation_step(self, batch, batch_idx): method validation_epoch_end (line 158) | def validation_epoch_end(self, outputs): method test_step (line 205) | def test_step(self, batch, batch_idx): method test_epoch_end (line 232) | def test_epoch_end(self, outputs): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/loftr/loftr/backbone/__init__.py function build_backbone (line 4) | def build_backbone(config): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/loftr/loftr/backbone/resnet_fpn.py function conv1x1 (line 5) | def conv1x1(in_planes, out_planes, stride=1): function conv3x3 (line 10) | def conv3x3(in_planes, out_planes, stride=1): class BasicBlock (line 15) | class BasicBlock(nn.Module): method __init__ (line 16) | def __init__(self, in_planes, planes, stride=1): method forward (line 32) | def forward(self, x): class ResNetFPN_8_2 (line 43) | class ResNetFPN_8_2(nn.Module): method __init__ (line 49) | def __init__(self, config): method _make_layer (line 92) | def _make_layer(self, block, dim, stride=1): method forward (line 100) | def forward(self, x): class ResNetFPN_16_4 (line 121) | class ResNetFPN_16_4(nn.Module): method __init__ (line 127) | def __init__(self, config): method _make_layer (line 172) | def _make_layer(self, block, dim, stride=1): method forward (line 180) | def forward(self, x): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/loftr/loftr/loftr.py class LoFTR (line 12) | class LoFTR(nn.Module): method __init__ (line 13) | def __init__(self, config): method forward (line 29) | def forward(self, data, coarse_only=False): method load_state_dict (line 79) | def load_state_dict(self, state_dict, *args, **kwargs): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/loftr/loftr/loftr_module/fine_preprocess.py class FinePreprocess (line 7) | class FinePreprocess(nn.Module): method __init__ (line 8) | def __init__(self, config): method _reset_parameters (line 24) | def _reset_parameters(self): method forward (line 29) | def forward(self, feat_f0, feat_f1, feat_c0, feat_c1, data): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/loftr/loftr/loftr_module/linear_attention.py function elu_feature_map (line 10) | def elu_feature_map(x): class LinearAttention (line 14) | class LinearAttention(Module): method __init__ (line 15) | def __init__(self, eps=1e-6): method forward (line 20) | def forward(self, queries, keys, values, q_mask=None, kv_mask=None): class FullAttention (line 50) | class FullAttention(Module): method __init__ (line 51) | def __init__(self, use_dropout=False, attention_dropout=0.1): method forward (line 56) | def forward(self, queries, keys, values, q_mask=None, kv_mask=None): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/loftr/loftr/loftr_module/transformer.py class LoFTREncoderLayer (line 7) | class LoFTREncoderLayer(nn.Module): method __init__ (line 8) | def __init__(self, method forward (line 35) | def forward(self, x, source, x_mask=None, source_mask=None): class LocalFeatureTransformer (line 61) | class LocalFeatureTransformer(nn.Module): method __init__ (line 64) | def __init__(self, config): method _reset_parameters (line 75) | def _reset_parameters(self): method forward (line 80) | def forward(self, feat0, feat1, mask0=None, mask1=None): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/loftr/loftr/utils/coarse_matching.py function mask_border (line 8) | def mask_border(m, b: int, v): function mask_border_with_padding (line 28) | def mask_border_with_padding(m, bd, v, p_m0, p_m1): function compute_max_candidates (line 46) | def compute_max_candidates(p_m0, p_m1): class CoarseMatching (line 59) | class CoarseMatching(nn.Module): method __init__ (line 60) | def __init__(self, config): method forward (line 87) | def forward(self, feat_c0, feat_c1, data, mask_c0=None, mask_c1=None): method get_coarse_match (line 151) | def get_coarse_match(self, conf_matrix, data): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/loftr/loftr/utils/cvpr_ds_config.py function lower_config (line 4) | def lower_config(yacs_cfg): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/loftr/loftr/utils/fine_matching.py class FineMatching (line 9) | class FineMatching(nn.Module): method __init__ (line 12) | def __init__(self): method forward (line 15) | def forward(self, feat_f0, feat_f1, data): method get_fine_match (line 63) | def get_fine_match(self, coords_normed, data): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/loftr/loftr/utils/geometry.py function warp_kpts (line 5) | def warp_kpts(kpts0, depth0, depth1, T_0to1, K0, K1): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/loftr/loftr/utils/position_encoding.py class PositionEncodingSine (line 6) | class PositionEncodingSine(nn.Module): method __init__ (line 11) | def __init__(self, d_model, max_shape=(256, 256), temp_bug_fix=True): method forward (line 37) | def forward(self, x): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/loftr/loftr/utils/supervision.py function mask_pts_at_padded_regions (line 14) | def mask_pts_at_padded_regions(grid_pt, mask): function spvs_coarse (line 22) | def spvs_coarse(data, config): function compute_supervision_coarse (line 112) | def compute_supervision_coarse(data, config): function spvs_fine (line 124) | def spvs_fine(data, config): function compute_supervision_fine (line 146) | def compute_supervision_fine(data, config): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/loftr/losses/loftr_loss.py class LoFTRLoss (line 7) | class LoFTRLoss(nn.Module): method __init__ (line 8) | def __init__(self, config): method compute_coarse_loss (line 22) | def compute_coarse_loss(self, conf, conf_gt, weight=None): method compute_fine_loss (line 98) | def compute_fine_loss(self, expec_f, expec_f_gt): method _compute_fine_loss_l2 (line 106) | def _compute_fine_loss_l2(self, expec_f, expec_f_gt): method _compute_fine_loss_l2_std (line 122) | def _compute_fine_loss_l2_std(self, expec_f, expec_f_gt): method compute_c_weight (line 153) | def compute_c_weight(self, data): method forward (line 161) | def forward(self, data): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/loftr/optimizers/__init__.py function build_optimizer (line 5) | def build_optimizer(model, config): function build_scheduler (line 17) | def build_scheduler(config, optimizer): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/loftr/tools/hloc_match_features.py function list_h5_names (line 26) | def list_h5_names(path): class MatchDataset (line 38) | class MatchDataset(Dataset): method __init__ (line 39) | def __init__( method __len__ (line 56) | def __len__(self): method __getitem__ (line 59) | def __getitem__(self, idx): method _parse_image_pairs (line 66) | def _parse_image_pairs(self, pairs_path): method _preprocess_image (line 73) | def _preprocess_image(self, img: np.array): class MatchSaver (line 82) | class MatchSaver(object): method __init__ (line 83) | def __init__(self, save_dir, save_fn, n_pairs, queue): method run (line 91) | def run(self): class MatchExtrator (line 114) | class MatchExtrator(object): method __init__ (line 115) | def __init__(self, args, rank, world_size, queue, coarse_hws_cached): method run (line 131) | def run(self): function extract_coarse_matches (line 145) | def extract_coarse_matches(matcher, data): function build_coarse_matches_hloc (line 165) | def build_coarse_matches_hloc(result, coarse_hws0): function check_consistent_image_sizes (line 181) | def check_consistent_image_sizes(image_dir, image_list): function compute_resize_max_area (line 187) | def compute_resize_max_area(h, w, max_area=1920000, df=8): function _build_pseudo_features (line 198) | def _build_pseudo_features(image_path, max_area=1920000, df=8, coarse_sc... function build_pseudo_features (line 211) | def build_pseudo_features( function get_parser (line 261) | def get_parser(): function get_default_args (line 281) | def get_default_args(): function build_dataset (line 285) | def build_dataset(args): function build_dataloader (line 297) | def build_dataloader(args, dataset, rank, world_size): function check_existence (line 305) | def check_existence(args, image_list): function process_instance (line 346) | def process_instance(args, image_list, block=True): # process a single ... function main (line 387) | def main(): # Testinng FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/loftr/utils/augment.py class DarkAug (line 4) | class DarkAug(object): method __init__ (line 9) | def __init__(self) -> None: method __call__ (line 18) | def __call__(self, x): class MobileAug (line 22) | class MobileAug(object): method __init__ (line 27) | def __init__(self): method __call__ (line 37) | def __call__(self, x): function build_augmentor (line 41) | def build_augmentor(method=None, **kwargs): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/loftr/utils/comm.py function get_world_size (line 22) | def get_world_size() -> int: function get_rank (line 30) | def get_rank() -> int: function get_local_rank (line 38) | def get_local_rank() -> int: function get_local_size (line 51) | def get_local_size() -> int: function is_main_process (line 64) | def is_main_process() -> bool: function synchronize (line 68) | def synchronize(): function _get_global_gloo_group (line 84) | def _get_global_gloo_group(): function _serialize_to_tensor (line 95) | def _serialize_to_tensor(data, group): function _pad_to_largest_tensor (line 113) | def _pad_to_largest_tensor(tensor, group): function all_gather (line 141) | def all_gather(data, group=None): function gather (line 179) | def gather(data, dst=0, group=None): function shared_random_seed (line 222) | def shared_random_seed(): function reduce_dict (line 236) | def reduce_dict(input_dict, average=True): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/loftr/utils/dataloader.py function get_local_split (line 6) | def get_local_split(items: list, world_size: int, rank: int, seed: int): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/loftr/utils/dataset.py function load_array_from_s3 (line 19) | def load_array_from_s3( function imread_gray (line 39) | def imread_gray(path, augment_fn=None, client=SCANNET_CLIENT): function get_resized_wh (line 55) | def get_resized_wh(w, h, resize=None): function get_divisible_wh (line 64) | def get_divisible_wh(w, h, df=None): function pad_bottom_right (line 72) | def pad_bottom_right(inp, pad_size, ret_mask=False): function read_megadepth_gray (line 94) | def read_megadepth_gray(path, resize=None, df=None, padding=False, augme... function read_megadepth_depth (line 128) | def read_megadepth_depth(path, pad_to=None): function read_scannet_gray (line 141) | def read_scannet_gray(path, resize=(640, 480), augment_fn=None): function read_scannet_depth (line 160) | def read_scannet_depth(path): function read_scannet_pose (line 170) | def read_scannet_pose(path): function read_scannet_intrinsic (line 181) | def read_scannet_intrinsic(path): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/loftr/utils/metrics.py function relative_pose_error (line 12) | def relative_pose_error(T_0to1, R, t, ignore_gt_t_thr=0.0): function symmetric_epipolar_distance (line 30) | def symmetric_epipolar_distance(pts0, pts1, E, K0, K1): function compute_symmetrical_epipolar_errors (line 50) | def compute_symmetrical_epipolar_errors(data): function estimate_pose (line 72) | def estimate_pose(kpts0, kpts1, K0, K1, thresh, conf=0.99999): function compute_pose_errors (line 101) | def compute_pose_errors(data, config): function error_auc (line 139) | def error_auc(errors, thresholds): function epidist_prec (line 159) | def epidist_prec(errors, thresholds, ret_dict=False): function aggregate_metrics (line 173) | def aggregate_metrics(metrics, epi_err_thr=5e-4): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/loftr/utils/misc.py function lower_config (line 13) | def lower_config(yacs_cfg): function upper_config (line 19) | def upper_config(dict_cfg): function log_on (line 25) | def log_on(condition, message, level): function get_rank_zero_only_logger (line 31) | def get_rank_zero_only_logger(logger: _Logger): function setup_gpus (line 43) | def setup_gpus(gpus: Union[str, int]) -> int: function flattenList (line 66) | def flattenList(x): function tqdm_joblib (line 71) | def tqdm_joblib(tqdm_object): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/loftr/utils/plotting.py function _compute_conf_thresh (line 7) | def _compute_conf_thresh(data): function make_matching_figure (line 20) | def make_matching_figure( function _make_evaluation_figure (line 68) | def _make_evaluation_figure(data, b_id, alpha='dynamic'): function _make_confidence_figure (line 107) | def _make_confidence_figure(data, b_id): function make_matching_figures (line 112) | def make_matching_figures(data, config, mode='evaluation'): function dynamic_alpha (line 136) | def dynamic_alpha(n_matches, function error_colormap (line 150) | def error_colormap(err, thr, alpha=1.0): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/loftr/utils/profiler.py class InferenceProfiler (line 7) | class InferenceProfiler(SimpleProfiler): method __init__ (line 13) | def __init__(self): method profile (line 20) | def profile(self, action_name: str) -> None: function build_profiler (line 30) | def build_profiler(name): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/test.py function parse_args (line 13) | def parse_args(): FILE: AutoReconForDens3R/third_party/AutoDecomp/third_party/LoFTR/train.py function parse_args (line 23) | def parse_args(): function main (line 57) | def main(): FILE: croco/datasets/crops/extract_crops_from_images.py function arg_parser (line 17) | def arg_parser(): function main (line 30) | def main(args): function load_crop_file (line 56) | def load_crop_file(path): function prepare_jobs (line 75) | def prepare_jobs(pairs, num_levels, num_pairs_in_dir): function load_image (line 99) | def load_image(path): function save_image_crops (line 107) | def save_image_crops(args, data): FILE: croco/datasets/habitat_sim/generate_from_metadata.py function generate_multiview_images_from_metadata (line 17) | def generate_multiview_images_from_metadata(metadata_filename, FILE: croco/datasets/habitat_sim/generate_multiview_images.py function generate_multiview_images_for_scene (line 16) | def generate_multiview_images_for_scene(scene_dataset_config_file, function create_commandline (line 119) | def create_commandline(scene_data, generate_depth, exist_ok=False): FILE: croco/datasets/habitat_sim/multiview_habitat_sim_generator.py function compute_camera_intrinsics (line 17) | def compute_camera_intrinsics(height, width, hfov): function compute_camera_pose_opencv_convention (line 22) | def compute_camera_pose_opencv_convention(camera_position, camera_orient... function compute_pointmap (line 27) | def compute_pointmap(depthmap, hfov): function compute_pointcloud (line 39) | def compute_pointcloud(depthmap, hfov, camera_position, camera_rotation): function compute_pointcloud_overlaps_scikit (line 50) | def compute_pointcloud_overlaps_scikit(pointcloud1, pointcloud2, distanc... function _append_camera_parameters (line 69) | def _append_camera_parameters(observation, hfov, camera_location, camera... function look_at (line 84) | def look_at(eye, center, up, return_cam2world=True): function look_at_for_habitat (line 106) | def look_at_for_habitat(eye, center, up, return_cam2world=True): function generate_orientation_noise (line 111) | def generate_orientation_noise(pan_range, tilt_range, roll_range): class NoNaviguableSpaceError (line 117) | class NoNaviguableSpaceError(RuntimeError): method __init__ (line 118) | def __init__(self, *args): class MultiviewHabitatSimGenerator (line 121) | class MultiviewHabitatSimGenerator: method __init__ (line 122) | def __init__(self, method _lazy_initialization (line 170) | def _lazy_initialization(self): method close (line 218) | def close(self): method __del__ (line 221) | def __del__(self): method __len__ (line 224) | def __len__(self): method sample_random_viewpoint (line 227) | def sample_random_viewpoint(self): method sample_other_random_viewpoint (line 237) | def sample_other_random_viewpoint(self, observed_point, nav_point): method is_other_pointcloud_overlapping (line 256) | def is_other_pointcloud_overlapping(self, ref_pointcloud, other_pointc... method is_other_viewpoint_overlapping (line 267) | def is_other_viewpoint_overlapping(self, ref_pointcloud, observation, ... method render_viewpoint (line 273) | def render_viewpoint(self, viewpoint_position, viewpoint_orientation): method __getitem__ (line 282) | def __getitem__(self, useless_idx): method generate_random_spiral_trajectory (line 352) | def generate_random_spiral_trajectory(self, images_count = 100, max_ra... FILE: croco/datasets/habitat_sim/paths.py function list_replicacad_scenes (line 27) | def list_replicacad_scenes(base_output_dir, base_path=SCENES_DATASET["re... function list_replica_cad_baked_lighting_scenes (line 42) | def list_replica_cad_baked_lighting_scenes(base_output_dir, base_path=SC... function list_replica_scenes (line 56) | def list_replica_scenes(base_output_dir, base_path): function list_scenes (line 72) | def list_scenes(base_output_dir, base_path): function list_scenes_available (line 103) | def list_scenes_available(base_output_dir, scenes_dataset_paths=SCENES_D... FILE: croco/datasets/pairs_dataset.py function load_image (line 10) | def load_image(impath): function load_pairs_from_cache_file (line 13) | def load_pairs_from_cache_file(fname, root=''): function load_pairs_from_list_file (line 20) | def load_pairs_from_list_file(fname, root=''): function write_cache_file (line 28) | def write_cache_file(fname, pairs, root=''): function parse_and_cache_all_pairs (line 41) | def parse_and_cache_all_pairs(dname, data_dir='./data/'): function dnames_to_image_pairs (line 61) | def dnames_to_image_pairs(dnames, data_dir='./data/'): class PairsDataset (line 85) | class PairsDataset(Dataset): method __init__ (line 87) | def __init__(self, dnames, trfs='', totensor=True, normalize=True, dat... method __len__ (line 92) | def __len__(self): method __getitem__ (line 95) | def __getitem__(self, index): FILE: croco/datasets/transforms.py class ComposePair (line 11) | class ComposePair(torchvision.transforms.Compose): method __call__ (line 12) | def __call__(self, img1, img2): class NormalizeBoth (line 17) | class NormalizeBoth(torchvision.transforms.Normalize): method forward (line 18) | def forward(self, img1, img2): class ToTensorBoth (line 23) | class ToTensorBoth(torchvision.transforms.ToTensor): method __call__ (line 24) | def __call__(self, img1, img2): class RandomCropPair (line 29) | class RandomCropPair(torchvision.transforms.RandomCrop): method forward (line 31) | def forward(self, img1, img2): class ColorJitterPair (line 36) | class ColorJitterPair(torchvision.transforms.ColorJitter): method __init__ (line 38) | def __init__(self, assymetric_prob, **kwargs): method jitter_one (line 41) | def jitter_one(self, img, fn_idx, brightness_factor, contrast_factor, ... method forward (line 53) | def forward(self, img1, img2): function get_pair_transforms (line 66) | def get_pair_transforms(transform_str, totensor=True, normalize=True): FILE: croco/models/blocks.py function _ntuple (line 24) | def _ntuple(n): function drop_path (line 32) | def drop_path(x, drop_prob: float = 0., training: bool = False, scale_by... class DropPath (line 44) | class DropPath(nn.Module): method __init__ (line 47) | def __init__(self, drop_prob: float = 0., scale_by_keep: bool = True): method forward (line 52) | def forward(self, x): method extra_repr (line 55) | def extra_repr(self): class Mlp (line 58) | class Mlp(nn.Module): method __init__ (line 60) | def __init__(self, in_features, hidden_features=None, out_features=Non... method forward (line 73) | def forward(self, x): class Attention (line 81) | class Attention(nn.Module): method __init__ (line 83) | def __init__(self, dim, rope=None, num_heads=8, qkv_bias=False, attn_d... method forward (line 94) | def forward(self, x, xpos): class Block (line 114) | class Block(nn.Module): method __init__ (line 116) | def __init__(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, drop=... method forward (line 127) | def forward(self, x, xpos): class ATMBlock (line 132) | class ATMBlock(nn.Module): method __init__ (line 134) | def __init__(self, dim, num_heads, qkv_bias=False, attn_drop=0., drop=... method forward (line 141) | def forward(self, tokens, each_grid_h, each_grid_w): class CrossAttention (line 167) | class CrossAttention(nn.Module): method __init__ (line 169) | def __init__(self, dim, rope=None, num_heads=8, qkv_bias=False, attn_d... method forward (line 184) | def forward(self, query, key, value, qpos, kpos): class DecoderBlock (line 206) | class DecoderBlock(nn.Module): method __init__ (line 208) | def __init__(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, drop=... method forward (line 221) | def forward(self, x, y, xpos, ypos): class PositionGetter (line 230) | class PositionGetter(object): method __init__ (line 233) | def __init__(self): method __call__ (line 236) | def __call__(self, b, h, w, device): class PatchEmbed (line 244) | class PatchEmbed(nn.Module): method __init__ (line 247) | def __init__(self, img_size=224, patch_size=16, in_chans=3, embed_dim=... method forward (line 262) | def forward(self, x): method _init_weights (line 273) | def _init_weights(self): FILE: croco/models/criterion.py class MaskedMSE (line 13) | class MaskedMSE(torch.nn.Module): method __init__ (line 15) | def __init__(self, norm_pix_loss=False, masked=True): method forward (line 24) | def forward(self, pred, mask, target): FILE: croco/models/croco.py class CroCoNet (line 21) | class CroCoNet(nn.Module): method __init__ (line 23) | def __init__(self, method interpolate_pose_encoding_dust3r (line 105) | def interpolate_pose_encoding_dust3r(self, x, pos_embed_code, img_w, i... method _set_patch_embed (line 159) | def _set_patch_embed(self, img_size=224, patch_size=16, enc_embed_dim=... method _set_mask_generator (line 162) | def _set_mask_generator(self, num_patches, mask_ratio): method _set_mask_token (line 165) | def _set_mask_token(self, dec_embed_dim): method _set_decoder (line 168) | def _set_decoder(self, enc_embed_dim, dec_embed_dim, dec_num_heads, de... method _set_prediction_head (line 180) | def _set_prediction_head(self, dec_embed_dim, patch_size): method initialize_weights (line 184) | def initialize_weights(self): method _init_weights (line 192) | def _init_weights(self, m): method _encode_image (line 202) | def _encode_image(self, image, do_mask=False, return_all_blocks=False): method _decoder (line 239) | def _decoder(self, feat1, pos1, masks1, feat2, pos2, return_all_blocks... method patchify (line 276) | def patchify(self, imgs): method unpatchify (line 291) | def unpatchify(self, x, channels=3): method forward (line 304) | def forward(self, img1, img2): FILE: croco/models/croco_downstream.py function croco_args_from_ckpt (line 13) | def croco_args_from_ckpt(ckpt): class CroCoDownstreamMonocularEncoder (line 23) | class CroCoDownstreamMonocularEncoder(CroCoNet): method __init__ (line 25) | def __init__(self, method _set_mask_generator (line 39) | def _set_mask_generator(self, *args, **kwargs): method _set_mask_token (line 43) | def _set_mask_token(self, *args, **kwargs): method _set_decoder (line 48) | def _set_decoder(self, *args, **kwargs): method _set_prediction_head (line 52) | def _set_prediction_head(self, *args, **kwargs): method forward (line 56) | def forward(self, img): class CroCoDownstreamBinocular (line 67) | class CroCoDownstreamBinocular(CroCoNet): method __init__ (line 69) | def __init__(self, method _set_mask_generator (line 81) | def _set_mask_generator(self, *args, **kwargs): method _set_mask_token (line 85) | def _set_mask_token(self, *args, **kwargs): method _set_prediction_head (line 90) | def _set_prediction_head(self, *args, **kwargs): method encode_image_pairs (line 94) | def encode_image_pairs(self, img1, img2, return_all_blocks=False): method forward (line 112) | def forward(self, img1, img2): FILE: croco/models/curope/curope.cpp function rope_2d_cpu (line 11) | void rope_2d_cpu( torch::Tensor tokens, const torch::Tensor positions, c... function rope_2d (line 49) | void rope_2d( torch::Tensor tokens, // B,N,H,D function PYBIND11_MODULE (line 67) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) { FILE: croco/models/curope/curope2d.py class cuRoPE2D_func (line 12) | class cuRoPE2D_func (torch.autograd.Function): method forward (line 15) | def forward(ctx, tokens, positions, base, F0=1): method backward (line 25) | def backward(ctx, grad_res): class cuRoPE2D (line 32) | class cuRoPE2D(torch.nn.Module): method __init__ (line 33) | def __init__(self, freq=100.0, F0=1.0): method forward (line 38) | def forward(self, tokens, positions): FILE: croco/models/dpt_block.py function pair (line 22) | def pair(t): function make_scratch (line 25) | def make_scratch(in_shape, out_shape, groups=1, expand=False): class ResidualConvUnit_custom (line 84) | class ResidualConvUnit_custom(nn.Module): method __init__ (line 87) | def __init__(self, features, activation, bn): method forward (line 126) | def forward(self, x): class FeatureFusionBlock_custom (line 149) | class FeatureFusionBlock_custom(nn.Module): method __init__ (line 152) | def __init__( method forward (line 194) | def forward(self, *xs): function make_fusion_block (line 225) | def make_fusion_block(features, use_bn, width_ratio=1): class Interpolate (line 236) | class Interpolate(nn.Module): method __init__ (line 239) | def __init__(self, scale_factor, mode, align_corners=False): method forward (line 252) | def forward(self, x): class DPTOutputAdapter (line 269) | class DPTOutputAdapter(nn.Module): method __init__ (line 285) | def __init__(self, method init (line 347) | def init(self, dim_tokens_enc=768): method adapt_tokens (line 418) | def adapt_tokens(self, encoder_tokens): method forward (line 425) | def forward(self, encoder_tokens: List[torch.Tensor], image_size): FILE: croco/models/head_downstream.py class PixelwiseTaskWithDPT (line 19) | class PixelwiseTaskWithDPT(nn.Module): method __init__ (line 26) | def __init__(self, *, hooks_idx=None, layer_dims=[96,192,384,768], method setup (line 36) | def setup(self, croconet): method forward (line 55) | def forward(self, x, img_info): FILE: croco/models/masking.py class RandomMask (line 12) | class RandomMask(nn.Module): method __init__ (line 17) | def __init__(self, num_patches, mask_ratio): method __call__ (line 22) | def __call__(self, x): FILE: croco/models/pos_embed.py function get_2d_sincos_pos_embed (line 22) | def get_2d_sincos_pos_embed(embed_dim, grid_size, n_cls_token=0): function get_2d_sincos_pos_embed_from_grid (line 40) | def get_2d_sincos_pos_embed_from_grid(embed_dim, grid): function get_1d_sincos_pos_embed_from_grid (line 51) | def get_1d_sincos_pos_embed_from_grid(embed_dim, pos): function interpolate_pos_embed (line 78) | def interpolate_pos_embed(model, checkpoint_model): class RoPE2D (line 112) | class RoPE2D(torch.nn.Module): method __init__ (line 114) | def __init__(self, freq=100.0, F0=1.0): method get_cos_sin (line 121) | def get_cos_sin(self, D, seq_len, device, dtype): method rotate_half (line 138) | def rotate_half(x): method apply_rope1d (line 142) | def apply_rope1d(self, tokens, pos1d, cos, sin): method forward (line 148) | def forward(self, tokens, positions): FILE: croco/pretrain.py function get_args_parser (line 37) | def get_args_parser(): function main (line 73) | def main(args): function train_one_epoch (line 190) | def train_one_epoch(model: torch.nn.Module, criterion: torch.nn.Module, FILE: croco/stereoflow/augmentor.py class StereoAugmentor (line 25) | class StereoAugmentor(object): method __init__ (line 27) | def __init__(self, crop_size, scale_prob=0.5, scale_xonly=True, lhth=8... method _random_scale (line 42) | def _random_scale(self, img1, img2, disp): method _random_crop (line 67) | def _random_crop(self, img1, img2, disp): method _random_vflip (line 78) | def _random_vflip(self, img1, img2, disp): method _random_rotate_shift_right (line 86) | def _random_rotate_shift_right(self, img2): method _random_color_contrast (line 98) | def _random_color_contrast(self, img1, img2): method _random_color_gamma (line 105) | def _random_color_gamma(self, img1, img2): method _random_color_brightness (line 112) | def _random_color_brightness(self, img1, img2): method _random_color_hue (line 119) | def _random_color_hue(self, img1, img2): method _random_color_saturation (line 126) | def _random_color_saturation(self, img1, img2): method _random_color (line 133) | def _random_color(self, img1, img2): method __call__ (line 151) | def __call__(self, img1, img2, disp, dataset_name): class FlowAugmentor (line 161) | class FlowAugmentor: method __init__ (line 163) | def __init__(self, crop_size, min_scale=-0.2, max_scale=0.5, spatial_a... method color_transform (line 182) | def color_transform(self, img1, img2): method _resize_flow (line 198) | def _resize_flow(self, flow, scale_x, scale_y, factor=1.0): method spatial_transform (line 234) | def spatial_transform(self, img1, img2, flow, dname): method __call__ (line 284) | def __call__(self, img1, img2, flow, dname): FILE: croco/stereoflow/criterion.py function _get_gtnorm (line 12) | def _get_gtnorm(gt): class L1Loss (line 20) | class L1Loss(nn.Module): method __init__ (line 22) | def __init__(self, max_gtnorm=None): method _error (line 27) | def _error(self, gt, predictions): method forward (line 30) | def forward(self, predictions, gt, inspect=False): class LaplacianLoss (line 41) | class LaplacianLoss(nn.Module): # used for CroCo-Stereo on ETH3D, d'=exp(d) method __init__ (line 43) | def __init__(self, max_gtnorm=None): method forward (line 48) | def forward(self, predictions, gt, conf): class LaplacianLossBounded (line 56) | class LaplacianLossBounded(nn.Module): # used for CroCo-Flow ; in the eq... method __init__ (line 57) | def __init__(self, max_gtnorm=10000., a=0.25, b=4.): method forward (line 63) | def forward(self, predictions, gt, conf): class LaplacianLossBounded2 (line 71) | class LaplacianLossBounded2(nn.Module): # used for CroCo-Stereo (except ... method __init__ (line 72) | def __init__(self, max_gtnorm=None, a=3.0, b=3.0): method forward (line 78) | def forward(self, predictions, gt, conf): class StereoMetrics (line 88) | class StereoMetrics(nn.Module): method __init__ (line 90) | def __init__(self, do_quantile=False): method forward (line 95) | def forward(self, predictions, gt): class FlowMetrics (line 113) | class FlowMetrics(nn.Module): method __init__ (line 114) | def __init__(self): method forward (line 118) | def forward(self, predictions, gt): class StereoDatasetMetrics (line 140) | class StereoDatasetMetrics(nn.Module): method __init__ (line 142) | def __init__(self): method reset (line 146) | def reset(self): method add_batch (line 152) | def add_batch(self, predictions, gt): method _compute_metrics (line 172) | def _compute_metrics(self): method get_results (line 180) | def get_results(self): class FlowDatasetMetrics (line 184) | class FlowDatasetMetrics(nn.Module): method __init__ (line 186) | def __init__(self): method reset (line 191) | def reset(self): method add_batch (line 201) | def add_batch(self, predictions, gt): method _compute_metrics (line 238) | def _compute_metrics(self): method get_results (line 249) | def get_results(self): FILE: croco/stereoflow/datasets_flow.py function flow_to_tensor (line 33) | def flow_to_tensor(disp): class FlowDataset (line 36) | class FlowDataset(data.Dataset): method __init__ (line 38) | def __init__(self, split, augmentor=False, crop_size=None, totensor=Tr... method prepare_data (line 51) | def prepare_data(self): method __len__ (line 57) | def __len__(self): method __getitem__ (line 60) | def __getitem__(self, index): method __rmul__ (line 88) | def __rmul__(self, v): method __str__ (line 93) | def __str__(self): method __repr__ (line 96) | def __repr__(self): method _set_root (line 104) | def _set_root(self): method _load_or_build_cache (line 108) | def _load_or_build_cache(self): class TartanAirDataset (line 120) | class TartanAirDataset(FlowDataset): method _prepare_data (line 122) | def _prepare_data(self): method _build_cache (line 132) | def _build_cache(self): class FlyingChairsDataset (line 139) | class FlyingChairsDataset(FlowDataset): method _prepare_data (line 141) | def _prepare_data(self): method _build_cache (line 151) | def _build_cache(self): class FlyingThingsDataset (line 160) | class FlyingThingsDataset(FlowDataset): method _prepare_data (line 162) | def _prepare_data(self): method _build_cache (line 172) | def _build_cache(self): class MPISintelDataset (line 202) | class MPISintelDataset(FlowDataset): method _prepare_data (line 204) | def _prepare_data(self): method _build_cache (line 214) | def _build_cache(self): method submission_save_pairname (line 234) | def submission_save_pairname(self, pairname, prediction, outdir, _time): method finalize_submission (line 240) | def finalize_submission(self, outdir): class SpringDataset (line 253) | class SpringDataset(FlowDataset): method _prepare_data (line 255) | def _prepare_data(self): method _build_cache (line 265) | def _build_cache(self): method submission_save_pairname (line 285) | def submission_save_pairname(self, pairname, prediction, outdir, time): method finalize_submission (line 293) | def finalize_submission(self, outdir): class Kitti12Dataset (line 307) | class Kitti12Dataset(FlowDataset): method _prepare_data (line 309) | def _prepare_data(self): method _build_cache (line 319) | def _build_cache(self): method submission_save_pairname (line 326) | def submission_save_pairname(self, pairname, prediction, outdir, time): method finalize_submission (line 333) | def finalize_submission(self, outdir): class Kitti15Dataset (line 341) | class Kitti15Dataset(FlowDataset): method _prepare_data (line 343) | def _prepare_data(self): method _build_cache (line 353) | def _build_cache(self): method submission_save_pairname (line 362) | def submission_save_pairname(self, pairname, prediction, outdir, time): method finalize_submission (line 369) | def finalize_submission(self, outdir): function _read_numpy_flow (line 378) | def _read_numpy_flow(filename): function _read_pfm_flow (line 381) | def _read_pfm_flow(filename): function readFlowFile (line 392) | def readFlowFile(filename): function writeFlowFile (line 425) | def writeFlowFile(flow,filename): function _read_kitti_flow (line 455) | def _read_kitti_flow(filename): function writeFlowKitti (line 467) | def writeFlowKitti(filename, uv): function writeFlo5File (line 473) | def writeFlo5File(flow, filename): function _read_hdf5_flow (line 477) | def _read_hdf5_flow(filename): function colorTest (line 491) | def colorTest(): function flowToColor (line 509) | def flowToColor(flow, maxflow=None, maxmaxflow=None, saturate=False): function flowMaxNorm (line 539) | def flowMaxNorm(flow): function _computeColor (line 550) | def _computeColor(flow, saturate=True): function get_train_dataset_flow (line 620) | def get_train_dataset_flow(dataset_str, augmentor=True, crop_size=None): function get_test_datasets_flow (line 628) | def get_test_datasets_flow(dataset_str): FILE: croco/stereoflow/datasets_stereo.py function img_to_tensor (line 44) | def img_to_tensor(img): function disp_to_tensor (line 48) | def disp_to_tensor(disp): class StereoDataset (line 51) | class StereoDataset(data.Dataset): method __init__ (line 53) | def __init__(self, split, augmentor=False, crop_size=None, totensor=Tr... method prepare_data (line 66) | def prepare_data(self): method __len__ (line 72) | def __len__(self): method __getitem__ (line 75) | def __getitem__(self, index): method __rmul__ (line 105) | def __rmul__(self, v): method __str__ (line 110) | def __str__(self): method __repr__ (line 113) | def __repr__(self): method _set_root (line 121) | def _set_root(self): method _load_or_build_cache (line 125) | def _load_or_build_cache(self): class CREStereoDataset (line 137) | class CREStereoDataset(StereoDataset): method _prepare_data (line 139) | def _prepare_data(self): method _build_cache (line 150) | def _build_cache(self): class SceneFlowDataset (line 156) | class SceneFlowDataset(StereoDataset): method _prepare_data (line 158) | def _prepare_data(self): method _build_cache (line 168) | def _build_cache(self): class Md21Dataset (line 203) | class Md21Dataset(StereoDataset): method _prepare_data (line 205) | def _prepare_data(self): method _build_cache (line 215) | def _build_cache(self): class Md14Dataset (line 228) | class Md14Dataset(StereoDataset): method _prepare_data (line 230) | def _prepare_data(self): method _build_cache (line 241) | def _build_cache(self): class Md06Dataset (line 255) | class Md06Dataset(StereoDataset): method _prepare_data (line 257) | def _prepare_data(self): method _build_cache (line 267) | def _build_cache(self): class Md05Dataset (line 283) | class Md05Dataset(StereoDataset): method _prepare_data (line 285) | def _prepare_data(self): method _build_cache (line 295) | def _build_cache(self): class MdEval3Dataset (line 311) | class MdEval3Dataset(StereoDataset): method _prepare_data (line 313) | def _prepare_data(self): method _build_cache (line 332) | def _build_cache(self): method submission_save_pairname (line 344) | def submission_save_pairname(self, pairname, prediction, outdir, time): method finalize_submission (line 354) | def finalize_submission(self, outdir): class ETH3DLowResDataset (line 360) | class ETH3DLowResDataset(StereoDataset): method _prepare_data (line 362) | def _prepare_data(self): method _build_cache (line 373) | def _build_cache(self): method submission_save_pairname (line 384) | def submission_save_pairname(self, pairname, prediction, outdir, time): method finalize_submission (line 394) | def finalize_submission(self, outdir): class BoosterDataset (line 400) | class BoosterDataset(StereoDataset): method _prepare_data (line 402) | def _prepare_data(self): method _build_cache (line 413) | def _build_cache(self): class SpringDataset (line 424) | class SpringDataset(StereoDataset): method _prepare_data (line 426) | def _prepare_data(self): method _build_cache (line 436) | def _build_cache(self): method submission_save_pairname (line 449) | def submission_save_pairname(self, pairname, prediction, outdir, time): method finalize_submission (line 456) | def finalize_submission(self, outdir): class Kitti12Dataset (line 468) | class Kitti12Dataset(StereoDataset): method _prepare_data (line 470) | def _prepare_data(self): method _build_cache (line 480) | def _build_cache(self): method submission_save_pairname (line 487) | def submission_save_pairname(self, pairname, prediction, outdir, time): method finalize_submission (line 495) | def finalize_submission(self, outdir): class Kitti15Dataset (line 502) | class Kitti15Dataset(StereoDataset): method _prepare_data (line 504) | def _prepare_data(self): method _build_cache (line 514) | def _build_cache(self): method submission_save_pairname (line 523) | def submission_save_pairname(self, pairname, prediction, outdir, time): method finalize_submission (line 531) | def finalize_submission(self, outdir): function _read_img (line 541) | def _read_img(filename): function _read_booster_disp (line 546) | def _read_booster_disp(filename): function _read_png_disp (line 551) | def _read_png_disp(filename, coef=1.0): function _read_pfm_disp (line 557) | def _read_pfm_disp(filename): function _read_npy_disp (line 562) | def _read_npy_disp(filename): function _read_crestereo_disp (line 565) | def _read_crestereo_disp(filename): return _read_png_disp(filename, coef... function _read_middlebury20052006_disp (line 566) | def _read_middlebury20052006_disp(filename): return _read_png_disp(filen... function _read_kitti_disp (line 567) | def _read_kitti_disp(filename): return _read_png_disp(filename, coef=256.0) function _read_hdf5_disp (line 574) | def _read_hdf5_disp(filename): function _read_pfm (line 581) | def _read_pfm(file): function writePFM (line 618) | def writePFM(file, image, scale=1): function writeDsp5File (line 647) | def writeDsp5File(disp, filename): function vis_disparity (line 654) | def vis_disparity(disp, m=None, M=None): function get_train_dataset_stereo (line 664) | def get_train_dataset_stereo(dataset_str, augmentor=True, crop_size=None): function get_test_datasets_stereo (line 672) | def get_test_datasets_stereo(dataset_str): FILE: croco/stereoflow/engine.py function split_prediction_conf (line 18) | def split_prediction_conf(predictions, with_conf=False): function train_one_epoch (line 25) | def train_one_epoch(model: torch.nn.Module, criterion: torch.nn.Module, ... function validate_one_epoch (line 104) | def validate_one_epoch(model: torch.nn.Module, function _resize_img (line 163) | def _resize_img(img, new_size): function _resize_stereo_or_flow (line 165) | def _resize_stereo_or_flow(data, new_size): function tiled_pred (line 179) | def tiled_pred(model, criterion, img1, img2, gt, function _overlapping (line 267) | def _overlapping(total, window, overlap=0.5): function _crop (line 273) | def _crop(img, sy, sx): FILE: croco/stereoflow/test.py function get_args_parser (line 30) | def get_args_parser(): function _load_model_and_criterion (line 51) | def _load_model_and_criterion(model_path, do_load_metrics, device): function _save_batch (line 82) | def _save_batch(pred, gt, pairnames, dataset, task, save, outdir, time, ... function main (line 128) | def main(args): FILE: croco/stereoflow/train.py function get_args_parser (line 37) | def get_args_parser(): function main (line 85) | def main(args): FILE: croco/utils/misc.py class SmoothedValue (line 27) | class SmoothedValue(object): method __init__ (line 32) | def __init__(self, window_size=20, fmt=None): method update (line 40) | def update(self, value, n=1): method synchronize_between_processes (line 45) | def synchronize_between_processes(self): method median (line 59) | def median(self): method avg (line 64) | def avg(self): method global_avg (line 69) | def global_avg(self): method max (line 73) | def max(self): method value (line 77) | def value(self): method __str__ (line 80) | def __str__(self): class MetricLogger (line 89) | class MetricLogger(object): method __init__ (line 90) | def __init__(self, delimiter="\t"): method update (line 94) | def update(self, **kwargs): method __getattr__ (line 103) | def __getattr__(self, attr): method __str__ (line 111) | def __str__(self): method synchronize_between_processes (line 119) | def synchronize_between_processes(self): method add_meter (line 123) | def add_meter(self, name, meter): method log_every (line 126) | def log_every(self, iterable, print_freq, header=None, max_iter=None): function setup_for_distributed (line 176) | def setup_for_distributed(is_master): function is_dist_avail_and_initialized (line 193) | def is_dist_avail_and_initialized(): function get_world_size (line 201) | def get_world_size(): function get_rank (line 207) | def get_rank(): function is_main_process (line 213) | def is_main_process(): function save_on_master (line 217) | def save_on_master(*args, **kwargs): function init_distributed_mode (line 222) | def init_distributed_mode(args): class NativeScalerWithGradNormCount (line 247) | class NativeScalerWithGradNormCount: method __init__ (line 250) | def __init__(self, enabled=True): method __call__ (line 253) | def __call__(self, loss, optimizer, clip_grad=None, parameters=None, c... method state_dict (line 272) | def state_dict(self): method load_state_dict (line 275) | def load_state_dict(self, state_dict): function get_grad_norm_ (line 279) | def get_grad_norm_(parameters, norm_type: float = 2.0) -> torch.Tensor: function save_model (line 296) | def save_model(args, epoch, model_without_ddp, optimizer, loss_scaler, f... function load_model (line 312) | def load_model(args, model_without_ddp, optimizer, loss_scaler): function all_reduce_mean (line 335) | def all_reduce_mean(x): function _replace (line 345) | def _replace(text, src, tgt, rm=''): function filename (line 360) | def filename( obj ): function _get_num_layer_for_vit (line 372) | def _get_num_layer_for_vit(var_name, enc_depth, dec_depth): function get_parameter_groups (line 392) | def get_parameter_groups(model, weight_decay, layer_decay=1.0, skip_list... function adjust_learning_rate (line 452) | def adjust_learning_rate(optimizer, epoch, args): FILE: dust3r/datasets/__init__.py function get_data_loader (line 7) | def get_data_loader(dataset, batch_size, num_workers=8, shuffle=True, dr... FILE: dust3r/datasets/base/base_stereo_view_dataset.py class BaseStereoViewDataset (line 17) | class BaseStereoViewDataset (EasyDataset): method __init__ (line 29) | def __init__(self, *, # only keyword arguments method __len__ (line 46) | def __len__(self): method get_stats (line 49) | def get_stats(self): method __repr__ (line 52) | def __repr__(self): method _get_views (line 60) | def _get_views(self, idx, resolution, rng): method __getitem__ (line 63) | def __getitem__(self, idx): method _set_resolutions (line 120) | def _set_resolutions(self, resolutions): method _crop_resize_if_necessary (line 137) | def _crop_resize_if_necessary(self, image, depthmap, intrinsics, resol... function is_good_type (line 199) | def is_good_type(key, v): function view_name (line 209) | def view_name(view, batch_index=None): function transpose_to_landscape (line 217) | def transpose_to_landscape(view): FILE: dust3r/datasets/base/batched_sampler.py class BatchedRandomSampler (line 12) | class BatchedRandomSampler: method __init__ (line 22) | def __init__(self, dataset, batch_size, pool_size, world_size=1, rank=... method __len__ (line 35) | def __len__(self): method set_epoch (line 38) | def set_epoch(self, epoch): method __iter__ (line 41) | def __iter__(self): function round_by (line 75) | def round_by(total, multiple, up=False): FILE: dust3r/datasets/base/easy_dataset.py class EasyDataset (line 11) | class EasyDataset: method __add__ (line 22) | def __add__(self, other): method __rmul__ (line 25) | def __rmul__(self, factor): method __rmatmul__ (line 28) | def __rmatmul__(self, factor): method set_epoch (line 31) | def set_epoch(self, epoch): method make_sampler (line 34) | def make_sampler(self, batch_size, shuffle=True, world_size=1, rank=0,... class MulDataset (line 41) | class MulDataset (EasyDataset): method __init__ (line 46) | def __init__(self, multiplicator, dataset): method __len__ (line 51) | def __len__(self): method __repr__ (line 54) | def __repr__(self): method __getitem__ (line 57) | def __getitem__(self, idx): method _resolutions (line 65) | def _resolutions(self): class ResizedDataset (line 69) | class ResizedDataset (EasyDataset): method __init__ (line 74) | def __init__(self, new_size, dataset): method __len__ (line 79) | def __len__(self): method __repr__ (line 82) | def __repr__(self): method set_epoch (line 89) | def set_epoch(self, epoch): method __getitem__ (line 102) | def __getitem__(self, idx): method _resolutions (line 111) | def _resolutions(self): class CatDataset (line 115) | class CatDataset (EasyDataset): method __init__ (line 119) | def __init__(self, datasets): method __len__ (line 125) | def __len__(self): method __repr__ (line 128) | def __repr__(self): method set_epoch (line 132) | def set_epoch(self, epoch): method __getitem__ (line 136) | def __getitem__(self, idx): method _resolutions (line 153) | def _resolutions(self): FILE: dust3r/datasets/base/mast3r_base_stereo_view_dataset.py class MASt3RBaseStereoViewDataset (line 24) | class MASt3RBaseStereoViewDataset(BaseStereoViewDataset): method __init__ (line 25) | def __init__(self, *, # only keyword arguments method _swap_view_aug (line 59) | def _swap_view_aug(self, views): method _crop_resize_if_necessary (line 63) | def _crop_resize_if_necessary(self, image, depthmap, intrinsics, resol... method generate_crops_from_pair (line 109) | def generate_crops_from_pair(self, view1, view2, resolution, aug_crop_... method __getNewitem__ (line 215) | def __getNewitem__(self, idx): method __getitem__ (line 224) | def __getitem__(self, idx): function transpose_to_landscape_no_flip (line 433) | def transpose_to_landscape_no_flip(view, rng, revert=False): function transpose_to_landscape (line 447) | def transpose_to_landscape(view, revert=False): function rotate_90_single_view (line 483) | def rotate_90_single_view(view, k=1): function rotate_90 (line 525) | def rotate_90(views, k=1): function motion_blur (line 529) | def motion_blur(img): FILE: dust3r/datasets/utils/cropping.py class ImageList (line 19) | class ImageList: method __init__ (line 23) | def __init__(self, images): method __len__ (line 32) | def __len__(self): method to_pil (line 35) | def to_pil(self): method size (line 39) | def size(self): method resize (line 44) | def resize(self, *args, **kwargs): method crop (line 47) | def crop(self, *args, **kwargs): method _dispatch (line 50) | def _dispatch(self, func, *args, **kwargs): function rescale_image_depthmap (line 54) | def rescale_image_depthmap(image, depthmap, camera_intrinsics, output_re... function camera_matrix_of_crop (line 102) | def camera_matrix_of_crop(input_camera_matrix, input_resolution, output_... function crop_image_depthmap (line 122) | def crop_image_depthmap(image, depthmap, camera_intrinsics, crop_bbox, m... function bbox_from_intrinsics_in_out (line 152) | def bbox_from_intrinsics_in_out(input_camera_matrix, output_camera_matri... FILE: dust3r/datasets/utils/mast3r_cropping.py function reciprocal_1d (line 12) | def reciprocal_1d(corres_1_to_2, corres_2_to_1, ret_recip=False): function extract_correspondences_from_pts3d (line 21) | def extract_correspondences_from_pts3d(view1, view2, target_n_corres, rn... function reproject_view (line 83) | def reproject_view(pts3d, view2): function reproject (line 88) | def reproject(pts3d, K, world2cam, shape): function ravel_xy (line 100) | def ravel_xy(pos, shape): function unravel_xy (line 108) | def unravel_xy(pos, shape): function _rotation_origin_to_pt (line 113) | def _rotation_origin_to_pt(target): function _dotmv (line 125) | def _dotmv(Trf, pts, ncol=None, norm=False): function crop_to_homography (line 164) | def crop_to_homography(K, crop, target_size=None): function gen_random_crops (line 199) | def gen_random_crops(imsize, n_crops, resolution, aug_crop, rng=np.random): function in2d_rect (line 219) | def in2d_rect(corres, crops): FILE: dust3r/heads/__init__.py function head_factory (line 10) | def head_factory(head_type, output_mode, net, has_conf=False): FILE: dust3r/heads/dpt_head.py class DPTOutputAdapter_fix (line 26) | class DPTOutputAdapter_fix(DPTOutputAdapter): method init (line 32) | def init(self, dim_tokens_enc=768): method forward (line 40) | def forward(self, encoder_tokens: List[torch.Tensor], image_size=None): class PixelwiseTaskWithDPT (line 79) | class PixelwiseTaskWithDPT(nn.Module): method __init__ (line 82) | def __init__(self, *, n_cls_token=0, hooks_idx=None, dim_tokens=None, method forward (line 102) | def forward(self, x, img_info): FILE: dust3r/heads/linear_head.py class LinearPts3d (line 12) | class LinearPts3d (nn.Module): method __init__ (line 18) | def __init__(self, net, has_conf=False): method setup (line 27) | def setup(self, croconet): method forward (line 30) | def forward(self, decout, img_shape): FILE: dust3r/heads/postprocess.py function postprocess (line 10) | def postprocess(out, depth_mode, conf_mode): function postprocess_mask (line 21) | def postprocess_mask(out, depth_mode, conf_mode): function reg_dense_depth (line 29) | def reg_dense_depth(xyz, mode): function reg_dense_conf (line 56) | def reg_dense_conf(x, mode): function reg_dense_normal (line 68) | def reg_dense_normal(normal, mode=None): function reg_dense_kappa (line 93) | def reg_dense_kappa(kappa, mode): FILE: dust3r/image_pairs.py function make_pairs (line 11) | def make_pairs(imgs, scene_graph='complete', prefilter=None, symmetrize=... function sel (line 50) | def sel(x, kept): function _filter_edges_seq (line 59) | def _filter_edges_seq(edges, seq_dis_thr, cyclic=False): function filter_pairs_seq (line 73) | def filter_pairs_seq(pairs, seq_dis_thr, cyclic=False): function filter_edges_seq (line 79) | def filter_edges_seq(view1, view2, pred1, pred2, seq_dis_thr, cyclic=Fal... FILE: dust3r/inference.py function load_model (line 29) | def load_model(model_path, device): function _interleave_imgs (line 44) | def _interleave_imgs(img1, img2): function make_batch_symmetric (line 56) | def make_batch_symmetric(batch): function loss_of_one_batch (line 62) | def loss_of_one_batch(batch, model, criterion, device, symmetrize_batch=... function inference (line 153) | def inference(pairs, model, device, batch_size=8, verbose=True): function check_if_same_size (line 172) | def check_if_same_size(pairs): function get_pred_pts3d (line 178) | def get_pred_pts3d(gt, pred, use_pose=False): function find_opt_scaling (line 203) | def find_opt_scaling(gt_pts1, gt_pts2, pr_pts1, pr_pts2=None, fit_mode='... FILE: dust3r/losses.py function Sum (line 21) | def Sum(*losses_and_masks): class BaseCriterion (line 33) | class BaseCriterion(nn.Module): method __init__ (line 34) | def __init__(self, reduction='mean'): class LLoss (line 39) | class LLoss (BaseCriterion): method forward (line 43) | def forward(self, a, b): method distance (line 55) | def distance(self, a, b): class L21Loss (line 59) | class L21Loss (LLoss): method distance (line 62) | def distance(self, a, b): class Criterion (line 69) | class Criterion (nn.Module): method __init__ (line 70) | def __init__(self, criterion=None): method get_name (line 75) | def get_name(self): method with_reduction (line 78) | def with_reduction(self, mode='none'): class MultiLoss (line 87) | class MultiLoss (nn.Module): method __init__ (line 94) | def __init__(self): method compute_loss (line 99) | def compute_loss(self, *args, **kwargs): method get_name (line 102) | def get_name(self): method __mul__ (line 105) | def __mul__(self, alpha): method __add__ (line 112) | def __add__(self, loss2): method __repr__ (line 121) | def __repr__(self): method forward (line 129) | def forward(self, *args, **kwargs): class Regr3D (line 147) | class Regr3D (Criterion, MultiLoss): method __init__ (line 158) | def __init__(self, criterion, norm_mode='avg_dis', gt_scale=False): method get_all_pts3d (line 163) | def get_all_pts3d(self, gt1, gt2, pred1, pred2, dist_clip=None): method compute_loss (line 190) | def compute_loss(self, gt1, gt2, pred1, pred2, **kw): class ConfLoss (line 202) | class ConfLoss (MultiLoss): method __init__ (line 213) | def __init__(self, pixel_loss, alpha=1): method get_name (line 219) | def get_name(self): method get_conf_log (line 222) | def get_conf_log(self, x): method get_conf_mask (line 225) | def get_conf_mask(self, gt1): method compute_loss (line 237) | def compute_loss(self, gt1, gt2, pred1, pred2, **kw): class Regr3D_ShiftInv (line 261) | class Regr3D_ShiftInv (Regr3D): method get_all_pts3d (line 265) | def get_all_pts3d(self, gt1, gt2, pred1, pred2): class Regr3D_ScaleInv (line 286) | class Regr3D_ScaleInv (Regr3D): method get_all_pts3d (line 291) | def get_all_pts3d(self, gt1, gt2, pred1, pred2): class Regr3D_ScaleShiftInv (line 317) | class Regr3D_ScaleShiftInv (Regr3D_ScaleInv, Regr3D_ShiftInv): FILE: dust3r/model.py class AsymmetricCroCo3DStereo (line 25) | class AsymmetricCroCo3DStereo (CroCoNet): method __init__ (line 31) | def __init__(self, method _set_patch_embed (line 55) | def _set_patch_embed(self, img_size=224, patch_size=16, enc_embed_dim=... method load_state_dict (line 58) | def load_state_dict(self, ckpt, **kw): method set_freeze (line 70) | def set_freeze(self, freeze): # this is for use by downstream models method _set_prediction_head (line 79) | def _set_prediction_head(self, *args, **kwargs): method _encode_image (line 83) | def _encode_image(self, image, true_shape): method _encode_image_pairs (line 101) | def _encode_image_pairs(self, img1, img2, true_shape1, true_shape2): method _encode_symmetrized (line 112) | def _encode_symmetrized(self, view1, view2): method _decoder (line 131) | def _decoder(self, f1, pos1, f2, pos2): method _downstream_head (line 156) | def _downstream_head(self, head_num, decout, img_shape): method forward (line 162) | def forward(self, view1, view2): FILE: dust3r/optim_factory.py function adjust_learning_rate_by_lr (line 9) | def adjust_learning_rate_by_lr(optimizer, lr): FILE: dust3r/patch_embed.py function get_patch_embed (line 14) | def get_patch_embed(patch_embed_cls, img_size, patch_size, enc_embed_dim): class PatchEmbedDust3R (line 20) | class PatchEmbedDust3R(PatchEmbed): method forward (line 21) | def forward(self, x, **kw): class ManyAR_PatchEmbed (line 33) | class ManyAR_PatchEmbed (PatchEmbed): method __init__ (line 39) | def __init__(self, img_size=224, patch_size=16, in_chans=3, embed_dim=... method forward (line 43) | def forward(self, img, true_shape, atm_size=None): FILE: dust3r/post_process.py function estimate_focal_knowing_depth (line 12) | def estimate_focal_knowing_depth(pts3d, pp, focal_mode='median', min_foc... FILE: dust3r/utils/device.py function todevice (line 11) | def todevice(batch, device, callback=None, non_blocking=False): function to_numpy (line 42) | def to_numpy(x): return todevice(x, 'numpy') function to_cpu (line 43) | def to_cpu(x): return todevice(x, 'cpu') function to_cuda (line 44) | def to_cuda(x): return todevice(x, 'cuda') function collate_with_cat (line 47) | def collate_with_cat(whatever, lists=False): function listify (line 75) | def listify(elems): FILE: dust3r/utils/geometry.py function xy_grid (line 15) | def xy_grid(W, H, device=None, origin=(0, 0), unsqueeze=None, cat_dim=-1... function geotrf (line 40) | def geotrf(Trf, pts, ncol=None, norm=False): function inv (line 104) | def inv(mat): function depthmap_to_pts3d (line 114) | def depthmap_to_pts3d(depth, pseudo_focal, pp=None, **_): function depthmap_to_camera_coordinates (line 165) | def depthmap_to_camera_coordinates(depthmap, camera_intrinsics, pseudo_f... function depthmap_to_absolute_camera_coordinates (line 200) | def depthmap_to_absolute_camera_coordinates(depthmap, camera_intrinsics,... function colmap_to_opencv_intrinsics (line 226) | def colmap_to_opencv_intrinsics(K): function opencv_to_colmap_intrinsics (line 239) | def opencv_to_colmap_intrinsics(K): function normalize_pointcloud (line 252) | def normalize_pointcloud(pts1, pts2, norm_mode='avg_dis', valid1=None, v... function get_joint_pointcloud_depth (line 316) | def get_joint_pointcloud_depth(z1, z2, valid_mask1, valid_mask2=None, qu... function get_joint_pointcloud_center_scale (line 331) | def get_joint_pointcloud_center_scale(pts1, pts2, valid_mask1=None, vali... function find_reciprocal_matches (line 348) | def find_reciprocal_matches(P1, P2): function get_med_dist_between_poses (line 367) | def get_med_dist_between_poses(poses): FILE: dust3r/utils/image.py function get_colormap (line 34) | def get_colormap(n): function exr2hdr (line 51) | def exr2hdr(exrpath): function inverse_normalize (line 72) | def inverse_normalize(tensor, mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5)): function img_to_arr (line 82) | def img_to_arr( img ): function imread_cv2 (line 87) | def imread_cv2(path, options=cv2.IMREAD_COLOR): function imread_pil (line 101) | def imread_pil(path): function rgb (line 109) | def rgb(ftensor, true_shape=None): function _resize_pil_image (line 128) | def _resize_pil_image(img, long_edge_size): function load_images (line 138) | def load_images(folder_or_list, size, square_ok=True, verbose=True): function normal2color (line 203) | def normal2color(normal_map): function color2normal (line 211) | def color2normal(normal_color): function depth2pcd_cam (line 219) | def depth2pcd_cam(depth, intrinsics_matrix, depth_valid_mask=None): function pcd2normal_numpy (line 233) | def pcd2normal_numpy(xyz): function colorize (line 252) | def colorize(value, vmin=None, vmax=None, cmap='gray_r', invalid_val=-99... FILE: dust3r/utils/misc.py function fill_default_args (line 10) | def fill_default_args(kwargs, func): function freeze_all_params (line 22) | def freeze_all_params(modules): function is_symmetrized (line 32) | def is_symmetrized(gt1, gt2): function flip (line 43) | def flip(tensor): function interleave (line 48) | def interleave(tensor1, tensor2): function transpose_to_landscape (line 54) | def transpose_to_landscape(head, activate=True): function transposed (line 99) | def transposed(dic): function invalid_to_nans (line 103) | def invalid_to_nans(arr, valid_mask, ndim=999): function invalid_to_zeros (line 112) | def invalid_to_zeros(arr, valid_mask, ndim=999): FILE: dust3r/utils/parallel.py function parallel_threads (line 12) | def parallel_threads(function, args, workers=0, star_args=False, kw_args... function parallel_processes (line 62) | def parallel_processes(*args, **kwargs): function starcall (line 70) | def starcall(args): function starstarcall (line 76) | def starstarcall(args): FILE: dust3r/utils/read_write_model.py class Image (line 49) | class Image(BaseImage): method qvec2rotmat (line 50) | def qvec2rotmat(self): function read_next_bytes (line 73) | def read_next_bytes(fid, num_bytes, format_char_sequence, endian_charact... function write_next_bytes (line 85) | def write_next_bytes(fid, data, format_char_sequence, endian_character="... function read_cameras_text (line 101) | def read_cameras_text(path): function read_cameras_binary (line 127) | def read_cameras_binary(path_to_model_file): function write_cameras_text (line 156) | def write_cameras_text(cameras, path): function write_cameras_binary (line 173) | def write_cameras_binary(cameras, path_to_model_file): function read_images_text (line 193) | def read_images_text(path): function read_images_binary (line 224) | def read_images_binary(path_to_model_file): function write_images_text (line 259) | def write_images_text(images, path): function write_images_binary (line 287) | def write_images_binary(images, path_to_model_file): function read_points3D_text (line 308) | def read_points3D_text(path): function read_points3D_binary (line 335) | def read_points3D_binary(path_to_model_file): function write_points3D_text (line 365) | def write_points3D_text(points3D, path): function write_points3D_binary (line 390) | def write_points3D_binary(points3D, path_to_model_file): function detect_model_format (line 409) | def detect_model_format(path, ext): function read_model (line 419) | def read_model(path, ext=""): function write_model (line 441) | def write_model(cameras, images, points3D, path, ext=".bin"): function qvec2rotmat (line 453) | def qvec2rotmat(qvec): function rotmat2qvec (line 466) | def rotmat2qvec(R): function main (line 480) | def main(): FILE: dust3r/viz.py function cat_3d (line 22) | def cat_3d(vecs): function show_raw_pointcloud (line 28) | def show_raw_pointcloud(pts3d, colors, point_size=2): function pts3d_to_trimesh (line 37) | def pts3d_to_trimesh(img, pts3d, valid=None): function cat_meshes (line 77) | def cat_meshes(meshes): function show_duster_pairs (line 89) | def show_duster_pairs(view1, view2, pred1, pred2): function auto_cam_size (line 114) | def auto_cam_size(im_poses): class SceneViz (line 118) | class SceneViz: method __init__ (line 119) | def __init__(self): method add_pointcloud (line 122) | def add_pointcloud(self, pts3d, color, mask=None): method add_camera (line 142) | def add_camera(self, pose_c2w, focal=None, color=(0, 0, 0), image=None... method add_cameras (line 147) | def add_cameras(self, poses, focals=None, images=None, imsizes=None, c... method show (line 154) | def show(self, point_size=2): function show_raw_pointcloud_with_cams (line 158) | def show_raw_pointcloud_with_cams(imgs, pts3d, mask, focals, cams2world, function add_scene_cam (line 192) | def add_scene_cam(scene, pose_c2w, edge_color, image=None, focal=None, i... function cat (line 261) | def cat(a, b): function uint8 (line 275) | def uint8(colors): function segment_sky (line 284) | def segment_sky(image): FILE: infer/demo_online.py function resize_image_max_size (line 40) | def resize_image_max_size(image, image_size): function get_two_view_recon (line 50) | def get_two_view_recon(output, one_image_flag): function get_two_view_prediction (line 86) | def get_two_view_prediction(filelist, out_dir): function get_sequence_prediction (line 131) | def get_sequence_prediction(filelist, out_dir): function handle_uploads (line 222) | def handle_uploads(input_video, input_images): function update_gallery_on_upload (line 293) | def update_gallery_on_upload(input_video, input_images): function gradio_demo (line 304) | def gradio_demo( function clear_fields (line 380) | def clear_fields(): function update_log (line 386) | def update_log(): function update_visualization (line 392) | def update_visualization( function main_demo (line 429) | def main_demo(args): function get_args (line 634) | def get_args(): FILE: infer/demo_utils.py class GlomapRecon (line 30) | class GlomapRecon: method __init__ (line 31) | def __init__(self, world_to_cam, intrinsics, points3d, imgs): class GlomapReconState (line 37) | class GlomapReconState: method __init__ (line 38) | def __init__(self, glomap_recon, should_delete=False, cache_dir=None, ... method __del__ (line 44) | def __del__(self): function add_scene_cam (line 54) | def add_scene_cam(scene, pose_c2w, edge_color, image=None, focal=None, i... function get_3D_model_from_scene (line 122) | def get_3D_model_from_scene(scene_state, transparent_cams=False, cam_siz... function get_glb_from_recon_scene (line 161) | def get_glb_from_recon_scene(reconstruction_path, img_dir, function predictions_to_glb (line 207) | def predictions_to_glb( function integrate_camera_into_scene (line 363) | def integrate_camera_into_scene( function apply_scene_alignment (line 413) | def apply_scene_alignment(scene_3d: trimesh.Scene, extrinsics_matrices: ... function get_opengl_conversion_matrix (line 437) | def get_opengl_conversion_matrix() -> np.ndarray: function transform_points (line 454) | def transform_points(transformation: np.ndarray, points: np.ndarray, dim... function compute_camera_faces (line 479) | def compute_camera_faces(cone_shape: trimesh.Trimesh) -> np.ndarray: function segment_sky (line 515) | def segment_sky(image_path, onnx_session, mask_filename=None): function run_skyseg (line 546) | def run_skyseg(onnx_session, input_size, image): function download_file_from_url (line 586) | def download_file_from_url(url, filename): function img2video (line 609) | def img2video(img_paths, video_path, fps=1): FILE: infer/dens3r_recon.py function infer_pair_matches_batch (line 28) | def infer_pair_matches_batch(imgs_pairs, model, img_dir, function recover_coordinate (line 85) | def recover_coordinate(match_info, size_info): function get_img_pairs (line 94) | def get_img_pairs(img_dir, mode="sequential", window_size=20): function save_img_pairs_txt (line 128) | def save_img_pairs_txt(img_pairs, sfm_dir): function colmap_recon (line 134) | def colmap_recon(imgs_keypts_dict : dict, matches_dict : dict, img_name_... function colmap_triangulation (line 168) | def colmap_triangulation(imgs_keypts_dict : dict, matches_dict : dict, i... function mast3r_match_tracking (line 197) | def mast3r_match_tracking(model : Any, function save_dict (line 314) | def save_dict(dict_to_save, filepath): function read_dict (line 320) | def read_dict(filepath): function manhattan_alignment (line 327) | def manhattan_alignment(image_dir, sfm_dir, output_dir): function Dens3RReconstruction (line 343) | def Dens3RReconstruction(path_or_model, img_dir, sfm_dir, function parse_args (line 379) | def parse_args(): function main (line 408) | def main(): FILE: infer/eval_scripts/eval_depth.py function normalized_depth_scale_and_shift (line 21) | def normalized_depth_scale_and_shift(prediction, target, mask): class RunningAverage (line 60) | class RunningAverage: method __init__ (line 61) | def __init__(self): method append (line 65) | def append(self, value): method get_value (line 69) | def get_value(self): class RunningAverageDict (line 73) | class RunningAverageDict: method __init__ (line 75) | def __init__(self): method update (line 78) | def update(self, new_dict): method get_value (line 90) | def get_value(self): function compute_errors (line 96) | def compute_errors(gt, pred): function compute_metrics (line 141) | def compute_metrics(gt_depth, pred, save_vis=None, interpolate=False, function colorize (line 177) | def colorize(value, vmin=None, vmax=None, cmap='gray_r', invalid_val=-99... function load_img_pairs (line 231) | def load_img_pairs(dataset_path=None, img_path_list=None, size=1024, pai... function get_pred (line 251) | def get_pred(model_path, img_list, save_path, device="cuda", size=1024, ... function eval_nyudv2 (line 274) | def eval_nyudv2(data_dir, save_dir, model_path, align_depth=True): function parse_args (line 337) | def parse_args(): function main (line 347) | def main(): FILE: infer/eval_scripts/eval_matching.py function recover_coordinate (line 18) | def recover_coordinate(match_info, size_info): function extract_match (line 28) | def extract_match(output, device="cuda", conf_thr=1.001, pixel_tol=5, su... function load_img_dict (line 62) | def load_img_dict(pair01_list, pair02_list, size=1024): function load_img_pairs (line 70) | def load_img_pairs(pair01_list, pair02_list, size=1024): function get_pred (line 80) | def get_pred(model_path, pair01_list, pair02_list, save_path, function read_gim_txt (line 103) | def read_gim_txt(path): function extract_gim_pairs (line 120) | def extract_gim_pairs(txt_list, sub="blendedmvs"): function eval_gim (line 133) | def eval_gim(model_path, data_dir, pred_save_dir, sub="blendedmvs"): function eval_scannet_1500 (line 194) | def eval_scannet_1500(model_path, data_dir, pred_save_dir): function parse_args (line 257) | def parse_args(): function main (line 267) | def main(): FILE: infer/eval_scripts/eval_normal.py function dot (line 20) | def dot(x, y): function length (line 35) | def length(x, eps=1e-20): function safe_normalize (line 50) | def safe_normalize(x, eps=1e-20): function compute_metric (line 60) | def compute_metric(gt_path, pred_path, save_vis_dir=None, data_type=None): function compute_matric_list (line 113) | def compute_matric_list(img_list, normal_gt_list, gt_dir, save_path, dat... function load_img_pairs (line 144) | def load_img_pairs(dataset_path=None, img_path_list=None, size=1024, pai... function get_pred (line 165) | def get_pred(model_path, dataset_path, save_path, device="cuda", size=10... function eval_nyudv2 (line 194) | def eval_nyudv2(data_dir, save_dir, model_path): function parse_args (line 255) | def parse_args(): function main (line 265) | def main(): FILE: infer/eval_scripts/matching_metrics.py function relative_pose_error (line 31) | def relative_pose_error(T_0to1, R, t, ignore_gt_t_thr=0.0): function symmetric_epipolar_distance (line 49) | def symmetric_epipolar_distance(pts0, pts1, E, K0, K1): function compute_symmetrical_epipolar_errors (line 69) | def compute_symmetrical_epipolar_errors(data): function estimate_pose (line 91) | def estimate_pose(kpts0, kpts1, K0, K1, thresh, conf=0.99999, return_con... function compute_pose_errors (line 122) | def compute_pose_errors(data, pixel_thr=0.5, conf=0.99999): function compute_pose_errors_numpy (line 158) | def compute_pose_errors_numpy(data, pixel_thr=0.5, conf=0.99999): function cal_geometry (line 191) | def cal_geometry(data, save_vis_path=None): function error_auc (line 234) | def error_auc(errors, thresholds=[5, 10, 20]): function epidist_prec (line 254) | def epidist_prec(errors, thresholds, ret_dict=False): function aggregate_metrics (line 268) | def aggregate_metrics(metrics, epi_err_thr=5e-4): function fast_make_matching_figure (line 292) | def fast_make_matching_figure(data): function fast_make_matching_overlay (line 337) | def fast_make_matching_overlay(data): function wrap_images (line 380) | def wrap_images(img0, img1, geo_info, geom_type): function plot_images (line 419) | def plot_images(imgs, titles=None, cmaps="gray", dpi=100, size=5, pad=0.5): function fig2im (line 453) | def fig2im(fig): function compute_geom (line 462) | def compute_geom(data, function vis_matching (line 512) | def vis_matching(img0, img1, kpts0, kpts1, sample_matching_num=None): FILE: infer/infer_normal_pts3d.py function make_pairs (line 25) | def make_pairs(img_list, mode="sequence", seq_len=5): function rescale_to_orig (line 60) | def rescale_to_orig(data, size_info): function get_avg_normal (line 73) | def get_avg_normal(normal_list, delete_bad_normal=True): function rotate_recover (line 90) | def rotate_recover(output): function infer_imgs (line 114) | def infer_imgs(model, img_dir, save_dir, function parse_args (line 208) | def parse_args(): function main (line 220) | def main(): FILE: mast3r/catmlp_dpt_head.py function reg_desc (line 23) | def reg_desc(desc, mode): function postprocess (line 31) | def postprocess(out, depth_mode, conf_mode, desc_dim=None, desc_mode='no... function postprocess_normal (line 49) | def postprocess_normal(out, normal_mode): function postprocess_semantic (line 56) | def postprocess_semantic(out, normal_mode): class Cat_MLP_LocalFeatures_DPT_Pts3d (line 62) | class Cat_MLP_LocalFeatures_DPT_Pts3d(PixelwiseTaskWithDPT): method __init__ (line 67) | def __init__(self, net, has_conf=False, local_feat_dim=16, hidden_dim_... method forward (line 91) | def forward(self, decout, img_shape): function mast3r_head_factory (line 118) | def mast3r_head_factory(head_type, output_mode, net, has_conf=False, has... FILE: mast3r/colmap_utils/database.py function image_ids_to_pair_id (line 115) | def image_ids_to_pair_id(image_id1, image_id2): function pair_id_to_image_ids (line 121) | def pair_id_to_image_ids(pair_id): function array_to_blob (line 127) | def array_to_blob(array): function blob_to_array (line 134) | def blob_to_array(blob, dtype, shape=(-1,)): class COLMAPDatabase (line 141) | class COLMAPDatabase(sqlite3.Connection): method connect (line 144) | def connect(database_path): method __init__ (line 148) | def __init__(self, *args, **kwargs): method add_camera (line 166) | def add_camera(self, model, width, height, params, method add_image (line 175) | def add_image(self, name, camera_id, method add_keypoints (line 184) | def add_keypoints(self, image_id, keypoints): method add_descriptors (line 193) | def add_descriptors(self, image_id, descriptors): method add_matches (line 199) | def add_matches(self, image_id1, image_id2, matches): method add_two_view_geometry (line 212) | def add_two_view_geometry(self, image_id1, image_id2, matches, function example_usage (line 236) | def example_usage(): FILE: mast3r/colmap_utils/database_utils.py function create_empty_db (line 23) | def create_empty_db(database_path: Path): function import_images (line 34) | def import_images(image_dir: Path, function get_image_ids (line 50) | def get_image_ids(database_path: Path) -> Dict[str, int]: function get_keypoints (line 58) | def get_keypoints(path: Path, name: str, function import_features (line 68) | def import_features(image_ids: Dict[str, int], function find_pair (line 82) | def find_pair(hfile: h5py.File, name0: str, name1: str): function get_matches (line 104) | def get_matches(path: Path, name0: str, name1: str) -> Tuple[np.ndarray]: function import_matches (line 116) | def import_matches(image_ids: Dict[str, int], class OutputCapture (line 146) | class OutputCapture: method __init__ (line 147) | def __init__(self, verbose: bool): method __enter__ (line 150) | def __enter__(self): method __exit__ (line 155) | def __exit__(self, exc_type, *args): function estimation_and_geometric_verification (line 162) | def estimation_and_geometric_verification(database_path: Path, function run_reconstruction (line 171) | def run_reconstruction(sfm_dir: Path, function glomap_run_mapper (line 206) | def glomap_run_mapper(glomap_bin, colmap_db_path, recon_path, image_root... function run_triangulation (line 229) | def run_triangulation(model_path: Path, function create_db_from_model (line 246) | def create_db_from_model(reconstruction: pycolmap.Reconstruction, function import_keypts (line 266) | def import_keypts(database_path: Path, function import_mts (line 283) | def import_mts(database_path: Path, FILE: mast3r/fast_nn.py function bruteforce_reciprocal_nns (line 17) | def bruteforce_reciprocal_nns(A, B, device='cuda', block_size=None, dist... class cdistMatcher (line 73) | class cdistMatcher: method __init__ (line 74) | def __init__(self, db_pts, device='cuda'): method query (line 78) | def query(self, queries, k=1, **kw): function merge_corres (line 87) | def merge_corres(idx1, idx2, shape1=None, shape2=None, ret_xy=True, ret_... function fast_reciprocal_NNs (line 109) | def fast_reciprocal_NNs(pts1, pts2, subsample_or_initxy1=8, ret_xy=True,... function extract_correspondences_nonsym (line 191) | def extract_correspondences_nonsym(A, B, confA, confB, subsample=8, devi... FILE: mast3r/image_pairs.py function make_pairs (line 11) | def make_pairs(imgs, scene_graph='complete', prefilter=None, symmetrize=... function sel (line 82) | def sel(x, kept): function _filter_edges_seq (line 91) | def _filter_edges_seq(edges, seq_dis_thr, cyclic=False): function filter_pairs_seq (line 105) | def filter_pairs_seq(pairs, seq_dis_thr, cyclic=False): function filter_edges_seq (line 111) | def filter_edges_seq(view1, view2, pred1, pred2, seq_dis_thr, cyclic=Fal... FILE: mast3r/losses.py function apply_log_to_norm (line 26) | def apply_log_to_norm(xyz): function normalized_depth_scale_and_shift (line 32) | def normalized_depth_scale_and_shift(prediction, target, mask): class Regr3D (line 74) | class Regr3D (Regr3D_dust3r): method __init__ (line 75) | def __init__(self, criterion, norm_mode='avg_dis', gt_scale=False, opt... method get_all_pts3d (line 93) | def get_all_pts3d(self, gt1, gt2, pred1, pred2, dist_clip=None): method get_all_pts3d_align (line 157) | def get_all_pts3d_align(self, gt1, gt2, pred1, pred2): method compute_loss (line 195) | def compute_loss(self, gt1, gt2, pred1, pred2, **kw): class AlignDepthLoss (line 236) | class AlignDepthLoss(MultiLoss): method __init__ (line 237) | def __init__(self, align_pts): method get_name (line 241) | def get_name(self): method compute_align_depth_loss (line 244) | def compute_align_depth_loss(self, gt1, gt2, pred1, pred2): method compute_loss (line 283) | def compute_loss(self, gt1, gt2, pred1, pred2, **kw): class RawDepthLoss (line 294) | class RawDepthLoss(MultiLoss): method __init__ (line 295) | def __init__(self): method get_name (line 298) | def get_name(self): method compute_depth_loss (line 301) | def compute_depth_loss(self, gt1, gt2, pred1, pred2): method compute_loss (line 339) | def compute_loss(self, gt1, gt2, pred1, pred2, **kw): class SegmentationLoss (line 349) | class SegmentationLoss(MultiLoss): method __init__ (line 350) | def __init__(self): method get_name (line 354) | def get_name(self): method compute_loss (line 357) | def compute_loss(self, gt1, gt2, pred1, pred2, **kw): class Regr3D_ShiftInv (line 371) | class Regr3D_ShiftInv (Regr3D): method get_all_pts3d (line 375) | def get_all_pts3d(self, gt1, gt2, pred1, pred2): class Regr3D_ScaleInv (line 396) | class Regr3D_ScaleInv (Regr3D): method get_all_pts3d (line 401) | def get_all_pts3d(self, gt1, gt2, pred1, pred2): class Regr3D_ScaleShiftInv (line 426) | class Regr3D_ScaleShiftInv (Regr3D_ScaleInv, Regr3D_ShiftInv): function get_similarities (line 431) | def get_similarities(desc1, desc2, euc=False): class MatchingCriterion (line 441) | class MatchingCriterion(BaseCriterion): method __init__ (line 442) | def __init__(self, reduction='mean', fp=torch.float32): method forward (line 446) | def forward(self, a, b, valid_matches=None, euc=False): method loss (line 462) | def loss(self, a, b, valid_matches=None): class InfoNCE (line 466) | class InfoNCE(MatchingCriterion): method __init__ (line 467) | def __init__(self, temperature=0.07, eps=1e-8, mode='all', **kwargs): method loss (line 474) | def loss(self, desc1, desc2, valid_matches=None, euc=False): class APLoss (line 505) | class APLoss (MatchingCriterion): method __init__ (line 515) | def __init__(self, nq='torch', min=0, max=1, euc=False, **kw): method compute_true_AP_sklearn (line 526) | def compute_true_AP_sklearn(scores, labels): method compute_true_AP_torch (line 542) | def compute_true_AP_torch(scores, labels): method loss (line 567) | def loss(self, desc1, desc2, valid_matches=None, euc=False): # if mat... class MatchingLoss (line 584) | class MatchingLoss (Criterion, MultiLoss): method __init__ (line 590) | def __init__(self, criterion, withconf=False, use_pts3d=False, negativ... method add_negatives (line 597) | def add_negatives(self, outdesc2, desc2, batchid, x2, y2): method get_confs (line 607) | def get_confs(self, pred1, pred2, sel1, sel2): method get_descs (line 619) | def get_descs(self, pred1, pred2): method get_matching_descs (line 626) | def get_matching_descs(self, gt1, gt2, pred1, pred2, **kw): method blockwise_criterion (line 649) | def blockwise_criterion(self, descs1, descs2, confs1, confs2, valid_ma... method compute_loss (line 686) | def compute_loss(self, gt1, gt2, pred1, pred2, **kw): class NormalMetric (line 698) | class NormalMetric(MultiLoss): method __init__ (line 699) | def __init__(self): method get_name (line 702) | def get_name(self): method compute_loss (line 705) | def compute_loss(self, gt1, gt2, pred1, pred2, **kw): class KappaNormalLoss (line 740) | class KappaNormalLoss(MultiLoss): method __init__ (line 741) | def __init__(self): method get_name (line 744) | def get_name(self): method compute_loss (line 747) | def compute_loss(self, gt1, gt2, pred1, pred2, **kw): class ABSNormalLoss (line 798) | class ABSNormalLoss(MultiLoss): method __init__ (line 799) | def __init__(self): method get_name (line 802) | def get_name(self): method compute_loss (line 805) | def compute_loss(self, gt1, gt2, pred1, pred2, **kw): class SSIMNormalLoss (line 839) | class SSIMNormalLoss(MultiLoss): method __init__ (line 840) | def __init__(self): method get_name (line 844) | def get_name(self): method compute_loss (line 847) | def compute_loss(self, gt1, gt2, pred1, pred2, **kw): class ABSNormalFromDepthLoss (line 890) | class ABSNormalFromDepthLoss(MultiLoss): method __init__ (line 891) | def __init__(self): method get_name (line 894) | def get_name(self): method pcd2normal_torch (line 897) | def pcd2normal_torch(self, xyz): method compute_loss (line 919) | def compute_loss(self, gt1, gt2, pred1, pred2, **kw): class ConfMatchingLoss (line 957) | class ConfMatchingLoss(ConfLoss): method __init__ (line 962) | def __init__(self, pixel_loss, alpha=1., confmode='prod', neg_conf_los... method aggregate_confs (line 968) | def aggregate_confs(self, confs1, confs2): # get the confidences resu... method compute_loss (line 977) | def compute_loss(self, gt1, gt2, pred1, pred2, **kw): FILE: mast3r/model.py function load_model (line 27) | def load_model(model_path, device, verbose=True): class AsymmetricMASt3R (line 46) | class AsymmetricMASt3R(AsymmetricCroCo3DStereo): method __init__ (line 47) | def __init__(self, desc_mode=('norm'), desc_conf=False, desc_conf_mode... method freeze_encoder_decoder (line 58) | def freeze_encoder_decoder(self): method freeze_matching_net (line 66) | def freeze_matching_net(self): method load_state_dict (line 71) | def load_state_dict(self, ckpt, **kw): method from_pretrained (line 77) | def from_pretrained(cls, pretrained_model_name_or_path, **kw): method set_downstream_head (line 83) | def set_downstream_head(self, output_mode, head_type, landscape_only, ... FILE: mast3r/retrieval/graph.py function farthest_point_sampling (line 10) | def farthest_point_sampling(dist, N=None, dist_thresh=None): function make_pairs_fps (line 42) | def make_pairs_fps(sim_mat, Na=20, tokK=1, dist_thresh=None): FILE: mast3r/retrieval/model.py function pcawhitenlearn_shrinkage (line 21) | def pcawhitenlearn_shrinkage(X, s=1.0): class Dust3rInputFromImageList (line 41) | class Dust3rInputFromImageList(torch.utils.data.Dataset): method __init__ (line 42) | def __init__(self, image_list, imsize=512): method __len__ (line 48) | def __len__(self): method __getitem__ (line 51) | def __getitem__(self, index): class Whitener (line 55) | class Whitener(nn.Module): method __init__ (line 56) | def __init__(self, dim, l2norm=None): method forward (line 62) | def forward(self, x): function weighted_spoc (line 79) | def weighted_spoc(feat, attn): function how_select_local (line 88) | def how_select_local(feat, attn, nfeat): class RetrievalModel (line 107) | class RetrievalModel(nn.Module): method __init__ (line 108) | def __init__(self, backbone, freeze_backbone=1, prewhiten=None, hdims=... method build_projector (line 144) | def build_projector(self, hdims, residual): method state_dict (line 159) | def state_dict(self, *args, destination=None, prefix='', keep_vars=Fal... method reinitialize_whitening (line 165) | def reinitialize_whitening(self, epoch, train_dataset, nimgs=5000, log... method extract_features_and_attention (line 221) | def extract_features_and_attention(self, x): method forward_local (line 230) | def forward_local(self, x): method forward_global (line 234) | def forward_global(self, x): method forward (line 238) | def forward(self, x): function identity (line 242) | def identity(x): # to avoid Can't pickle local object 'extract_local_fe... function extract_local_features (line 247) | def extract_local_features(model, images, imsize, seed=0, tocpu=False, m... FILE: mast3r/retrieval/processor.py class FaissCpuL2Index (line 22) | class FaissCpuL2Index(asmk.index.FaissL2Index): method __init__ (line 23) | def __init__(self, gpu_id): method _faiss_index_flat (line 27) | def _faiss_index_flat(self, dim): function get_args_parser (line 36) | def get_args_parser(): function get_impaths (line 46) | def get_impaths(imlistfile): function get_impaths_from_imdir (line 53) | def get_impaths_from_imdir(imdir, extensions=['png', 'jpg', 'PNG', 'JPG']): function get_impaths_from_imdir_or_imlistfile (line 59) | def get_impaths_from_imdir_or_imlistfile(input_imdir_or_imlistfile): class Retriever (line 66) | class Retriever(object): method __init__ (line 67) | def __init__(self, modelname, backbone=None, device='cuda'): method __call__ (line 102) | def __call__(self, input_imdir_or_imlistfile, outfile=None): FILE: mast3r/ssim.py function gaussian (line 8) | def gaussian(window_size, sigma): function create_window (line 12) | def create_window(window_size, channel): function _ssim (line 18) | def _ssim(img1, img2, window, window_size, channel, size_average = True): class SSIM (line 40) | class SSIM(torch.nn.Module): method __init__ (line 41) | def __init__(self, window_size = 11, size_average = True): method forward (line 48) | def forward(self, img1, img2): function ssim (line 67) | def ssim(img1, img2, window_size = 11, size_average = True): FILE: mast3r/utils/coarse_to_fine.py function crop_tag (line 10) | def crop_tag(cell): function crop_slice (line 14) | def crop_slice(cell): function _start_pos (line 18) | def _start_pos(total_size, win_size, overlap): function multiple_of_16 (line 29) | def multiple_of_16(x): function _make_overlapping_grid (line 33) | def _make_overlapping_grid(H, W, size, overlap): function _cell_size (line 43) | def _cell_size(cell2): function _norm_windows (line 50) | def _norm_windows(cell2, H2, W2, forced_resolution=None): function _weight_pixels (line 91) | def _weight_pixels(cell, pix, assigned, gauss_var=2): function pos2d_in_rect (line 104) | def pos2d_in_rect(p1, cell1): function _score_cell (line 111) | def _score_cell(cell1, H2, W2, p1, p2, min_corres=10, forced_resolution=... function greedy_selection (line 156) | def greedy_selection(corres_weights, target=0.9): function select_pairs_of_crops (line 184) | def select_pairs_of_crops(img_q, img_b, pos2d_in_query, pos2d_in_ref, ma... FILE: mast3r/utils/collate.py function cat_collate_tensor_fn (line 14) | def cat_collate_tensor_fn(batch, *, collate_fn_map): function cat_collate_list_fn (line 18) | def cat_collate_list_fn(batch, *, collate_fn_map: Optional[Dict[Union[Ty... function cat_collate (line 28) | def cat_collate(batch, *, collate_fn_map: Optional[Dict[Union[Type, Tupl... FILE: mast3r/utils/misc.py function mkdir_for (line 11) | def mkdir_for(f): function hash_md5 (line 16) | def hash_md5(s):