Hugging Face's logo

Spaces:
HorizonRobotics
/
3D-Fixer
Running on Zero

App Files Community
3D-Fixer
File size: 3,320 Bytes
afea36f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
# This file is modified from TRELLIS:
# https://github.com/microsoft/TRELLIS
# Original license: MIT
# Copyright (c) the TRELLIS authors
# Modifications Copyright (c) 2026 Ze-Xin Yin, Robot labs of Horizon Robotics, and D-Robotics.

import importlib

__attributes = {
    'SparseStructure': 'sparse_structure',
    
    'SparseFeat2Render': 'sparse_feat2render',
    'SLat2Render':'structured_latent2render',
    'Slat2RenderGeo':'structured_latent2render',
    
    'SparseStructureLatent': 'sparse_structure_latent',
    'TextConditionedSparseStructureLatent': 'sparse_structure_latent',
    'ImageConditionedSparseStructureLatent': 'sparse_structure_latent',
    
    'SLat': 'structured_latent',
    'TextConditionedSLat': 'structured_latent',
    'ImageConditionedSLat': 'structured_latent',

    'ImageConditionedSparseStructureLatentRandRot': 'sparse_structure_latent_random_rot',
    'ImageConditionedSLatRandRot': 'structured_latent_random_rot',
    'SparseFeat2RenderRandRot': 'sparse_feat2render_random_rot',
    'Slat2RenderGeoRandRot': 'structured_latent2render_random_rot',

    'ObjectImageConditionedSparseStructureVoxel': 'scene_sparse_structure_latent_obj_pretrain',
    'SceneImageConditionedVoxel': 'scene_sparse_structure_latent',
    'SceneConditionedSLat': 'scene_structured_latent',
}

__submodules = []

__all__ = list(__attributes.keys()) + __submodules

def __getattr__(name):
    if name not in globals():
        if name in __attributes:
            module_name = __attributes[name]
            module = importlib.import_module(f".{module_name}", __name__)
            globals()[name] = getattr(module, name)
        elif name in __submodules:
            module = importlib.import_module(f".{name}", __name__)
            globals()[name] = module
        else:
            raise AttributeError(f"module {__name__} has no attribute {name}")
    return globals()[name]


# For Pylance
if __name__ == '__main__':
    from .sparse_structure import SparseStructure
    
    from .sparse_feat2render import SparseFeat2Render
    from .structured_latent2render import (
        SLat2Render,
        Slat2RenderGeo,
    )
    
    from .sparse_structure_latent import (
        SparseStructureLatent,
        TextConditionedSparseStructureLatent,
        ImageConditionedSparseStructureLatent,
    )
    
    from .structured_latent import (
        SLat,
        TextConditionedSLat,
        ImageConditionedSLat,
    )
    
    # rot mesh
    from .sparse_structure_latent_random_rot import (
        ImageConditionedSparseStructureLatentRandRot
    )

    # rot SLAT
    from .structured_latent_random_rot import (
        ImageConditionedSLatRandRot
    )

    # VAE gs dec
    from .sparse_feat2render_random_rot import (
        SparseFeat2RenderRandRot
    )

    # VAE mesh dec
    from .structured_latent2render_random_rot import (
        Slat2RenderGeoRandRot
    )

    # object-level pre-training
    from .scene_sparse_structure_latent_obj_pretrain import (
        ObjectImageConditionedSparseStructureVoxel
    )

    # scene-level training dataloader for stage 1
    from .scene_sparse_structure_latent import (
        SceneImageConditionedVoxel
    )

    # scene-level training dataloader for stage 2
    from .scene_structured_latent import (
        SceneConditionedSLat
    )