mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-12-24 00:00:34 +08:00
Compare commits
1 Commits
v0.7.2
...
classner-p
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
e7c1f026ea |
@@ -117,7 +117,7 @@ jobs:
|
||||
- run:
|
||||
name: build
|
||||
no_output_timeout: 20m
|
||||
command: MAX_JOBS=15 TEST_FLAG=--no-test python3 packaging/build_conda.py
|
||||
command: MAX_JOBS=15 TEST_FLAG=--no-test packaging/build_conda.sh
|
||||
- store_artifacts:
|
||||
path: /opt/conda/conda-bld/linux-64
|
||||
- persist_to_workspace:
|
||||
@@ -154,12 +154,12 @@ jobs:
|
||||
export JUST_TESTRUN=1
|
||||
VARS_TO_PASS="-e PYTHON_VERSION -e BUILD_VERSION -e PYTORCH_VERSION -e CU_VERSION -e JUST_TESTRUN"
|
||||
|
||||
docker run --gpus all --ipc=host -v $(pwd):/remote -w /remote ${VARS_TO_PASS} ${TESTRUN_DOCKER_IMAGE} python3 ./packaging/build_conda.py
|
||||
docker run --gpus all --ipc=host -v $(pwd):/remote -w /remote ${VARS_TO_PASS} ${TESTRUN_DOCKER_IMAGE} ./packaging/build_conda.sh
|
||||
|
||||
binary_macos_wheel:
|
||||
<<: *binary_common
|
||||
macos:
|
||||
xcode: "13.4.1"
|
||||
xcode: "12.0"
|
||||
steps:
|
||||
- checkout
|
||||
- run:
|
||||
@@ -182,23 +182,23 @@ workflows:
|
||||
# context: DOCKERHUB_TOKEN
|
||||
{{workflows()}}
|
||||
- binary_linux_conda_cuda:
|
||||
name: testrun_conda_cuda_py38_cu102_pyt190
|
||||
name: testrun_conda_cuda_py37_cu102_pyt190
|
||||
context: DOCKERHUB_TOKEN
|
||||
python_version: "3.8"
|
||||
python_version: "3.7"
|
||||
pytorch_version: '1.9.0'
|
||||
cu_version: "cu102"
|
||||
- binary_macos_wheel:
|
||||
cu_version: cpu
|
||||
name: macos_wheel_py3.8_cpu
|
||||
name: macos_wheel_py37_cpu
|
||||
python_version: '3.7'
|
||||
pytorch_version: '1.12.0'
|
||||
- binary_macos_wheel:
|
||||
cu_version: cpu
|
||||
name: macos_wheel_py38_cpu
|
||||
python_version: '3.8'
|
||||
pytorch_version: '1.13.0'
|
||||
pytorch_version: '1.12.0'
|
||||
- binary_macos_wheel:
|
||||
cu_version: cpu
|
||||
name: macos_wheel_py3.9_cpu
|
||||
name: macos_wheel_py39_cpu
|
||||
python_version: '3.9'
|
||||
pytorch_version: '1.13.0'
|
||||
- binary_macos_wheel:
|
||||
cu_version: cpu
|
||||
name: macos_wheel_py3.10_cpu
|
||||
python_version: '3.10'
|
||||
pytorch_version: '1.13.0'
|
||||
pytorch_version: '1.12.0'
|
||||
|
||||
@@ -117,7 +117,7 @@ jobs:
|
||||
- run:
|
||||
name: build
|
||||
no_output_timeout: 20m
|
||||
command: MAX_JOBS=15 TEST_FLAG=--no-test python3 packaging/build_conda.py
|
||||
command: MAX_JOBS=15 TEST_FLAG=--no-test packaging/build_conda.sh
|
||||
- store_artifacts:
|
||||
path: /opt/conda/conda-bld/linux-64
|
||||
- persist_to_workspace:
|
||||
@@ -154,12 +154,12 @@ jobs:
|
||||
export JUST_TESTRUN=1
|
||||
VARS_TO_PASS="-e PYTHON_VERSION -e BUILD_VERSION -e PYTORCH_VERSION -e CU_VERSION -e JUST_TESTRUN"
|
||||
|
||||
docker run --gpus all --ipc=host -v $(pwd):/remote -w /remote ${VARS_TO_PASS} ${TESTRUN_DOCKER_IMAGE} python3 ./packaging/build_conda.py
|
||||
docker run --gpus all --ipc=host -v $(pwd):/remote -w /remote ${VARS_TO_PASS} ${TESTRUN_DOCKER_IMAGE} ./packaging/build_conda.sh
|
||||
|
||||
binary_macos_wheel:
|
||||
<<: *binary_common
|
||||
macos:
|
||||
xcode: "13.4.1"
|
||||
xcode: "12.0"
|
||||
steps:
|
||||
- checkout
|
||||
- run:
|
||||
@@ -180,6 +180,205 @@ workflows:
|
||||
jobs:
|
||||
# - main:
|
||||
# context: DOCKERHUB_TOKEN
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu101
|
||||
name: linux_conda_py37_cu101_pyt180
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.8.0
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu102
|
||||
name: linux_conda_py37_cu102_pyt180
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.8.0
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu111
|
||||
name: linux_conda_py37_cu111_pyt180
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.8.0
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu101
|
||||
name: linux_conda_py37_cu101_pyt181
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.8.1
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu102
|
||||
name: linux_conda_py37_cu102_pyt181
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.8.1
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu111
|
||||
name: linux_conda_py37_cu111_pyt181
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.8.1
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu102
|
||||
name: linux_conda_py37_cu102_pyt190
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.9.0
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu111
|
||||
name: linux_conda_py37_cu111_pyt190
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.9.0
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu102
|
||||
name: linux_conda_py37_cu102_pyt191
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.9.1
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu111
|
||||
name: linux_conda_py37_cu111_pyt191
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.9.1
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu102
|
||||
name: linux_conda_py37_cu102_pyt1100
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.10.0
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu111
|
||||
name: linux_conda_py37_cu111_pyt1100
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.10.0
|
||||
- binary_linux_conda:
|
||||
conda_docker_image: pytorch/conda-builder:cuda113
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu113
|
||||
name: linux_conda_py37_cu113_pyt1100
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.10.0
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu102
|
||||
name: linux_conda_py37_cu102_pyt1101
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.10.1
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu111
|
||||
name: linux_conda_py37_cu111_pyt1101
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.10.1
|
||||
- binary_linux_conda:
|
||||
conda_docker_image: pytorch/conda-builder:cuda113
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu113
|
||||
name: linux_conda_py37_cu113_pyt1101
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.10.1
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu102
|
||||
name: linux_conda_py37_cu102_pyt1102
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.10.2
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu111
|
||||
name: linux_conda_py37_cu111_pyt1102
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.10.2
|
||||
- binary_linux_conda:
|
||||
conda_docker_image: pytorch/conda-builder:cuda113
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu113
|
||||
name: linux_conda_py37_cu113_pyt1102
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.10.2
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu102
|
||||
name: linux_conda_py37_cu102_pyt1110
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.11.0
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu111
|
||||
name: linux_conda_py37_cu111_pyt1110
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.11.0
|
||||
- binary_linux_conda:
|
||||
conda_docker_image: pytorch/conda-builder:cuda113
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu113
|
||||
name: linux_conda_py37_cu113_pyt1110
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.11.0
|
||||
- binary_linux_conda:
|
||||
conda_docker_image: pytorch/conda-builder:cuda115
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu115
|
||||
name: linux_conda_py37_cu115_pyt1110
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.11.0
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu102
|
||||
name: linux_conda_py37_cu102_pyt1120
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.12.0
|
||||
- binary_linux_conda:
|
||||
conda_docker_image: pytorch/conda-builder:cuda113
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu113
|
||||
name: linux_conda_py37_cu113_pyt1120
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.12.0
|
||||
- binary_linux_conda:
|
||||
conda_docker_image: pytorch/conda-builder:cuda116
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu116
|
||||
name: linux_conda_py37_cu116_pyt1120
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.12.0
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu101
|
||||
name: linux_conda_py38_cu101_pyt180
|
||||
python_version: '3.8'
|
||||
pytorch_version: 1.8.0
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu102
|
||||
name: linux_conda_py38_cu102_pyt180
|
||||
python_version: '3.8'
|
||||
pytorch_version: 1.8.0
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu111
|
||||
name: linux_conda_py38_cu111_pyt180
|
||||
python_version: '3.8'
|
||||
pytorch_version: 1.8.0
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu101
|
||||
name: linux_conda_py38_cu101_pyt181
|
||||
python_version: '3.8'
|
||||
pytorch_version: 1.8.1
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu102
|
||||
name: linux_conda_py38_cu102_pyt181
|
||||
python_version: '3.8'
|
||||
pytorch_version: 1.8.1
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu111
|
||||
name: linux_conda_py38_cu111_pyt181
|
||||
python_version: '3.8'
|
||||
pytorch_version: 1.8.1
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu102
|
||||
@@ -307,40 +506,42 @@ workflows:
|
||||
name: linux_conda_py38_cu116_pyt1120
|
||||
python_version: '3.8'
|
||||
pytorch_version: 1.12.0
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu101
|
||||
name: linux_conda_py39_cu101_pyt180
|
||||
python_version: '3.9'
|
||||
pytorch_version: 1.8.0
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu102
|
||||
name: linux_conda_py38_cu102_pyt1121
|
||||
python_version: '3.8'
|
||||
pytorch_version: 1.12.1
|
||||
name: linux_conda_py39_cu102_pyt180
|
||||
python_version: '3.9'
|
||||
pytorch_version: 1.8.0
|
||||
- binary_linux_conda:
|
||||
conda_docker_image: pytorch/conda-builder:cuda113
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu113
|
||||
name: linux_conda_py38_cu113_pyt1121
|
||||
python_version: '3.8'
|
||||
pytorch_version: 1.12.1
|
||||
cu_version: cu111
|
||||
name: linux_conda_py39_cu111_pyt180
|
||||
python_version: '3.9'
|
||||
pytorch_version: 1.8.0
|
||||
- binary_linux_conda:
|
||||
conda_docker_image: pytorch/conda-builder:cuda116
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu116
|
||||
name: linux_conda_py38_cu116_pyt1121
|
||||
python_version: '3.8'
|
||||
pytorch_version: 1.12.1
|
||||
cu_version: cu101
|
||||
name: linux_conda_py39_cu101_pyt181
|
||||
python_version: '3.9'
|
||||
pytorch_version: 1.8.1
|
||||
- binary_linux_conda:
|
||||
conda_docker_image: pytorch/conda-builder:cuda116
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu116
|
||||
name: linux_conda_py38_cu116_pyt1130
|
||||
python_version: '3.8'
|
||||
pytorch_version: 1.13.0
|
||||
cu_version: cu102
|
||||
name: linux_conda_py39_cu102_pyt181
|
||||
python_version: '3.9'
|
||||
pytorch_version: 1.8.1
|
||||
- binary_linux_conda:
|
||||
conda_docker_image: pytorch/conda-builder:cuda117
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu117
|
||||
name: linux_conda_py38_cu117_pyt1130
|
||||
python_version: '3.8'
|
||||
pytorch_version: 1.13.0
|
||||
cu_version: cu111
|
||||
name: linux_conda_py39_cu111_pyt181
|
||||
python_version: '3.9'
|
||||
pytorch_version: 1.8.1
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu102
|
||||
@@ -468,40 +669,6 @@ workflows:
|
||||
name: linux_conda_py39_cu116_pyt1120
|
||||
python_version: '3.9'
|
||||
pytorch_version: 1.12.0
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu102
|
||||
name: linux_conda_py39_cu102_pyt1121
|
||||
python_version: '3.9'
|
||||
pytorch_version: 1.12.1
|
||||
- binary_linux_conda:
|
||||
conda_docker_image: pytorch/conda-builder:cuda113
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu113
|
||||
name: linux_conda_py39_cu113_pyt1121
|
||||
python_version: '3.9'
|
||||
pytorch_version: 1.12.1
|
||||
- binary_linux_conda:
|
||||
conda_docker_image: pytorch/conda-builder:cuda116
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu116
|
||||
name: linux_conda_py39_cu116_pyt1121
|
||||
python_version: '3.9'
|
||||
pytorch_version: 1.12.1
|
||||
- binary_linux_conda:
|
||||
conda_docker_image: pytorch/conda-builder:cuda116
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu116
|
||||
name: linux_conda_py39_cu116_pyt1130
|
||||
python_version: '3.9'
|
||||
pytorch_version: 1.13.0
|
||||
- binary_linux_conda:
|
||||
conda_docker_image: pytorch/conda-builder:cuda117
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu117
|
||||
name: linux_conda_py39_cu117_pyt1130
|
||||
python_version: '3.9'
|
||||
pytorch_version: 1.13.0
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu102
|
||||
@@ -548,58 +715,24 @@ workflows:
|
||||
name: linux_conda_py310_cu116_pyt1120
|
||||
python_version: '3.10'
|
||||
pytorch_version: 1.12.0
|
||||
- binary_linux_conda:
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu102
|
||||
name: linux_conda_py310_cu102_pyt1121
|
||||
python_version: '3.10'
|
||||
pytorch_version: 1.12.1
|
||||
- binary_linux_conda:
|
||||
conda_docker_image: pytorch/conda-builder:cuda113
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu113
|
||||
name: linux_conda_py310_cu113_pyt1121
|
||||
python_version: '3.10'
|
||||
pytorch_version: 1.12.1
|
||||
- binary_linux_conda:
|
||||
conda_docker_image: pytorch/conda-builder:cuda116
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu116
|
||||
name: linux_conda_py310_cu116_pyt1121
|
||||
python_version: '3.10'
|
||||
pytorch_version: 1.12.1
|
||||
- binary_linux_conda:
|
||||
conda_docker_image: pytorch/conda-builder:cuda116
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu116
|
||||
name: linux_conda_py310_cu116_pyt1130
|
||||
python_version: '3.10'
|
||||
pytorch_version: 1.13.0
|
||||
- binary_linux_conda:
|
||||
conda_docker_image: pytorch/conda-builder:cuda117
|
||||
context: DOCKERHUB_TOKEN
|
||||
cu_version: cu117
|
||||
name: linux_conda_py310_cu117_pyt1130
|
||||
python_version: '3.10'
|
||||
pytorch_version: 1.13.0
|
||||
- binary_linux_conda_cuda:
|
||||
name: testrun_conda_cuda_py38_cu102_pyt190
|
||||
name: testrun_conda_cuda_py37_cu102_pyt190
|
||||
context: DOCKERHUB_TOKEN
|
||||
python_version: "3.8"
|
||||
python_version: "3.7"
|
||||
pytorch_version: '1.9.0'
|
||||
cu_version: "cu102"
|
||||
- binary_macos_wheel:
|
||||
cu_version: cpu
|
||||
name: macos_wheel_py3.8_cpu
|
||||
name: macos_wheel_py37_cpu
|
||||
python_version: '3.7'
|
||||
pytorch_version: '1.12.0'
|
||||
- binary_macos_wheel:
|
||||
cu_version: cpu
|
||||
name: macos_wheel_py38_cpu
|
||||
python_version: '3.8'
|
||||
pytorch_version: '1.13.0'
|
||||
pytorch_version: '1.12.0'
|
||||
- binary_macos_wheel:
|
||||
cu_version: cpu
|
||||
name: macos_wheel_py3.9_cpu
|
||||
name: macos_wheel_py39_cpu
|
||||
python_version: '3.9'
|
||||
pytorch_version: '1.13.0'
|
||||
- binary_macos_wheel:
|
||||
cu_version: cpu
|
||||
name: macos_wheel_py3.10_cpu
|
||||
python_version: '3.10'
|
||||
pytorch_version: '1.13.0'
|
||||
pytorch_version: '1.12.0'
|
||||
|
||||
@@ -20,6 +20,8 @@ from packaging import version
|
||||
# version of pytorch.
|
||||
# Pytorch 1.4 also supports cuda 10.0 but we no longer build for cuda 10.0 at all.
|
||||
CONDA_CUDA_VERSIONS = {
|
||||
"1.8.0": ["cu101", "cu102", "cu111"],
|
||||
"1.8.1": ["cu101", "cu102", "cu111"],
|
||||
"1.9.0": ["cu102", "cu111"],
|
||||
"1.9.1": ["cu102", "cu111"],
|
||||
"1.10.0": ["cu102", "cu111", "cu113"],
|
||||
@@ -27,8 +29,6 @@ CONDA_CUDA_VERSIONS = {
|
||||
"1.10.2": ["cu102", "cu111", "cu113"],
|
||||
"1.11.0": ["cu102", "cu111", "cu113", "cu115"],
|
||||
"1.12.0": ["cu102", "cu113", "cu116"],
|
||||
"1.12.1": ["cu102", "cu113", "cu116"],
|
||||
"1.13.0": ["cu116", "cu117"],
|
||||
}
|
||||
|
||||
|
||||
@@ -41,8 +41,6 @@ def conda_docker_image_for_cuda(cuda_version):
|
||||
return "pytorch/conda-builder:cuda115"
|
||||
if cuda_version == "cu116":
|
||||
return "pytorch/conda-builder:cuda116"
|
||||
if cuda_version == "cu117":
|
||||
return "pytorch/conda-builder:cuda117"
|
||||
raise ValueError("Unknown cuda version")
|
||||
|
||||
|
||||
@@ -66,7 +64,7 @@ def pytorch_versions_for_python(python_version):
|
||||
def workflows(prefix="", filter_branch=None, upload=False, indentation=6):
|
||||
w = []
|
||||
for btype in ["conda"]:
|
||||
for python_version in ["3.8", "3.9", "3.10"]:
|
||||
for python_version in ["3.7", "3.8", "3.9", "3.10"]:
|
||||
for pytorch_version in pytorch_versions_for_python(python_version):
|
||||
for cu_version in CONDA_CUDA_VERSIONS[pytorch_version]:
|
||||
w += workflow_pair(
|
||||
|
||||
7
.github/CONTRIBUTING.md
vendored
7
.github/CONTRIBUTING.md
vendored
@@ -16,13 +16,14 @@ We do not always accept new features, and we take the following factors into con
|
||||
|
||||
When sending a PR, please ensure you complete the following steps:
|
||||
|
||||
1. Fork the repo and create your branch from `main`. Follow the instructions
|
||||
1. Fork the repo and create your branch from `master`. Follow the instructions
|
||||
in [INSTALL.md](../INSTALL.md) to build the repo.
|
||||
2. If you've added code that should be tested, add tests.
|
||||
3. If you've changed any APIs, please update the documentation.
|
||||
4. Ensure the test suite passes, by running this from the project root:
|
||||
4. Ensure the test suite passes:
|
||||
```
|
||||
python -m unittest discover -v -s tests -t .
|
||||
cd pytorch3d/tests
|
||||
python -m unittest -v
|
||||
```
|
||||
5. Make sure your code lints by running `dev/linter.sh` from the project root.
|
||||
6. If a PR contains multiple orthogonal changes, split it into multiple separate PRs.
|
||||
|
||||
BIN
.github/implicitron_config.gif
vendored
BIN
.github/implicitron_config.gif
vendored
Binary file not shown.
|
Before Width: | Height: | Size: 5.0 MiB |
27
.github/workflows/stale.yml
vendored
Normal file
27
.github/workflows/stale.yml
vendored
Normal file
@@ -0,0 +1,27 @@
|
||||
name: Mark stale issues and pull requests
|
||||
|
||||
on:
|
||||
schedule:
|
||||
- cron: '31 5 * * *'
|
||||
|
||||
jobs:
|
||||
stale:
|
||||
|
||||
runs-on: ubuntu-latest
|
||||
permissions:
|
||||
issues: write
|
||||
pull-requests: write
|
||||
|
||||
steps:
|
||||
- uses: actions/stale@v3
|
||||
with:
|
||||
repo-token: ${{ secrets.GITHUB_TOKEN }}
|
||||
exempt-issue-labels: 'enhancement,how-to'
|
||||
stale-issue-message: 'This issue is stale because it has been open 30 days with no activity. Remove stale label or comment or this will be closed in 5 days.'
|
||||
stale-pr-message: 'This PR is stale because it has been open 45 days with no activity. Remove stale label or comment or this will be closed in 10 days.'
|
||||
close-issue-message: 'This issue was closed because it has been stalled for 5 days with no activity.'
|
||||
close-pr-message: 'This PR was closed because it has been stalled for 10 days with no activity.'
|
||||
days-before-issue-stale: 30
|
||||
days-before-pr-stale: 45
|
||||
days-before-issue-close: 5
|
||||
days-before-pr-close: 10
|
||||
12
INSTALL.md
12
INSTALL.md
@@ -8,8 +8,8 @@
|
||||
The core library is written in PyTorch. Several components have underlying implementation in CUDA for improved performance. A subset of these components have CPU implementations in C++/PyTorch. It is advised to use PyTorch3D with GPU support in order to use all the features.
|
||||
|
||||
- Linux or macOS or Windows
|
||||
- Python 3.8, 3.9 or 3.10
|
||||
- PyTorch 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2, 1.11.0, 1.12.0, 1.12.1 or 1.13.0.
|
||||
- Python 3.6, 3.7, 3.8 or 3.9
|
||||
- PyTorch 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2, 1.11.0 or 1.12.0.
|
||||
- torchvision that matches the PyTorch installation. You can install them together as explained at pytorch.org to make sure of this.
|
||||
- gcc & g++ ≥ 4.9
|
||||
- [fvcore](https://github.com/facebookresearch/fvcore)
|
||||
@@ -21,11 +21,11 @@ The runtime dependencies can be installed by running:
|
||||
```
|
||||
conda create -n pytorch3d python=3.9
|
||||
conda activate pytorch3d
|
||||
conda install -c pytorch pytorch=1.9.1 torchvision cudatoolkit=11.6
|
||||
conda install -c pytorch pytorch=1.9.1 torchvision cudatoolkit=10.2
|
||||
conda install -c fvcore -c iopath -c conda-forge fvcore iopath
|
||||
```
|
||||
|
||||
For the CUB build time dependency, which you only need if you have CUDA older than 11.7, if you are using conda, you can continue with
|
||||
For the CUB build time dependency, if you are using conda, you can continue with
|
||||
```
|
||||
conda install -c bottler nvidiacub
|
||||
```
|
||||
@@ -78,14 +78,14 @@ Or, to install a nightly (non-official, alpha) build:
|
||||
conda install pytorch3d -c pytorch3d-nightly
|
||||
```
|
||||
### 2. Install from PyPI, on Mac only.
|
||||
This works with pytorch 1.13.0 only. The build is CPU only.
|
||||
This works with pytorch 1.12.0 only. The build is CPU only.
|
||||
```
|
||||
pip install pytorch3d
|
||||
```
|
||||
|
||||
### 3. Install wheels for Linux
|
||||
We have prebuilt wheels with CUDA for Linux for PyTorch 1.11.0, for each of the supported CUDA versions,
|
||||
for Python 3.8 and 3.9. This is for ease of use on Google Colab.
|
||||
for Python 3.7, 3.8 and 3.9. This is for ease of use on Google Colab.
|
||||
These are installed in a special way.
|
||||
For example, to install for Python 3.8, PyTorch 1.11.0 and CUDA 11.3
|
||||
```
|
||||
|
||||
14
README.md
14
README.md
@@ -12,7 +12,6 @@ Key features include:
|
||||
- Data structure for storing and manipulating triangle meshes
|
||||
- Efficient operations on triangle meshes (projective transformations, graph convolution, sampling, loss functions)
|
||||
- A differentiable mesh renderer
|
||||
- Implicitron, see [its README](projects/implicitron_trainer), a framework for new-view synthesis via implicit representations.
|
||||
|
||||
PyTorch3D is designed to integrate smoothly with deep learning methods for predicting and manipulating 3D data.
|
||||
For this reason, all operators in PyTorch3D:
|
||||
@@ -56,11 +55,6 @@ Get started with PyTorch3D by trying one of the tutorial notebooks.
|
||||
|:------------------------------------------------------------:|:--------------------------------------------------:|
|
||||
| [Fit Textured Volume](https://github.com/facebookresearch/pytorch3d/blob/main/docs/tutorials/fit_textured_volume.ipynb)| [Fit A Simple Neural Radiance Field](https://github.com/facebookresearch/pytorch3d/blob/main/docs/tutorials/fit_simple_neural_radiance_field.ipynb)|
|
||||
|
||||
| <img src="https://raw.githubusercontent.com/facebookresearch/pytorch3d/main/.github/fit_textured_volume.gif" width="310"/> | <img src="https://raw.githubusercontent.com/facebookresearch/pytorch3d/main/.github/implicitron_config.gif" width="310" height="310"/>
|
||||
|:------------------------------------------------------------:|:--------------------------------------------------:|
|
||||
| [Fit Textured Volume in Implicitron](https://github.com/facebookresearch/pytorch3d/blob/main/docs/tutorials/implicitron_volumes.ipynb)| [Implicitron Config System](https://github.com/facebookresearch/pytorch3d/blob/main/docs/tutorials/implicitron_config_system.ipynb)|
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
@@ -99,8 +93,6 @@ In alphabetical order:
|
||||
|
||||
* Amitav Baruah
|
||||
* Steve Branson
|
||||
* Krzysztof Chalupka
|
||||
* Jiali Duan
|
||||
* Luya Gao
|
||||
* Georgia Gkioxari
|
||||
* Taylor Gordon
|
||||
@@ -144,12 +136,6 @@ If you are using the pulsar backend for sphere-rendering (the `PulsarPointRender
|
||||
|
||||
Please see below for a timeline of the codebase updates in reverse chronological order. We are sharing updates on the releases as well as research projects which are built with PyTorch3D. The changelogs for the releases are available under [`Releases`](https://github.com/facebookresearch/pytorch3d/releases), and the builds can be installed using `conda` as per the instructions in [INSTALL.md](INSTALL.md).
|
||||
|
||||
**[Oct 23rd 2022]:** PyTorch3D [v0.7.1](https://github.com/facebookresearch/pytorch3d/releases/tag/v0.7.1) released.
|
||||
|
||||
**[Aug 10th 2022]:** PyTorch3D [v0.7.0](https://github.com/facebookresearch/pytorch3d/releases/tag/v0.7.0) released with Implicitron and MeshRasterizerOpenGL.
|
||||
|
||||
**[Apr 28th 2022]:** PyTorch3D [v0.6.2](https://github.com/facebookresearch/pytorch3d/releases/tag/v0.6.2) released
|
||||
|
||||
**[Dec 16th 2021]:** PyTorch3D [v0.6.1](https://github.com/facebookresearch/pytorch3d/releases/tag/v0.6.1) released
|
||||
|
||||
**[Oct 6th 2021]:** PyTorch3D [v0.6.0](https://github.com/facebookresearch/pytorch3d/releases/tag/v0.6.0) released
|
||||
|
||||
@@ -17,8 +17,7 @@ It has no dependencies.
|
||||
|
||||
def get_test_files() -> List[Path]:
|
||||
root = Path(__file__).parent.parent
|
||||
dirs = ["tests", "projects/implicitron_trainer"]
|
||||
return [i for dir in dirs for i in (root / dir).glob("**/test*.py")]
|
||||
return list((root / "tests").glob("**/test*.py"))
|
||||
|
||||
|
||||
def tests_from_file(path: Path, base: str) -> List[str]:
|
||||
|
||||
@@ -1,162 +0,0 @@
|
||||
#!/usr/bin/env python3
|
||||
# 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.
|
||||
|
||||
"""
|
||||
This script makes the stubs for implicitron in docs/modules.
|
||||
"""
|
||||
|
||||
from pathlib import Path
|
||||
|
||||
ROOT_DIR = Path(__file__).resolve().parent.parent
|
||||
|
||||
|
||||
def paths_to_modules(paths):
|
||||
"""
|
||||
Given an iterable of paths, return equivalent list of modules.
|
||||
"""
|
||||
return [
|
||||
str(i.relative_to(ROOT_DIR))[:-3].replace("/", ".")
|
||||
for i in paths
|
||||
if "__pycache__" not in str(i)
|
||||
]
|
||||
|
||||
|
||||
def create_one_file(title, description, sources, dest_file):
|
||||
with open(dest_file, "w") as f:
|
||||
print(title, file=f)
|
||||
print("=" * len(title), file=f)
|
||||
print(file=f)
|
||||
print(description, file=f)
|
||||
for source in sources:
|
||||
if source.find("._") != -1:
|
||||
# ignore internal modules including __init__.py
|
||||
continue
|
||||
print(f"\n.. automodule:: {source}", file=f)
|
||||
print(" :members:", file=f)
|
||||
print(" :undoc-members:", file=f)
|
||||
print(" :show-inheritance:", file=f)
|
||||
|
||||
|
||||
def iterate_directory(directory_path, dest):
|
||||
"""
|
||||
Create a file for each module in the given path
|
||||
"""
|
||||
toc = []
|
||||
if not dest.exists():
|
||||
dest.mkdir()
|
||||
for file in sorted(directory_path.glob("*.py")):
|
||||
if file.stem.startswith("_"):
|
||||
continue
|
||||
module = paths_to_modules([file])
|
||||
create_one_file(module[0], file.stem, module, dest / f"{file.stem}.rst")
|
||||
toc.append(file.stem)
|
||||
|
||||
for subdir in directory_path.iterdir():
|
||||
if not subdir.is_dir():
|
||||
continue
|
||||
if subdir.name == "fb":
|
||||
continue
|
||||
if subdir.name.startswith("_"):
|
||||
continue
|
||||
iterate_directory(subdir, dest / (subdir.name))
|
||||
toc.append(f"{subdir.name}/index")
|
||||
|
||||
paths_to_modules_ = paths_to_modules([directory_path.with_suffix(".XX")])
|
||||
if len(paths_to_modules_) == 0:
|
||||
return
|
||||
title = paths_to_modules_[0]
|
||||
|
||||
with open(dest / "index.rst", "w") as f:
|
||||
print(title, file=f)
|
||||
print("=" * len(title), file=f)
|
||||
print("\n.. toctree::\n", file=f)
|
||||
for item in toc:
|
||||
print(f" {item}", file=f)
|
||||
|
||||
|
||||
def make_directory_index(title: str, directory_path: Path):
|
||||
index_file = directory_path / "index.rst"
|
||||
directory_rsts = sorted(directory_path.glob("*.rst"))
|
||||
subdirs = sorted([f for f in directory_path.iterdir() if f.is_dir()])
|
||||
with open(index_file, "w") as f:
|
||||
print(title, file=f)
|
||||
print("=" * len(title), file=f)
|
||||
print("\n.. toctree::\n", file=f)
|
||||
for subdir in subdirs:
|
||||
print(f" {subdir.stem}/index.rst", file=f)
|
||||
for rst in directory_rsts:
|
||||
if rst.stem == "index":
|
||||
continue
|
||||
print(f" {rst.stem}", file=f)
|
||||
|
||||
|
||||
def do_implicitron():
|
||||
DEST_DIR = Path(__file__).resolve().parent / "modules/implicitron"
|
||||
|
||||
iterate_directory(ROOT_DIR / "pytorch3d/implicitron/models", DEST_DIR / "models")
|
||||
|
||||
unwanted_tools = ["configurable", "depth_cleanup", "utils"]
|
||||
tools_sources = sorted(ROOT_DIR.glob("pytorch3d/implicitron/tools/*.py"))
|
||||
tools_modules = [
|
||||
str(i.relative_to(ROOT_DIR))[:-3].replace("/", ".")
|
||||
for i in tools_sources
|
||||
if i.stem not in unwanted_tools
|
||||
]
|
||||
create_one_file(
|
||||
"pytorch3d.implicitron.tools",
|
||||
"Tools for implicitron",
|
||||
tools_modules,
|
||||
DEST_DIR / "tools.rst",
|
||||
)
|
||||
|
||||
dataset_files = sorted(ROOT_DIR.glob("pytorch3d/implicitron/dataset/*.py"))
|
||||
basic_dataset = [
|
||||
"dataset_base",
|
||||
"dataset_map_provider",
|
||||
"data_loader_map_provider",
|
||||
"data_source",
|
||||
"scene_batch_sampler",
|
||||
]
|
||||
basic_dataset_modules = [
|
||||
f"pytorch3d.implicitron.dataset.{i}" for i in basic_dataset
|
||||
]
|
||||
create_one_file(
|
||||
"pytorch3d.implicitron.dataset in general",
|
||||
"Basics of data for implicitron",
|
||||
basic_dataset_modules,
|
||||
DEST_DIR / "data_basics.rst",
|
||||
)
|
||||
|
||||
specific_dataset_files = [
|
||||
i for i in dataset_files if i.stem.find("_dataset_map_provider") != -1
|
||||
]
|
||||
create_one_file(
|
||||
"pytorch3d.implicitron.dataset specific datasets",
|
||||
"specific datasets",
|
||||
paths_to_modules(specific_dataset_files),
|
||||
DEST_DIR / "datasets.rst",
|
||||
)
|
||||
|
||||
evaluation_files = sorted(ROOT_DIR.glob("pytorch3d/implicitron/evaluation/*.py"))
|
||||
create_one_file(
|
||||
"pytorch3d.implicitron.evaluation",
|
||||
"evaluation",
|
||||
paths_to_modules(evaluation_files),
|
||||
DEST_DIR / "evaluation.rst",
|
||||
)
|
||||
|
||||
make_directory_index("pytorch3d.implicitron", DEST_DIR)
|
||||
|
||||
|
||||
def iterate_toplevel_module(name: str) -> None:
|
||||
dest_dir = Path(__file__).resolve().parent / "modules" / name
|
||||
iterate_directory(ROOT_DIR / "pytorch3d" / name, dest_dir)
|
||||
|
||||
|
||||
do_implicitron()
|
||||
iterate_toplevel_module("renderer")
|
||||
iterate_toplevel_module("vis")
|
||||
@@ -1,29 +0,0 @@
|
||||
pytorch3d.implicitron.dataset in general
|
||||
========================================
|
||||
|
||||
Basics of data for implicitron
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.dataset.dataset_base
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.dataset.dataset_map_provider
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.dataset.data_loader_map_provider
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.dataset.data_source
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.dataset.scene_batch_sampler
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,29 +0,0 @@
|
||||
pytorch3d.implicitron.dataset specific datasets
|
||||
===============================================
|
||||
|
||||
specific datasets
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.dataset.blender_dataset_map_provider
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.dataset.json_index_dataset_map_provider
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.dataset.json_index_dataset_map_provider_v2
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.dataset.llff_dataset_map_provider
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.dataset.rendered_mesh_dataset_map_provider
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,14 +0,0 @@
|
||||
pytorch3d.implicitron.evaluation
|
||||
================================
|
||||
|
||||
evaluation
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.evaluation.evaluate_new_view_synthesis
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.evaluation.evaluator
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,10 +0,0 @@
|
||||
pytorch3d.implicitron
|
||||
=====================
|
||||
|
||||
.. toctree::
|
||||
|
||||
models/index.rst
|
||||
data_basics
|
||||
datasets
|
||||
evaluation
|
||||
tools
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.implicitron.models.base_model
|
||||
=======================================
|
||||
|
||||
base_model
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.models.base_model
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.implicitron.models.feature_extractor.feature_extractor
|
||||
================================================================
|
||||
|
||||
feature_extractor
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.models.feature_extractor.feature_extractor
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,7 +0,0 @@
|
||||
pytorch3d.implicitron.models.feature_extractor
|
||||
==============================================
|
||||
|
||||
.. toctree::
|
||||
|
||||
feature_extractor
|
||||
resnet_feature_extractor
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.implicitron.models.feature_extractor.resnet_feature_extractor
|
||||
=======================================================================
|
||||
|
||||
resnet_feature_extractor
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.models.feature_extractor.resnet_feature_extractor
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.implicitron.models.generic_model
|
||||
==========================================
|
||||
|
||||
generic_model
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.models.generic_model
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.implicitron.models.global_encoder.autodecoder
|
||||
=======================================================
|
||||
|
||||
autodecoder
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.models.global_encoder.autodecoder
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.implicitron.models.global_encoder.global_encoder
|
||||
==========================================================
|
||||
|
||||
global_encoder
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.models.global_encoder.global_encoder
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,7 +0,0 @@
|
||||
pytorch3d.implicitron.models.global_encoder
|
||||
===========================================
|
||||
|
||||
.. toctree::
|
||||
|
||||
autodecoder
|
||||
global_encoder
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.implicitron.models.implicit_function.base
|
||||
===================================================
|
||||
|
||||
base
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.models.implicit_function.base
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.implicitron.models.implicit_function.decoding_functions
|
||||
=================================================================
|
||||
|
||||
decoding_functions
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.models.implicit_function.decoding_functions
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.implicitron.models.implicit_function.idr_feature_field
|
||||
================================================================
|
||||
|
||||
idr_feature_field
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.models.implicit_function.idr_feature_field
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,13 +0,0 @@
|
||||
pytorch3d.implicitron.models.implicit_function
|
||||
==============================================
|
||||
|
||||
.. toctree::
|
||||
|
||||
base
|
||||
decoding_functions
|
||||
idr_feature_field
|
||||
neural_radiance_field
|
||||
scene_representation_networks
|
||||
utils
|
||||
voxel_grid
|
||||
voxel_grid_implicit_function
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.implicitron.models.implicit_function.neural_radiance_field
|
||||
====================================================================
|
||||
|
||||
neural_radiance_field
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.models.implicit_function.neural_radiance_field
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.implicitron.models.implicit_function.scene_representation_networks
|
||||
============================================================================
|
||||
|
||||
scene_representation_networks
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.models.implicit_function.scene_representation_networks
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.implicitron.models.implicit_function.utils
|
||||
====================================================
|
||||
|
||||
utils
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.models.implicit_function.utils
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.implicitron.models.implicit_function.voxel_grid
|
||||
=========================================================
|
||||
|
||||
voxel_grid
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.models.implicit_function.voxel_grid
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.implicitron.models.implicit_function.voxel_grid_implicit_function
|
||||
===========================================================================
|
||||
|
||||
voxel_grid_implicit_function
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.models.implicit_function.voxel_grid_implicit_function
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,15 +0,0 @@
|
||||
pytorch3d.implicitron.models
|
||||
============================
|
||||
|
||||
.. toctree::
|
||||
|
||||
base_model
|
||||
generic_model
|
||||
metrics
|
||||
model_dbir
|
||||
feature_extractor/index
|
||||
global_encoder/index
|
||||
implicit_function/index
|
||||
renderer/index
|
||||
view_pooler/index
|
||||
visualization/index
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.implicitron.models.metrics
|
||||
====================================
|
||||
|
||||
metrics
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.models.metrics
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.implicitron.models.model_dbir
|
||||
=======================================
|
||||
|
||||
model_dbir
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.models.model_dbir
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.implicitron.models.renderer.base
|
||||
==========================================
|
||||
|
||||
base
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.models.renderer.base
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,14 +0,0 @@
|
||||
pytorch3d.implicitron.models.renderer
|
||||
=====================================
|
||||
|
||||
.. toctree::
|
||||
|
||||
base
|
||||
lstm_renderer
|
||||
multipass_ea
|
||||
ray_point_refiner
|
||||
ray_sampler
|
||||
ray_tracing
|
||||
raymarcher
|
||||
rgb_net
|
||||
sdf_renderer
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.implicitron.models.renderer.lstm_renderer
|
||||
===================================================
|
||||
|
||||
lstm_renderer
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.models.renderer.lstm_renderer
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.implicitron.models.renderer.multipass_ea
|
||||
==================================================
|
||||
|
||||
multipass_ea
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.models.renderer.multipass_ea
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.implicitron.models.renderer.ray_point_refiner
|
||||
=======================================================
|
||||
|
||||
ray_point_refiner
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.models.renderer.ray_point_refiner
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.implicitron.models.renderer.ray_sampler
|
||||
=================================================
|
||||
|
||||
ray_sampler
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.models.renderer.ray_sampler
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.implicitron.models.renderer.ray_tracing
|
||||
=================================================
|
||||
|
||||
ray_tracing
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.models.renderer.ray_tracing
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.implicitron.models.renderer.raymarcher
|
||||
================================================
|
||||
|
||||
raymarcher
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.models.renderer.raymarcher
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.implicitron.models.renderer.rgb_net
|
||||
=============================================
|
||||
|
||||
rgb_net
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.models.renderer.rgb_net
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.implicitron.models.renderer.sdf_renderer
|
||||
==================================================
|
||||
|
||||
sdf_renderer
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.models.renderer.sdf_renderer
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.implicitron.models.view_pooler.feature_aggregator
|
||||
===========================================================
|
||||
|
||||
feature_aggregator
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.models.view_pooler.feature_aggregator
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,8 +0,0 @@
|
||||
pytorch3d.implicitron.models.view_pooler
|
||||
========================================
|
||||
|
||||
.. toctree::
|
||||
|
||||
feature_aggregator
|
||||
view_pooler
|
||||
view_sampler
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.implicitron.models.view_pooler.view_pooler
|
||||
====================================================
|
||||
|
||||
view_pooler
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.models.view_pooler.view_pooler
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.implicitron.models.view_pooler.view_sampler
|
||||
=====================================================
|
||||
|
||||
view_sampler
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.models.view_pooler.view_sampler
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,6 +0,0 @@
|
||||
pytorch3d.implicitron.models.visualization
|
||||
==========================================
|
||||
|
||||
.. toctree::
|
||||
|
||||
render_flyaround
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.implicitron.models.visualization.render_flyaround
|
||||
===========================================================
|
||||
|
||||
render_flyaround
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.models.visualization.render_flyaround
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,64 +0,0 @@
|
||||
pytorch3d.implicitron.tools
|
||||
===========================
|
||||
|
||||
Tools for implicitron
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.tools.camera_utils
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.tools.circle_fitting
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.tools.config
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.tools.eval_video_trajectory
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.tools.image_utils
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.tools.metric_utils
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.tools.model_io
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.tools.point_cloud_utils
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.tools.rasterize_mc
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.tools.stats
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.tools.video_writer
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
|
||||
.. automodule:: pytorch3d.implicitron.tools.vis_utils
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -12,5 +12,4 @@ API Documentation
|
||||
utils
|
||||
datasets
|
||||
common
|
||||
vis/index
|
||||
implicitron/index
|
||||
vis
|
||||
|
||||
@@ -1,9 +1,7 @@
|
||||
pytorch3d.renderer.blending
|
||||
===========================
|
||||
|
||||
blending
|
||||
===========================
|
||||
|
||||
.. automodule:: pytorch3d.renderer.blending
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.renderer.camera_conversions
|
||||
=====================================
|
||||
|
||||
camera_conversions
|
||||
|
||||
.. automodule:: pytorch3d.renderer.camera_conversions
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.renderer.camera_utils
|
||||
===============================
|
||||
|
||||
camera_utils
|
||||
|
||||
.. automodule:: pytorch3d.renderer.camera_utils
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +1,7 @@
|
||||
pytorch3d.renderer.cameras
|
||||
==========================
|
||||
|
||||
cameras
|
||||
===========================
|
||||
|
||||
.. automodule:: pytorch3d.renderer.cameras
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.renderer.compositing
|
||||
==============================
|
||||
|
||||
compositing
|
||||
|
||||
.. automodule:: pytorch3d.renderer.compositing
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.renderer.fisheyecameras
|
||||
=================================
|
||||
|
||||
fisheyecameras
|
||||
|
||||
.. automodule:: pytorch3d.renderer.fisheyecameras
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.renderer.implicit.harmonic_embedding
|
||||
==============================================
|
||||
|
||||
harmonic_embedding
|
||||
|
||||
.. automodule:: pytorch3d.renderer.implicit.harmonic_embedding
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,11 +0,0 @@
|
||||
pytorch3d.renderer.implicit
|
||||
===========================
|
||||
|
||||
.. toctree::
|
||||
|
||||
harmonic_embedding
|
||||
raymarching
|
||||
raysampling
|
||||
renderer
|
||||
sample_pdf
|
||||
utils
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.renderer.implicit.raymarching
|
||||
=======================================
|
||||
|
||||
raymarching
|
||||
|
||||
.. automodule:: pytorch3d.renderer.implicit.raymarching
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.renderer.implicit.raysampling
|
||||
=======================================
|
||||
|
||||
raysampling
|
||||
|
||||
.. automodule:: pytorch3d.renderer.implicit.raysampling
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.renderer.implicit.renderer
|
||||
====================================
|
||||
|
||||
renderer
|
||||
|
||||
.. automodule:: pytorch3d.renderer.implicit.renderer
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.renderer.implicit.sample_pdf
|
||||
======================================
|
||||
|
||||
sample_pdf
|
||||
|
||||
.. automodule:: pytorch3d.renderer.implicit.sample_pdf
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.renderer.implicit.utils
|
||||
=================================
|
||||
|
||||
utils
|
||||
|
||||
.. automodule:: pytorch3d.renderer.implicit.utils
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,19 +1,15 @@
|
||||
pytorch3d.renderer
|
||||
==================
|
||||
pytorch3d.renderer
|
||||
===========================
|
||||
|
||||
.. toctree::
|
||||
|
||||
blending
|
||||
camera_conversions
|
||||
camera_utils
|
||||
rasterizer
|
||||
cameras
|
||||
compositing
|
||||
fisheyecameras
|
||||
lighting
|
||||
materials
|
||||
splatter_blend
|
||||
utils
|
||||
implicit/index
|
||||
mesh/index
|
||||
opengl/index
|
||||
points/index
|
||||
texturing
|
||||
blending
|
||||
shading
|
||||
shader
|
||||
renderer
|
||||
utils
|
||||
@@ -1,9 +1,6 @@
|
||||
pytorch3d.renderer.lighting
|
||||
===========================
|
||||
|
||||
lighting
|
||||
===========================
|
||||
|
||||
.. automodule:: pytorch3d.renderer.lighting
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
:undoc-members:
|
||||
@@ -1,9 +1,7 @@
|
||||
pytorch3d.renderer.materials
|
||||
============================
|
||||
|
||||
materials
|
||||
===========================
|
||||
|
||||
.. automodule:: pytorch3d.renderer.materials
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.renderer.mesh.clip
|
||||
============================
|
||||
|
||||
clip
|
||||
|
||||
.. automodule:: pytorch3d.renderer.mesh.clip
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,13 +0,0 @@
|
||||
pytorch3d.renderer.mesh
|
||||
=======================
|
||||
|
||||
.. toctree::
|
||||
|
||||
clip
|
||||
rasterize_meshes
|
||||
rasterizer
|
||||
renderer
|
||||
shader
|
||||
shading
|
||||
textures
|
||||
utils
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.renderer.mesh.rasterize_meshes
|
||||
========================================
|
||||
|
||||
rasterize_meshes
|
||||
|
||||
.. automodule:: pytorch3d.renderer.mesh.rasterize_meshes
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.renderer.mesh.rasterizer
|
||||
==================================
|
||||
|
||||
rasterizer
|
||||
|
||||
.. automodule:: pytorch3d.renderer.mesh.rasterizer
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.renderer.mesh.renderer
|
||||
================================
|
||||
|
||||
renderer
|
||||
|
||||
.. automodule:: pytorch3d.renderer.mesh.renderer
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.renderer.mesh.textures
|
||||
================================
|
||||
|
||||
textures
|
||||
|
||||
.. automodule:: pytorch3d.renderer.mesh.textures
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.renderer.mesh.utils
|
||||
=============================
|
||||
|
||||
utils
|
||||
|
||||
.. automodule:: pytorch3d.renderer.mesh.utils
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,7 +0,0 @@
|
||||
pytorch3d.renderer.opengl
|
||||
=========================
|
||||
|
||||
.. toctree::
|
||||
|
||||
opengl_utils
|
||||
rasterizer_opengl
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.renderer.opengl.opengl_utils
|
||||
======================================
|
||||
|
||||
opengl_utils
|
||||
|
||||
.. automodule:: pytorch3d.renderer.opengl.opengl_utils
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.renderer.opengl.rasterizer_opengl
|
||||
===========================================
|
||||
|
||||
rasterizer_opengl
|
||||
|
||||
.. automodule:: pytorch3d.renderer.opengl.rasterizer_opengl
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.renderer.points.compositor
|
||||
====================================
|
||||
|
||||
compositor
|
||||
|
||||
.. automodule:: pytorch3d.renderer.points.compositor
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,10 +0,0 @@
|
||||
pytorch3d.renderer.points
|
||||
=========================
|
||||
|
||||
.. toctree::
|
||||
|
||||
compositor
|
||||
rasterize_points
|
||||
rasterizer
|
||||
renderer
|
||||
pulsar/index
|
||||
@@ -1,7 +0,0 @@
|
||||
pytorch3d.renderer.points.pulsar
|
||||
================================
|
||||
|
||||
.. toctree::
|
||||
|
||||
renderer
|
||||
unified
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.renderer.points.pulsar.renderer
|
||||
=========================================
|
||||
|
||||
renderer
|
||||
|
||||
.. automodule:: pytorch3d.renderer.points.pulsar.renderer
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.renderer.points.pulsar.unified
|
||||
========================================
|
||||
|
||||
unified
|
||||
|
||||
.. automodule:: pytorch3d.renderer.points.pulsar.unified
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.renderer.points.rasterize_points
|
||||
==========================================
|
||||
|
||||
rasterize_points
|
||||
|
||||
.. automodule:: pytorch3d.renderer.points.rasterize_points
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.renderer.points.rasterizer
|
||||
====================================
|
||||
|
||||
rasterizer
|
||||
|
||||
.. automodule:: pytorch3d.renderer.points.rasterizer
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.renderer.points.renderer
|
||||
==================================
|
||||
|
||||
renderer
|
||||
|
||||
.. automodule:: pytorch3d.renderer.points.renderer
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
10
docs/modules/renderer/rasterizer.rst
Normal file
10
docs/modules/renderer/rasterizer.rst
Normal file
@@ -0,0 +1,10 @@
|
||||
rasterizer
|
||||
===========================
|
||||
|
||||
.. automodule:: pytorch3d.renderer.mesh.rasterize_meshes
|
||||
:members:
|
||||
:undoc-members:
|
||||
|
||||
.. automodule:: pytorch3d.renderer.mesh.rasterizer
|
||||
:members:
|
||||
:undoc-members:
|
||||
@@ -1,9 +1,7 @@
|
||||
pytorch3d.renderer.mesh.shader
|
||||
==============================
|
||||
|
||||
shader
|
||||
===========================
|
||||
|
||||
.. automodule:: pytorch3d.renderer.mesh.shader
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
|
||||
@@ -1,9 +1,7 @@
|
||||
pytorch3d.renderer.mesh.shading
|
||||
===============================
|
||||
|
||||
shading
|
||||
===========================
|
||||
|
||||
.. automodule:: pytorch3d.renderer.mesh.shading
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.renderer.splatter_blend
|
||||
=================================
|
||||
|
||||
splatter_blend
|
||||
|
||||
.. automodule:: pytorch3d.renderer.splatter_blend
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
7
docs/modules/renderer/texturing.rst
Normal file
7
docs/modules/renderer/texturing.rst
Normal file
@@ -0,0 +1,7 @@
|
||||
texturing
|
||||
===========================
|
||||
|
||||
.. automodule:: pytorch3d.renderer.mesh.texturing
|
||||
:members:
|
||||
:undoc-members:
|
||||
|
||||
@@ -1,9 +1,6 @@
|
||||
pytorch3d.renderer.utils
|
||||
========================
|
||||
|
||||
utils
|
||||
===========================
|
||||
|
||||
.. automodule:: pytorch3d.renderer.utils
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
:undoc-members:
|
||||
6
docs/modules/vis.rst
Normal file
6
docs/modules/vis.rst
Normal file
@@ -0,0 +1,6 @@
|
||||
pytorch3d.vis
|
||||
===========================
|
||||
|
||||
.. automodule:: pytorch3d.vis
|
||||
:members:
|
||||
:undoc-members:
|
||||
@@ -1,7 +0,0 @@
|
||||
pytorch3d.vis
|
||||
=============
|
||||
|
||||
.. toctree::
|
||||
|
||||
plotly_vis
|
||||
texture_vis
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.vis.plotly_vis
|
||||
========================
|
||||
|
||||
plotly_vis
|
||||
|
||||
.. automodule:: pytorch3d.vis.plotly_vis
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -1,9 +0,0 @@
|
||||
pytorch3d.vis.texture_vis
|
||||
=========================
|
||||
|
||||
texture_vis
|
||||
|
||||
.. automodule:: pytorch3d.vis.texture_vis
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
@@ -62,7 +62,7 @@ Below we compare the performance for Objectron (in C++) and our algorithm, in C+
|
||||
```python
|
||||
from pytorch3d.ops import box3d_overlap
|
||||
# Assume inputs: boxes1 (M, 8, 3) and boxes2 (N, 8, 3)
|
||||
intersection_vol, iou_3d = box3d_overlap(boxes1, boxes2)
|
||||
intersection_vol, iou_3d = box3d_overal(boxes1, boxes2)
|
||||
```
|
||||
|
||||
For more details, read [iou_box3d.py](https://github.com/facebookresearch/pytorch3d/blob/main/pytorch3d/ops/iou_box3d.py).
|
||||
|
||||
@@ -5,8 +5,7 @@ sphinx_rtd_theme
|
||||
sphinx_markdown_tables
|
||||
mock
|
||||
numpy
|
||||
iopath
|
||||
fvcore
|
||||
git+git://github.com/facebookresearch/fvcore.git
|
||||
git+git://github.com/facebookresearch/iopath.git
|
||||
https://download.pytorch.org/whl/cpu/torchvision-0.8.2%2Bcpu-cp37-cp37m-linux_x86_64.whl
|
||||
https://download.pytorch.org/whl/cpu/torch-1.7.1%2Bcpu-cp37-cp37m-linux_x86_64.whl
|
||||
omegaconf
|
||||
|
||||
@@ -89,7 +89,7 @@
|
||||
"except ModuleNotFoundError:\n",
|
||||
" need_pytorch3d=True\n",
|
||||
"if need_pytorch3d:\n",
|
||||
" if torch.__version__.startswith(\"1.13.\") and sys.platform.startswith(\"linux\"):\n",
|
||||
" if torch.__version__.startswith(\"1.11.\") and sys.platform.startswith(\"linux\"):\n",
|
||||
" # We try to install PyTorch3D via a released wheel.\n",
|
||||
" pyt_version_str=torch.__version__.split(\"+\")[0].replace(\".\", \"\")\n",
|
||||
" version_str=\"\".join([\n",
|
||||
|
||||
@@ -76,7 +76,7 @@
|
||||
"except ModuleNotFoundError:\n",
|
||||
" need_pytorch3d=True\n",
|
||||
"if need_pytorch3d:\n",
|
||||
" if torch.__version__.startswith(\"1.13.\") and sys.platform.startswith(\"linux\"):\n",
|
||||
" if torch.__version__.startswith(\"1.11.\") and sys.platform.startswith(\"linux\"):\n",
|
||||
" # We try to install PyTorch3D via a released wheel.\n",
|
||||
" pyt_version_str=torch.__version__.split(\"+\")[0].replace(\".\", \"\")\n",
|
||||
" version_str=\"\".join([\n",
|
||||
|
||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user