20 lines
501 B
Python
20 lines
501 B
Python
"""
|
|
ComfyUI-D-DFFNet: Defocus Blur Detection Custom Node for ComfyUI.
|
|
|
|
This node wraps the D-DFFNet (Depth-aware Defocus Blur Detection Network)
|
|
model from the paper "Depth and DOF Cues Make A Better Defocus Blur Detector"
|
|
(ICME 2023).
|
|
"""
|
|
|
|
from .dffnet_node import DDFFNetNode
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"DDFFNet": DDFFNetNode,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"DDFFNet": "D-DFFNet Defocus Blur Detection",
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|