subraoulandClaude
c04e18c99f
- 4 nodes: Classical enhancement, Save 16-bit TIFF, ABCD, deepDeband - Classical methods: Bilateral+Dither, Gradient Domain, Multi-scale Fusion, Fast Edge-Aware - Deep learning: ABCD (3 models), deepDeband (2 variants) - Batch processing support - Comprehensive documentation Based on: - ABCD: https://github.com/WooKyoungHan/ABCD (CVPR 2023) - deepDeband: https://github.com/RaymondLZhou/deepDeband (ICIP 2022) 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
27 lines
975 B
Python
27 lines
975 B
Python
"""
|
|
ComfyUI Bit Depth Enhancement
|
|
Classical and deep learning-based bit-depth enhancement for professional cinematography workflows
|
|
|
|
This is the ComfyUI entry point that loads the custom nodes.
|
|
"""
|
|
|
|
from .bit_depth_enhancement.nodes.classical import BitDepthEnhancementClassical, Save16BitTIFF
|
|
from .bit_depth_enhancement.nodes.abcd import ABCD_BitDepthEnhancement
|
|
from .bit_depth_enhancement.nodes.deband import DeepDeband
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"BitDepthEnhancementClassical": BitDepthEnhancementClassical,
|
|
"Save16BitTIFF": Save16BitTIFF,
|
|
"ABCD_BitDepthEnhancement": ABCD_BitDepthEnhancement,
|
|
"DeepDeband": DeepDeband,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"BitDepthEnhancementClassical": "Bit Depth Enhancement (Classical)",
|
|
"Save16BitTIFF": "Save 16-bit TIFF",
|
|
"ABCD_BitDepthEnhancement": "ABCD Bit-Depth Enhancement (8→16)",
|
|
"DeepDeband": "deepDeband (Banding Removal)",
|
|
}
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
|