Add inline quaternion functions into app_new.py

README.md

@@ -12,4 +12,16 @@ python_version: "3.10"
 
 # MoMask: Text-to-Motion Generation
 
-Generate 3D human skeleton animations from text descriptions.
+Generate 3D human skeleton animations from text descriptions using [MoMask](https://github.com/EricGuo5513/momask-codes).
+
+## Model Architecture (ONNX INT8, ~151MB total)
+| Model | Size | Precision |
+|-------|------|-----------|
+| CLIP Text Encoder | 62MB | INT8 |
+| Mask Transformer | 20MB | INT8 |
+| Residual Transformer | 20MB | INT8 |
+| VQ-VAE Decoder | 44MB | FP32 |
+| Length Estimator | 441KB | INT8 |
+
+## Usage
+Enter a text description and click Generate.

app_new.py

@@ -17,9 +17,28 @@ from matplotlib.animation import FuncAnimation
 import mpl_toolkits.mplot3d.axes3d as p3
 from mpl_toolkits.mplot3d.art3d import Poly3DCollection
 from pathlib import Path
-# Add parent to path for imports
-sys.path.insert(0, str(Path(__file__).parent))
-from common.quaternion import qinv, qrot
+# ============ Quaternion Operations ============
+def qinv(q):
+    """Invert quaternion"""
+    assert q.shape[-1] == 4
+    mask = torch.ones_like(q)
+    mask[..., 1:] = -mask[..., 1:]
+    return q * mask
+
+def qrot(q, v):
+    """Rotate vector(s) v by quaternion(s) q"""
+    assert q.shape[-1] == 4
+    assert v.shape[-1] == 3
+    assert q.shape[:-1] == v.shape[:-1]
+    
+    original_shape = list(v.shape)
+    q = q.contiguous().view(-1, 4)
+    v = v.contiguous().view(-1, 3)
+    
+    qvec = q[:, 1:]
+    uv = torch.cross(qvec, v, dim=1)
+    uuv = torch.cross(qvec, uv, dim=1)
+    return (v + 2 * (q[:, :1] * uv + uuv)).view(original_shape)
 
 # ============ Configuration ============
 ONNX_DIR = Path(__file__).parent / "onnx_models"