Commit d388881f authored by Emilien Garreau's avatar Emilien Garreau Committed by Facebook GitHub Bot
Browse files

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
parent 33b49ceb
...@@ -17,58 +17,68 @@ import torch ...@@ -17,58 +17,68 @@ import torch
import tqdm import tqdm
from omegaconf import DictConfig from omegaconf import DictConfig
from pytorch3d.common.compat import prod from pytorch3d.common.compat import prod
from pytorch3d.implicitron.models.metrics import (
RegularizationMetricsBase, from pytorch3d.implicitron.models.base_model import (
ViewMetricsBase, ImplicitronModelBase,
ImplicitronRender,
) )
from pytorch3d.implicitron.models.renderer.base import ImplicitronRayBundle from pytorch3d.implicitron.models.feature_extractor import FeatureExtractorBase
from pytorch3d.implicitron.tools import image_utils, vis_utils from pytorch3d.implicitron.models.feature_extractor.resnet_feature_extractor import ( # noqa
from pytorch3d.implicitron.tools.config import ( ResNetFeatureExtractor,
expand_args_fields,
registry,
run_auto_creation,
) )
from pytorch3d.implicitron.tools.rasterize_mc import rasterize_sparse_ray_bundle from pytorch3d.implicitron.models.global_encoder.global_encoder import GlobalEncoderBase
from pytorch3d.implicitron.tools.utils import cat_dataclass from pytorch3d.implicitron.models.implicit_function.base import ImplicitFunctionBase
from pytorch3d.renderer import utils as rend_utils from pytorch3d.implicitron.models.implicit_function.idr_feature_field import ( # noqa
IdrFeatureField,
from pytorch3d.renderer.cameras import CamerasBase )
from pytorch3d.implicitron.models.implicit_function.neural_radiance_field import ( # noqa
if TYPE_CHECKING:
from visdom import Visdom
from .base_model import ImplicitronModelBase, ImplicitronRender
from .feature_extractor import FeatureExtractorBase
from .feature_extractor.resnet_feature_extractor import ResNetFeatureExtractor # noqa
from .global_encoder.global_encoder import GlobalEncoderBase
from .implicit_function.base import ImplicitFunctionBase
from .implicit_function.idr_feature_field import IdrFeatureField # noqa
from .implicit_function.neural_radiance_field import ( # noqa
NeRFormerImplicitFunction, NeRFormerImplicitFunction,
NeuralRadianceFieldImplicitFunction, NeuralRadianceFieldImplicitFunction,
) )
from .implicit_function.scene_representation_networks import ( # noqa from pytorch3d.implicitron.models.implicit_function.scene_representation_networks import ( # noqa
SRNHyperNetImplicitFunction, SRNHyperNetImplicitFunction,
SRNImplicitFunction, SRNImplicitFunction,
) )
from .implicit_function.voxel_grid_implicit_function import ( # noqa from pytorch3d.implicitron.models.implicit_function.voxel_grid_implicit_function import ( # noqa
VoxelGridImplicitFunction, VoxelGridImplicitFunction,
) )
from pytorch3d.implicitron.models.metrics import (
RegularizationMetricsBase,
ViewMetricsBase,
)
from .renderer.base import ( from pytorch3d.implicitron.models.renderer.base import (
BaseRenderer, BaseRenderer,
EvaluationMode, EvaluationMode,
ImplicitFunctionWrapper, ImplicitFunctionWrapper,
ImplicitronRayBundle,
RendererOutput, RendererOutput,
RenderSamplingMode, RenderSamplingMode,
) )
from .renderer.lstm_renderer import LSTMRenderer # noqa from pytorch3d.implicitron.models.renderer.lstm_renderer import LSTMRenderer # noqa
from .renderer.multipass_ea import MultiPassEmissionAbsorptionRenderer # noqa from pytorch3d.implicitron.models.renderer.multipass_ea import ( # noqa
from .renderer.ray_sampler import RaySamplerBase MultiPassEmissionAbsorptionRenderer,
from .renderer.sdf_renderer import SignedDistanceFunctionRenderer # noqa )
from .view_pooler.view_pooler import ViewPooler from pytorch3d.implicitron.models.renderer.ray_sampler import RaySamplerBase
from pytorch3d.implicitron.models.renderer.sdf_renderer import ( # noqa
SignedDistanceFunctionRenderer,
)
from pytorch3d.implicitron.models.view_pooler.view_pooler import ViewPooler
from pytorch3d.implicitron.tools import image_utils, vis_utils
from pytorch3d.implicitron.tools.config import (
expand_args_fields,
registry,
run_auto_creation,
)
from pytorch3d.implicitron.tools.rasterize_mc import rasterize_sparse_ray_bundle
from pytorch3d.implicitron.tools.utils import cat_dataclass
from pytorch3d.renderer import utils as rend_utils
from pytorch3d.renderer.cameras import CamerasBase
if TYPE_CHECKING:
from visdom import Visdom
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment