|
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