31 lines
822 B
Python
31 lines
822 B
Python
# __init__.py
|
|
import os
|
|
import sys
|
|
|
|
from .nodes.image_feeder import ImageFeeder
|
|
from .nodes.image_filters import HighPassFilter
|
|
from .nodes.face_wrapper import FaceWrapper
|
|
from .nodes.face_fit_and_restore import FaceFitAndRestore
|
|
|
|
# Add current directory to system path
|
|
sys.path.append(os.path.dirname(os.path.realpath(__file__)))
|
|
|
|
# Get the path to the current directory
|
|
NODE_PATH = os.path.dirname(os.path.realpath(__file__))
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"FaceFitAndRestore": FaceFitAndRestore,
|
|
"FaceWrapper": FaceWrapper,
|
|
"HighPassFilter": HighPassFilter,
|
|
"ImageFeeder": ImageFeeder
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"FaceFitAndRestore": "Face Fit or Restore",
|
|
"FaceWrapper": "Face Wrapper",
|
|
"HighPassFilter": "High Pass Filter (HPF)",
|
|
"ImageFeeder": "Image Feeder"
|
|
}
|
|
|
|
__version__ = "1.1.0"
|