423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-08-02 03:42:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

setup.py for implicitron_trainer

Browse Source 
Summary: Enable `pytorch3d_implicitron_runner` executable

Reviewed By: shapovalov

Differential Revision: D34754902

fbshipit-source-id: 213f3e9183e3f7dd7b4df16ad77d95fbc971d625
This commit is contained in:
Jeremy Reizenstein 2022-03-28 04:50:26 -07:00 committed by Facebook GitHub Bot
parent 97894fb37b
commit 0c3bed55be
3 changed files with 19 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
projects/__init__.py Normal file
View File

@ -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.

5
projects/implicitron_trainer/__init__.py Normal file
View File

@ -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.

10
setup.py
View File

@ -129,6 +129,7 @@ if os.getenv("PYTORCH3D_NO_NINJA", "0") == "1":
else: else:
BuildExtension = torch.utils.cpp_extension.BuildExtension BuildExtension = torch.utils.cpp_extension.BuildExtension
trainer = "projects.implicitron_trainer"
setup( setup(
name="pytorch3d", name="pytorch3d",
@ -138,12 +139,19 @@ setup(
description="PyTorch3D is FAIR's library of reusable components " description="PyTorch3D is FAIR's library of reusable components "
"for deep Learning with 3D data.", "for deep Learning with 3D data.",
packages=find_packages( packages=find_packages(
exclude=("configs", "tests", "tests.*", "docs.*", "projects.*") exclude=("configs", "tests", "tests.*", "docs.*", "projects.nerf.*")
), ),
install_requires=["fvcore", "iopath"], install_requires=["fvcore", "iopath"],
extras_require={ extras_require={
"all": ["matplotlib", "tqdm>4.29.0", "imageio", "ipywidgets"], "all": ["matplotlib", "tqdm>4.29.0", "imageio", "ipywidgets"],
"dev": ["flake8", "isort", "black==19.3b0"], "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(), ext_modules=get_extensions(),
cmdclass={"build_ext": BuildExtension}, cmdclass={"build_ext": BuildExtension},