pyfaceau 1.3.9__tar.gz → 1.3.10__tar.gz

{pyfaceau-1.3.9 → pyfaceau-1.3.10}/MANIFEST.in

@@ -21,9 +21,9 @@ global-exclude *.pyo
 global-exclude __pycache__
 global-exclude .DS_Store
 
-# Exclude weights (too large for PyPI, users must download separately)
-exclude weights/*
-recursive-exclude weights *
+# Include weights for git-based installs (but these are excluded from PyPI sdist)
+# Users installing from git will get weights; PyPI users must download separately
+recursive-include weights *.txt *.dat
 
 # Exclude development files
 exclude .gitignore

pyfaceau-1.3.10/PKG-INFO

@@ -0,0 +1,243 @@
+Metadata-Version: 2.4
+Name: pyfaceau
+Version: 1.3.10
+Summary: Pure Python OpenFace 2.2 AU extraction with CLNF landmark refinement
+Home-page: https://github.com/johnwilsoniv/face-analysis
+Author: John Wilson
+Author-email: 
+License: CC BY-NC 4.0
+Project-URL: Homepage, https://github.com/johnwilsoniv/pyfaceau
+Project-URL: Documentation, https://github.com/johnwilsoniv/pyfaceau
+Project-URL: Repository, https://github.com/johnwilsoniv/pyfaceau
+Project-URL: Bug Tracker, https://github.com/johnwilsoniv/pyfaceau/issues
+Keywords: facial-action-units,openface,computer-vision,facial-analysis,emotion-recognition
+Classifier: Development Status :: 5 - Production/Stable
+Classifier: Intended Audience :: Science/Research
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Scientific/Engineering :: Image Recognition
+Classifier: License :: Other/Proprietary License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: numpy>=1.20.0
+Requires-Dist: opencv-python>=4.5.0
+Requires-Dist: pandas>=1.3.0
+Requires-Dist: onnxruntime>=1.10.0
+Requires-Dist: scipy>=1.7.0
+Requires-Dist: scikit-learn>=1.0.0
+Requires-Dist: tqdm>=4.62.0
+Requires-Dist: numba>=0.56.0
+Requires-Dist: pyfhog>=0.1.0
+Requires-Dist: pyclnf>=0.2.0
+Requires-Dist: pymtcnn>=0.1.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+Requires-Dist: black>=22.0.0; extra == "dev"
+Requires-Dist: flake8>=4.0.0; extra == "dev"
+Provides-Extra: accel
+Requires-Dist: onnxruntime-coreml>=1.10.0; extra == "accel"
+Provides-Extra: training
+Requires-Dist: torch>=2.0.0; extra == "training"
+Requires-Dist: h5py>=3.0.0; extra == "training"
+Dynamic: home-page
+Dynamic: license-file
+Dynamic: requires-python
+
+# pyfaceau
+
+A python-based implementation of OpenFace 2.2's Facial Action Unit extraction pipeline with an accurate dlib substitute (ptmtcnn, pyclnf).
+
+**Accuracy: r = 0.97 correlation with C++ OpenFace 2.2**
+
+## Installation
+
+### Option 1: Install from GitHub (Recommended for Development)
+
+This includes model weights:
+
+```bash
+# Clone repository with weights
+git clone https://github.com/johnwilsoniv/pyfaceau.git
+cd pyfaceau
+
+# Create conda environment (recommended)
+conda create -n pyfaceau python=3.11
+conda activate pyfaceau
+
+# Install in editable mode
+pip install -e .
+```
+
+### Option 2: Install from PyPI
+
+```bash
+pip install pyfaceau
+```
+
+This automatically installs dependencies:
+- [pyclnf](https://github.com/johnwilsoniv/pyclnf) - Facial landmark detection (68 points)
+- [pymtcnn](https://github.com/johnwilsoniv/pymtcnn) - Face detection
+- [pyfhog](https://github.com/johnwilsoniv/pyfhog) - FHOG feature extraction
+
+### Model Weights
+
+Model weights (~50MB) are downloaded automatically on first use. If auto-download fails, download manually:
+
+```bash
+# Download weights
+python -m pyfaceau.download_weights
+
+# Or after pip install
+pyfaceau-download-weights
+```
+
+Weights are stored in `~/.pyfaceau/weights/`. You can customize this location:
+
+```bash
+# Set custom weights directory
+export PYFACEAU_WEIGHTS_DIR=/path/to/your/weights
+```
+
+### Troubleshooting: "PDM file not found"
+
+If you see this error:
+1. **First try**: Run `python -m pyfaceau.download_weights`
+2. **Manual fix**: Copy the `weights/` folder from the GitHub repo to `~/.pyfaceau/weights/`
+3. **Alternative**: Set `PYFACEAU_WEIGHTS_DIR` environment variable to point to your weights
+
+## Quick Start
+
+### Video Processing (Recommended)
+
+```python
+from pyfaceau import OpenFaceProcessor
+
+# Initialize processor
+processor = OpenFaceProcessor(verbose=True)
+
+# Process video to CSV (same format as OpenFace)
+processor.process_video("input.mp4", "output.csv")
+```
+
+### Batch Processing
+
+```python
+from pyfaceau import process_videos
+
+# Process all videos in a directory
+process_videos(
+    directory_path="/path/to/videos",
+    output_dir="/path/to/output"
+)
+```
+
+### Frame-by-Frame Processing
+
+```python
+from pyfaceau import FullPythonAUPipeline
+from pathlib import Path
+import cv2
+
+# Initialize pipeline with model paths
+weights_dir = Path("weights")
+pipeline = FullPythonAUPipeline(
+    pdm_file=str(weights_dir / "In-the-wild_aligned_PDM_68.txt"),
+    au_models_dir=str(weights_dir / "AU_predictors"),
+    triangulation_file=str(weights_dir / "tris_68_full.txt"),
+    patch_expert_file=str(weights_dir / "svr_patches_0.25_general.txt")
+)
+
+# Process single frame
+image = cv2.imread("face.jpg")
+image_rgb = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+
+result = pipeline.process_frame(image_rgb, frame_num=0)
+
+if result['success']:
+    print("AU intensities:", result['au_intensities'])
+    print("Landmarks shape:", result['landmarks_2d'].shape)  # (68, 2)
+    print("Pose (pitch, yaw, roll):", result['pose'])
+```
+
+## Output Format
+
+### CSV Output Columns
+
+The output CSV matches OpenFace format:
+- `frame` - Frame number
+- `timestamp` - Time in seconds
+- `confidence` - Detection confidence
+- `success` - Whether face was detected
+- `AU01_r` through `AU45_r` - AU intensities (0.0 - 5.0)
+- `pose_Rx`, `pose_Ry`, `pose_Rz` - Head pose in radians
+- `x_0` through `x_67`, `y_0` through `y_67` - 68 landmark coordinates
+
+### Action Units
+
+17 facial action units with intensity values (0.0 - 5.0):
+
+| AU | Description |
+|----|-------------|
+| AU01 | Inner Brow Raiser |
+| AU02 | Outer Brow Raiser |
+| AU04 | Brow Lowerer |
+| AU05 | Upper Lid Raiser |
+| AU06 | Cheek Raiser |
+| AU07 | Lid Tightener |
+| AU09 | Nose Wrinkler |
+| AU10 | Upper Lip Raiser |
+| AU12 | Lip Corner Puller |
+| AU14 | Dimpler |
+| AU15 | Lip Corner Depressor |
+| AU17 | Chin Raiser |
+| AU20 | Lip Stretcher |
+| AU23 | Lip Tightener |
+| AU25 | Lips Part |
+| AU26 | Jaw Drop |
+| AU45 | Blink |
+
+## Accuracy
+
+Validated against C++ OpenFace 2.2
+
+| Metric | Correlation |
+|--------|-------------|
+| **Overall Mean** | r = 0.97 |
+| **Overall Median** | r = 0.996 |
+| Static AUs | r = 0.98 |
+| Dynamic AUs | r = 0.96 |
+
+Per-AU correlations:
+- AU01: 0.997, AU02: 0.999, AU04: 0.989, AU05: 0.999
+- AU06: 0.999, AU07: 0.996, AU09: 0.997, AU10: 0.994
+- AU12: 0.998, AU14: 0.974, AU15: 0.893, AU17: 0.948
+- AU20: 0.817, AU23: 0.996, AU25: 0.984, AU26: 0.902, AU45: 0.998
+
+## Requirements
+
+- Python 3.8+
+- numpy
+- opencv-python
+- torch
+- scipy
+
+## Acknowledgments
+
+Based on OpenFace 2.2:
+
+> Baltrusaitis, T., Zadeh, A., Lim, Y. C., & Morency, L. P. (2018). OpenFace 2.0: Facial Behavior Analysis Toolkit. IEEE International Conference on Automatic Face and Gesture Recognition.
+
+## Citation
+
+If you use this in research, please cite:
+
+> Wilson IV, J., Rosenberg, J., Gray, M. L., & Razavi, C. R. (2025). A split-face computer vision/machine learning assessment of facial paralysis using facial action units. *Facial Plastic Surgery & Aesthetic Medicine*. https://doi.org/10.1177/26893614251394382
+
+## License
+
+CC BY-NC 4.0 - Free for non-commercial use with attribution.

pyfaceau-1.3.10/README.md

@@ -0,0 +1,193 @@
+# pyfaceau
+
+A python-based implementation of OpenFace 2.2's Facial Action Unit extraction pipeline with an accurate dlib substitute (ptmtcnn, pyclnf).
+
+**Accuracy: r = 0.97 correlation with C++ OpenFace 2.2**
+
+## Installation
+
+### Option 1: Install from GitHub (Recommended for Development)
+
+This includes model weights:
+
+```bash
+# Clone repository with weights
+git clone https://github.com/johnwilsoniv/pyfaceau.git
+cd pyfaceau
+
+# Create conda environment (recommended)
+conda create -n pyfaceau python=3.11
+conda activate pyfaceau
+
+# Install in editable mode
+pip install -e .
+```
+
+### Option 2: Install from PyPI
+
+```bash
+pip install pyfaceau
+```
+
+This automatically installs dependencies:
+- [pyclnf](https://github.com/johnwilsoniv/pyclnf) - Facial landmark detection (68 points)
+- [pymtcnn](https://github.com/johnwilsoniv/pymtcnn) - Face detection
+- [pyfhog](https://github.com/johnwilsoniv/pyfhog) - FHOG feature extraction
+
+### Model Weights
+
+Model weights (~50MB) are downloaded automatically on first use. If auto-download fails, download manually:
+
+```bash
+# Download weights
+python -m pyfaceau.download_weights
+
+# Or after pip install
+pyfaceau-download-weights
+```
+
+Weights are stored in `~/.pyfaceau/weights/`. You can customize this location:
+
+```bash
+# Set custom weights directory
+export PYFACEAU_WEIGHTS_DIR=/path/to/your/weights
+```
+
+### Troubleshooting: "PDM file not found"
+
+If you see this error:
+1. **First try**: Run `python -m pyfaceau.download_weights`
+2. **Manual fix**: Copy the `weights/` folder from the GitHub repo to `~/.pyfaceau/weights/`
+3. **Alternative**: Set `PYFACEAU_WEIGHTS_DIR` environment variable to point to your weights
+
+## Quick Start
+
+### Video Processing (Recommended)
+
+```python
+from pyfaceau import OpenFaceProcessor
+
+# Initialize processor
+processor = OpenFaceProcessor(verbose=True)
+
+# Process video to CSV (same format as OpenFace)
+processor.process_video("input.mp4", "output.csv")
+```
+
+### Batch Processing
+
+```python
+from pyfaceau import process_videos
+
+# Process all videos in a directory
+process_videos(
+    directory_path="/path/to/videos",
+    output_dir="/path/to/output"
+)
+```
+
+### Frame-by-Frame Processing
+
+```python
+from pyfaceau import FullPythonAUPipeline
+from pathlib import Path
+import cv2
+
+# Initialize pipeline with model paths
+weights_dir = Path("weights")
+pipeline = FullPythonAUPipeline(
+    pdm_file=str(weights_dir / "In-the-wild_aligned_PDM_68.txt"),
+    au_models_dir=str(weights_dir / "AU_predictors"),
+    triangulation_file=str(weights_dir / "tris_68_full.txt"),
+    patch_expert_file=str(weights_dir / "svr_patches_0.25_general.txt")
+)
+
+# Process single frame
+image = cv2.imread("face.jpg")
+image_rgb = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+
+result = pipeline.process_frame(image_rgb, frame_num=0)
+
+if result['success']:
+    print("AU intensities:", result['au_intensities'])
+    print("Landmarks shape:", result['landmarks_2d'].shape)  # (68, 2)
+    print("Pose (pitch, yaw, roll):", result['pose'])
+```
+
+## Output Format
+
+### CSV Output Columns
+
+The output CSV matches OpenFace format:
+- `frame` - Frame number
+- `timestamp` - Time in seconds
+- `confidence` - Detection confidence
+- `success` - Whether face was detected
+- `AU01_r` through `AU45_r` - AU intensities (0.0 - 5.0)
+- `pose_Rx`, `pose_Ry`, `pose_Rz` - Head pose in radians
+- `x_0` through `x_67`, `y_0` through `y_67` - 68 landmark coordinates
+
+### Action Units
+
+17 facial action units with intensity values (0.0 - 5.0):
+
+| AU | Description |
+|----|-------------|
+| AU01 | Inner Brow Raiser |
+| AU02 | Outer Brow Raiser |
+| AU04 | Brow Lowerer |
+| AU05 | Upper Lid Raiser |
+| AU06 | Cheek Raiser |
+| AU07 | Lid Tightener |
+| AU09 | Nose Wrinkler |
+| AU10 | Upper Lip Raiser |
+| AU12 | Lip Corner Puller |
+| AU14 | Dimpler |
+| AU15 | Lip Corner Depressor |
+| AU17 | Chin Raiser |
+| AU20 | Lip Stretcher |
+| AU23 | Lip Tightener |
+| AU25 | Lips Part |
+| AU26 | Jaw Drop |
+| AU45 | Blink |
+
+## Accuracy
+
+Validated against C++ OpenFace 2.2
+
+| Metric | Correlation |
+|--------|-------------|
+| **Overall Mean** | r = 0.97 |
+| **Overall Median** | r = 0.996 |
+| Static AUs | r = 0.98 |
+| Dynamic AUs | r = 0.96 |
+
+Per-AU correlations:
+- AU01: 0.997, AU02: 0.999, AU04: 0.989, AU05: 0.999
+- AU06: 0.999, AU07: 0.996, AU09: 0.997, AU10: 0.994
+- AU12: 0.998, AU14: 0.974, AU15: 0.893, AU17: 0.948
+- AU20: 0.817, AU23: 0.996, AU25: 0.984, AU26: 0.902, AU45: 0.998
+
+## Requirements
+
+- Python 3.8+
+- numpy
+- opencv-python
+- torch
+- scipy
+
+## Acknowledgments
+
+Based on OpenFace 2.2:
+
+> Baltrusaitis, T., Zadeh, A., Lim, Y. C., & Morency, L. P. (2018). OpenFace 2.0: Facial Behavior Analysis Toolkit. IEEE International Conference on Automatic Face and Gesture Recognition.
+
+## Citation
+
+If you use this in research, please cite:
+
+> Wilson IV, J., Rosenberg, J., Gray, M. L., & Razavi, C. R. (2025). A split-face computer vision/machine learning assessment of facial paralysis using facial action units. *Facial Plastic Surgery & Aesthetic Medicine*. https://doi.org/10.1177/26893614251394382
+
+## License
+
+CC BY-NC 4.0 - Free for non-commercial use with attribution.

pyfaceau-1.3.10/pyfaceau/__init__.py

@@ -0,0 +1,46 @@
+"""
+pyfaceau - Pure Python OpenFace 2.2 AU Extraction
+
+A complete Python implementation of OpenFace 2.2's AU extraction pipeline
+with high-performance parallel processing support and CLNF landmark refinement.
+"""
+
+__version__ = "1.3.10"
+
+# Weight management functions can be imported without heavy dependencies
+from .download_weights import (
+    download_weights,
+    ensure_weights,
+    get_weights_dir,
+    weights_exist
+)
+
+
+def __getattr__(name):
+    """Lazy import of heavy modules to speed up package loading."""
+    if name == 'FullPythonAUPipeline':
+        from .pipeline import FullPythonAUPipeline
+        return FullPythonAUPipeline
+    elif name == 'ParallelAUPipeline':
+        from .parallel_pipeline import ParallelAUPipeline
+        return ParallelAUPipeline
+    elif name == 'OpenFaceProcessor':
+        from .processor import OpenFaceProcessor
+        return OpenFaceProcessor
+    elif name == 'process_videos':
+        from .processor import process_videos
+        return process_videos
+    raise AttributeError(f"module 'pyfaceau' has no attribute '{name}'")
+
+
+__all__ = [
+    'FullPythonAUPipeline',
+    'ParallelAUPipeline',
+    'OpenFaceProcessor',
+    'process_videos',
+    # Weight management
+    'download_weights',
+    'ensure_weights',
+    'get_weights_dir',
+    'weights_exist',
+]