|
17 | 17 | import tqdm
|
18 | 18 | from omegaconf import DictConfig
|
19 | 19 | from pytorch3d.common.compat import prod
|
20 |
| -from pytorch3d.implicitron.models.metrics import ( |
21 |
| - RegularizationMetricsBase, |
22 |
| - ViewMetricsBase, |
| 20 | + |
| 21 | +from pytorch3d.implicitron.models.base_model import ( |
| 22 | + ImplicitronModelBase, |
| 23 | + ImplicitronRender, |
23 | 24 | )
|
24 |
| -from pytorch3d.implicitron.models.renderer.base import ImplicitronRayBundle |
25 |
| -from pytorch3d.implicitron.tools import image_utils, vis_utils |
26 |
| -from pytorch3d.implicitron.tools.config import ( |
27 |
| - expand_args_fields, |
28 |
| - registry, |
29 |
| - run_auto_creation, |
| 25 | +from pytorch3d.implicitron.models.feature_extractor import FeatureExtractorBase |
| 26 | +from pytorch3d.implicitron.models.feature_extractor.resnet_feature_extractor import ( # noqa |
| 27 | + ResNetFeatureExtractor, |
30 | 28 | )
|
31 |
| -from pytorch3d.implicitron.tools.rasterize_mc import rasterize_sparse_ray_bundle |
32 |
| -from pytorch3d.implicitron.tools.utils import cat_dataclass |
33 |
| -from pytorch3d.renderer import utils as rend_utils |
34 |
| - |
35 |
| -from pytorch3d.renderer.cameras import CamerasBase |
36 |
| - |
37 |
| -if TYPE_CHECKING: |
38 |
| - from visdom import Visdom |
39 |
| - |
40 |
| -from .base_model import ImplicitronModelBase, ImplicitronRender |
41 |
| -from .feature_extractor import FeatureExtractorBase |
42 |
| -from .feature_extractor.resnet_feature_extractor import ResNetFeatureExtractor # noqa |
43 |
| -from .global_encoder.global_encoder import GlobalEncoderBase |
44 |
| -from .implicit_function.base import ImplicitFunctionBase |
45 |
| -from .implicit_function.idr_feature_field import IdrFeatureField # noqa |
46 |
| -from .implicit_function.neural_radiance_field import ( # noqa |
| 29 | +from pytorch3d.implicitron.models.global_encoder.global_encoder import GlobalEncoderBase |
| 30 | +from pytorch3d.implicitron.models.implicit_function.base import ImplicitFunctionBase |
| 31 | +from pytorch3d.implicitron.models.implicit_function.idr_feature_field import ( # noqa |
| 32 | + IdrFeatureField, |
| 33 | +) |
| 34 | +from pytorch3d.implicitron.models.implicit_function.neural_radiance_field import ( # noqa |
47 | 35 | NeRFormerImplicitFunction,
|
48 | 36 | NeuralRadianceFieldImplicitFunction,
|
49 | 37 | )
|
50 |
| -from .implicit_function.scene_representation_networks import ( # noqa |
| 38 | +from pytorch3d.implicitron.models.implicit_function.scene_representation_networks import ( # noqa |
51 | 39 | SRNHyperNetImplicitFunction,
|
52 | 40 | SRNImplicitFunction,
|
53 | 41 | )
|
54 |
| -from .implicit_function.voxel_grid_implicit_function import ( # noqa |
| 42 | +from pytorch3d.implicitron.models.implicit_function.voxel_grid_implicit_function import ( # noqa |
55 | 43 | VoxelGridImplicitFunction,
|
56 | 44 | )
|
| 45 | +from pytorch3d.implicitron.models.metrics import ( |
| 46 | + RegularizationMetricsBase, |
| 47 | + ViewMetricsBase, |
| 48 | +) |
57 | 49 |
|
58 |
| -from .renderer.base import ( |
| 50 | +from pytorch3d.implicitron.models.renderer.base import ( |
59 | 51 | BaseRenderer,
|
60 | 52 | EvaluationMode,
|
61 | 53 | ImplicitFunctionWrapper,
|
| 54 | + ImplicitronRayBundle, |
62 | 55 | RendererOutput,
|
63 | 56 | RenderSamplingMode,
|
64 | 57 | )
|
65 |
| -from .renderer.lstm_renderer import LSTMRenderer # noqa |
66 |
| -from .renderer.multipass_ea import MultiPassEmissionAbsorptionRenderer # noqa |
67 |
| -from .renderer.ray_sampler import RaySamplerBase |
68 |
| -from .renderer.sdf_renderer import SignedDistanceFunctionRenderer # noqa |
69 |
| -from .view_pooler.view_pooler import ViewPooler |
| 58 | +from pytorch3d.implicitron.models.renderer.lstm_renderer import LSTMRenderer # noqa |
| 59 | +from pytorch3d.implicitron.models.renderer.multipass_ea import ( # noqa |
| 60 | + MultiPassEmissionAbsorptionRenderer, |
| 61 | +) |
| 62 | +from pytorch3d.implicitron.models.renderer.ray_sampler import RaySamplerBase |
| 63 | +from pytorch3d.implicitron.models.renderer.sdf_renderer import ( # noqa |
| 64 | + SignedDistanceFunctionRenderer, |
| 65 | +) |
| 66 | +from pytorch3d.implicitron.models.view_pooler.view_pooler import ViewPooler |
| 67 | +from pytorch3d.implicitron.tools import image_utils, vis_utils |
| 68 | +from pytorch3d.implicitron.tools.config import ( |
| 69 | + expand_args_fields, |
| 70 | + registry, |
| 71 | + run_auto_creation, |
| 72 | +) |
| 73 | + |
| 74 | +from pytorch3d.implicitron.tools.rasterize_mc import rasterize_sparse_ray_bundle |
| 75 | +from pytorch3d.implicitron.tools.utils import cat_dataclass |
| 76 | +from pytorch3d.renderer import utils as rend_utils |
| 77 | +from pytorch3d.renderer.cameras import CamerasBase |
70 | 78 |
|
71 | 79 |
|
| 80 | +if TYPE_CHECKING: |
| 81 | + from visdom import Visdom |
72 | 82 | logger = logging.getLogger(__name__)
|
73 | 83 |
|
74 | 84 |
|
|
0 commit comments