diff --git a/projects/__init__.py b/projects/__init__.py new file mode 100644 index 00000000..2e41cd71 --- /dev/null +++ b/projects/__init__.py @@ -0,0 +1,5 @@ +# Copyright (c) Meta Platforms, Inc. and affiliates. +# All rights reserved. +# +# This source code is licensed under the BSD-style license found in the +# LICENSE file in the root directory of this source tree. diff --git a/projects/implicitron_trainer/__init__.py b/projects/implicitron_trainer/__init__.py new file mode 100644 index 00000000..2e41cd71 --- /dev/null +++ b/projects/implicitron_trainer/__init__.py @@ -0,0 +1,5 @@ +# Copyright (c) Meta Platforms, Inc. and affiliates. +# All rights reserved. +# +# This source code is licensed under the BSD-style license found in the +# LICENSE file in the root directory of this source tree. diff --git a/setup.py b/setup.py index 00706f04..87902b57 100755 --- a/setup.py +++ b/setup.py @@ -129,6 +129,7 @@ if os.getenv("PYTORCH3D_NO_NINJA", "0") == "1": else: BuildExtension = torch.utils.cpp_extension.BuildExtension +trainer = "projects.implicitron_trainer" setup( name="pytorch3d", @@ -138,12 +139,19 @@ setup( description="PyTorch3D is FAIR's library of reusable components " "for deep Learning with 3D data.", packages=find_packages( - exclude=("configs", "tests", "tests.*", "docs.*", "projects.*") + exclude=("configs", "tests", "tests.*", "docs.*", "projects.nerf.*") ), install_requires=["fvcore", "iopath"], extras_require={ "all": ["matplotlib", "tqdm>4.29.0", "imageio", "ipywidgets"], "dev": ["flake8", "isort", "black==19.3b0"], + "implicitron": ["hydra-core>=1.1", "visdom", "lpips", "matplotlib"], + }, + entry_points={ + "console_scripts": [ + f"pytorch3d_implicitron_runner={trainer}.experiment", + f"pytorch3d_implicitron_visualizer={trainer}.visualize_reconstruction", + ] }, ext_modules=get_extensions(), cmdclass={"build_ext": BuildExtension},