14 lines
254 B
Python
14 lines
254 B
Python
import os
|
|
import sys
|
|
|
|
sys.path.append(os.path.dirname(__file__))
|
|
|
|
from coreml_suite import CoreMLLoaderUNet
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"CoreMLUNetLoader": CoreMLLoaderUNet,
|
|
}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"CoreMLUNetLoader": "Load Core ML UNet"
|
|
}
|