Skip to content

Commit d388881

Browse files
EmGarrfacebook-github-bot
authored andcommitted
Replace relative imports in generic_model.py with absolute ones
Summary: - Replace all the relative imports for generic models to absolute import: (from . import y => from pytorch3.x import y) Reviewed By: shapovalov Differential Revision: D43620682 fbshipit-source-id: 937318b339b5020d17b511a891c7b000ff659328
1 parent 33b49ce commit d388881

File tree

1 file changed

+43
-33
lines changed

1 file changed

+43
-33
lines changed

pytorch3d/implicitron/models/generic_model.py

+43-33
Original file line numberDiff line numberDiff line change
@@ -17,58 +17,68 @@
1717
import tqdm
1818
from omegaconf import DictConfig
1919
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,
2324
)
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,
3028
)
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
4735
NeRFormerImplicitFunction,
4836
NeuralRadianceFieldImplicitFunction,
4937
)
50-
from .implicit_function.scene_representation_networks import ( # noqa
38+
from pytorch3d.implicitron.models.implicit_function.scene_representation_networks import ( # noqa
5139
SRNHyperNetImplicitFunction,
5240
SRNImplicitFunction,
5341
)
54-
from .implicit_function.voxel_grid_implicit_function import ( # noqa
42+
from pytorch3d.implicitron.models.implicit_function.voxel_grid_implicit_function import ( # noqa
5543
VoxelGridImplicitFunction,
5644
)
45+
from pytorch3d.implicitron.models.metrics import (
46+
RegularizationMetricsBase,
47+
ViewMetricsBase,
48+
)
5749

58-
from .renderer.base import (
50+
from pytorch3d.implicitron.models.renderer.base import (
5951
BaseRenderer,
6052
EvaluationMode,
6153
ImplicitFunctionWrapper,
54+
ImplicitronRayBundle,
6255
RendererOutput,
6356
RenderSamplingMode,
6457
)
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
7078

7179

80+
if TYPE_CHECKING:
81+
from visdom import Visdom
7282
logger = logging.getLogger(__name__)
7383

7484

0 commit comments

Comments
 (0)