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>
50 lines
2.8 KiB
C
50 lines
2.8 KiB
C
// 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.
|
|
|
|
#pragma once
|
|
|
|
//------------------------------------------------------------------------
|
|
// Constants and helpers.
|
|
|
|
#define IP_FWD_MAX_KERNEL_BLOCK_WIDTH 8
|
|
#define IP_FWD_MAX_KERNEL_BLOCK_HEIGHT 8
|
|
#define IP_GRAD_MAX_KERNEL_BLOCK_WIDTH 8
|
|
#define IP_GRAD_MAX_KERNEL_BLOCK_HEIGHT 8
|
|
#define IP_MAX_DIFF_ATTRS 32
|
|
|
|
//------------------------------------------------------------------------
|
|
// CUDA kernel params.
|
|
|
|
struct InterpolateKernelParams
|
|
{
|
|
const int* tri; // Incoming triangle buffer.
|
|
const float* attr; // Incoming attribute buffer.
|
|
const float* rast; // Incoming rasterizer output buffer.
|
|
const float* rastDB; // Incoming rasterizer output buffer for bary derivatives.
|
|
const float* dy; // Incoming attribute gradients.
|
|
const float* dda; // Incoming attr diff gradients.
|
|
float* out; // Outgoing interpolated attributes.
|
|
float* outDA; // Outgoing texcoord major axis lengths.
|
|
float* gradAttr; // Outgoing attribute gradients.
|
|
float* gradRaster; // Outgoing rasterizer gradients.
|
|
float* gradRasterDB; // Outgoing rasterizer bary diff gradients.
|
|
int numTriangles; // Number of triangles.
|
|
int numVertices; // Number of vertices.
|
|
int numAttr; // Number of total vertex attributes.
|
|
int numDiffAttr; // Number of attributes to differentiate.
|
|
int width; // Image width.
|
|
int height; // Image height.
|
|
int depth; // Minibatch size.
|
|
int attrBC; // 0=normal, 1=attr is broadcast.
|
|
int instance_mode; // 0=normal, 1=instance mode.
|
|
int diff_attrs_all; // 0=normal, 1=produce pixel differentials for all attributes.
|
|
int diffAttrs[IP_MAX_DIFF_ATTRS]; // List of attributes to differentiate.
|
|
};
|
|
|
|
//------------------------------------------------------------------------
|