liyang
ae28d55f81
Source: RGB-D -> Dyn-HaMR -> L20 retargeting -> FoundationPose -> reference repair -> SPIDER, documented in docs/PIPELINE_LATEST.md and docs/SETUP_AND_WEIGHTS.md. Adds FoundationPose and nvdiffrast upstream snapshots, requirements/pipeline_venv.txt and the FoundationPose weight manifest/downloader. Assets (Git LFS): weights/ (WiLoR detector, HandFlow denoiser, UniDepth-L), FoundationPose checkpoints, HaMeR checkpoint, Dyn-HaMR HMP model and BMC constraints, L20 URDF/meshes, the 20260915_171525 D405 recording and the two box CADs. MANO models are not redistributed (third_party/hamer/_DATA/data/mano/README.txt). Environments, caches and run outputs excluded. Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
56 lines
2.2 KiB
Python
56 lines
2.2 KiB
Python
# Copyright (c) 2020, NVIDIA CORPORATION. All rights reserved.
|
|
#
|
|
# NVIDIA CORPORATION and its licensors retain all intellectual property
|
|
# and proprietary rights in and to this software, related documentation
|
|
# and any modifications thereto. Any use, reproduction, disclosure or
|
|
# distribution of this software and related documentation without an express
|
|
# license agreement from NVIDIA CORPORATION is strictly prohibited.
|
|
|
|
import setuptools
|
|
import os
|
|
|
|
# Print an error message if there's no PyTorch installed.
|
|
try:
|
|
from torch.utils.cpp_extension import BuildExtension, CUDAExtension
|
|
except ImportError:
|
|
# This happens if the user runs 'pip install' with default build isolation
|
|
# OR if they simply don't have torch installed at all.
|
|
print("\n\n" + "*" * 70)
|
|
print("ERROR! Cannot compile nvdiffrast CUDA extension. Please ensure that:\n")
|
|
print("1. You have PyTorch installed")
|
|
print("2. You run 'pip install' with --no-build-isolation flag")
|
|
print("*" * 70 + "\n\n")
|
|
exit(1)
|
|
|
|
setuptools.setup(
|
|
ext_modules=[
|
|
CUDAExtension(
|
|
"_nvdiffrast_c",
|
|
sources=[
|
|
"csrc/common/antialias.cu",
|
|
"csrc/common/common.cpp",
|
|
"csrc/common/cudaraster/impl/Buffer.cpp",
|
|
"csrc/common/cudaraster/impl/CudaRaster.cpp",
|
|
"csrc/common/cudaraster/impl/RasterImpl.cpp",
|
|
"csrc/common/cudaraster/impl/RasterImpl_kernel.cu",
|
|
"csrc/common/interpolate.cu",
|
|
"csrc/common/rasterize.cu",
|
|
"csrc/common/texture.cpp",
|
|
"csrc/common/texture_kernel.cu",
|
|
"csrc/torch/torch_antialias.cpp",
|
|
"csrc/torch/torch_bindings.cpp",
|
|
"csrc/torch/torch_interpolate.cpp",
|
|
"csrc/torch/torch_rasterize.cpp",
|
|
"csrc/torch/torch_texture.cpp",
|
|
],
|
|
extra_compile_args={
|
|
"cxx": ["-DNVDR_TORCH"]
|
|
# Disable warnings in torch headers.
|
|
+ (["/wd4067", "/wd4624", "/wd4996"] if os.name == "nt" else []),
|
|
"nvcc": ["-DNVDR_TORCH", "-lineinfo"],
|
|
},
|
|
)
|
|
],
|
|
cmdclass={"build_ext": BuildExtension},
|
|
)
|