Structural Affinity

Reference implementation of Structural Affinity: An Algorithm for Coherent Cage-Based Shape Design (Aviraj Newatia, Shengan Yang, Hantao Zhong, Rika Antonova, University of Cambridge).

Project page · arXiv · Paper (PDF)

Summary

In cage-based deformation (CBD), each handle of the control cage moves independently. This can cause spiky, distorted and self-intersecting shapes.

Structural Affinity couples the handle motion with a second cage:

• The inner cage is the control cage of the CBD pipeline. It has N handles.
• The outer cage is a coarser convex cage that strictly encloses the inner cage. It has M vertices.

The outer cage gives an affinity matrix C (N × N). C tells how much each handle follows the other handles and is calculated once, offline. At deformation time, the method does one sparse matrix product.

Requirements

• Python 3.10 or later.
• numpy and scipy.
• Optional: PyTorch, for structural_affinity_torch.py. matplotlib, for the examples.

The implementation is one file: structural_affinity.py. Copy this file into your project. For a PyTorch pipeline, also copy structural_affinity_torch.py.

Quick start

1. Make the outer cage from the inner cage:

   import structural_affinity as sa
   
   outer_verts, outer_faces = sa.icosahedral_outer_cage(cage_verts)
   

2. Calculate the affinity. Do this step one time for each inner cage:

   affinity = sa.StructuralAffinity(cage_verts, outer_verts, outer_faces, alpha=0.3, density=0.4)
   

3. Apply the affinity to the handle displacements D (N, 3):

   deformed_cage = cage_verts + affinity.propagate(D)
   

Outer cage

The outer cage must:

• Be convex.
• Strictly enclose all handles of the inner cage.

If the outer cage does not obey these rules, StructuralAffinity raises ValueError.

Three functions make an outer cage. Each function returns (outer_verts, outer_faces):

function	vertices	paper experiment
icosahedral_outer_cage(cage_verts, padding=0.1)	12 in 3D, 7 in 2D	Neural Cages
box_outer_cage(cage_verts, padding=0.05)	8 in 3D, 4 in 2D	Robot fingertips
convex_outer_cage(cage_verts, padding=0.05)	the hull vertices of the inner cage	Wing optimization

A custom outer cage can also be made as long as it satisfies the requirements above, these are just our provided implementations.

Note: for a convex inner cage, convex_outer_cage keeps all inner handles. The outer cage is then not coarser than the inner cage.

To use your own outer cage, supply the arrays directly:

affinity = sa.StructuralAffinity(cage_verts, my_outer_verts, my_outer_faces, alpha=0.3, density=0.4)

• my_outer_verts: shape (M, 3).
• my_outer_faces: shape (F, 3). The faces must have a consistent orientation.

The outer cage controls which handles move together. The result of the method depends on this choice.

Parameters

parameter	range	effect
alpha	≥ 0	Affinity strength. 0 gives plain CBD. A larger value makes the handles move more together.
density	[0, 1]	Fraction of off-diagonal pairs in C to keep, largest values first. 0 removes all coupling. 1 keeps C dense.

Start with alpha in [0.1, 0.45] and density in [0.2, 0.7].

Values in the paper:

experiment	alpha	density	outer cage
Neural Cages (chair / table / car)	0.3 / 0.2 / 0.4	0.4	icosahedral_outer_cage
Robot fingertips	0.35	0.15	box_outer_cage, padding 0.05
Wing, perturbed starts	0.05	0.025	convex_outer_cage

Integration into a CBD pipeline

Structural Affinity changes only the step that adds the displacements to the inner cage:

# before
deformed_cage = cage_verts + D
# after
deformed_cage = cage_verts + affinity.propagate(D)

propagate also accepts a batch with shape (B, N, d).

To save C and use it again, use StructuralAffinity.from_affinity_matrix(C, alpha).

PyTorch

AffinityPropagation is a torch.nn.Module. Its gradients go through the propagation to the offsets.

from structural_affinity_torch import AffinityPropagation

propagate = AffinityPropagation(affinity.C, alpha=affinity.alpha).to(device)
deformed_cage = cage + propagate(net(x))        # net(x) gives the offsets, shape (B, N, 3)

Method

step	time	operation	function
1	offline	Make a convex outer cage that strictly encloses the inner cage.	icosahedral_outer_cage, box_outer_cage, convex_outer_cage
2	offline	Calculate the mean-value coordinates W (N, M) of each inner handle in the outer cage.	mvc
3	offline	Calculate C_ij = Σ_k min(w_ik, w_jk). C is symmetric, in [0, 1], with a diagonal of 1.	affinity_matrix
4	offline	Keep the diagonal and the largest off-diagonal entries of C.	sparsify
5	each step	Calculate ΔV = A D, with A = (1 − α) I + α Cᵀ.	propagate

Steps 2 to 4 are performed in the constructor of StructuralAffinity constructor.

Properties:

• A handle with displacement D_i moves by D_i.
• Each other handle j also moves by α · C_ij · D_i.
• The cost of step 5 is O(N + L). L is the number of off-diagonal entries in C.
• For a dense C and α < 1, A is invertible. Thus all cage motions are possible. Structural Affinity changes which motions are easy to find. It does not remove motions.

以上为 README 的开头部分。其余内容请阅读原文。 查看完整 README ↗