✨ Features: - Complete 6-node suite for 3D face reconstruction - FLAME model integration with multi-constraint optimization - UV coordinate and surface normal prediction - Professional mesh export (OBJ, PLY, STL) - Comprehensive documentation and examples 🎯 Nodes included: - 🔧 Pixel3DMM Model Loader - 🎭 3D Face Reconstructor - 🗺️ UV Coordinate Predictor - 📐 Surface Normal Predictor - 🔥 FLAME Parameter Optimizer - 📦 3D Mesh Exporter 📚 Documentation: - Complete installation guide - Usage examples and workflows - Troubleshooting and support - Ready-to-use ComfyUI workflow files 🧪 Quality: - Comprehensive testing and validation - ComfyUI standard compliance - Cross-platform support (Windows, macOS, Linux) - Production-ready with robust error handling Generated by MCP Multi-Agent System Research → DevOps → Coding → Testing → Documentation
45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
"""
|
|
Pixel3DMM ComfyUI Custom Nodes
|
|
Generated by MCP Multi-Agent System - Coding Agent
|
|
|
|
This package provides ComfyUI nodes for 3D face reconstruction using the Pixel3DMM method.
|
|
"""
|
|
|
|
from .pixel3dmm_loader import Pixel3DMMLoader
|
|
from .face_reconstructor import FaceReconstructor3D
|
|
from .flame_optimizer import FLAMEOptimizer
|
|
from .mesh_exporter import MeshExporter
|
|
from .uv_predictor import UVPredictor
|
|
from .normal_predictor import NormalPredictor
|
|
|
|
# Node class mappings for ComfyUI
|
|
NODE_CLASS_MAPPINGS = {
|
|
"Pixel3DMMLoader": Pixel3DMMLoader,
|
|
"FaceReconstructor3D": FaceReconstructor3D,
|
|
"FLAMEOptimizer": FLAMEOptimizer,
|
|
"MeshExporter": MeshExporter,
|
|
"UVPredictor": UVPredictor,
|
|
"NormalPredictor": NormalPredictor,
|
|
}
|
|
|
|
# Display names for ComfyUI interface
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"Pixel3DMMLoader": "🔧 Pixel3DMM Model Loader",
|
|
"FaceReconstructor3D": "🎭 3D Face Reconstructor",
|
|
"FLAMEOptimizer": "🔥 FLAME Parameter Optimizer",
|
|
"MeshExporter": "📦 3D Mesh Exporter",
|
|
"UVPredictor": "🗺️ UV Coordinate Predictor",
|
|
"NormalPredictor": "📐 Surface Normal Predictor",
|
|
}
|
|
|
|
__all__ = [
|
|
"NODE_CLASS_MAPPINGS",
|
|
"NODE_DISPLAY_NAME_MAPPINGS",
|
|
"Pixel3DMMLoader",
|
|
"FaceReconstructor3D",
|
|
"FLAMEOptimizer",
|
|
"MeshExporter",
|
|
"UVPredictor",
|
|
"NormalPredictor",
|
|
]
|