soorena62
20 lines
576 B
Python
20 lines
576 B
Python
from core.workflow import Graph
|
|
from nodes.image_reader import ImageReader
|
|
from nodes.feature_extractor import FeatureExtractor
|
|
from nodes.feature_writer import FeatureWriter
|
|
|
|
|
|
graph = Graph()
|
|
|
|
reader = ImageReader(modality="CT")
|
|
extractor = FeatureExtractor()
|
|
writer = FeatureWriter()
|
|
|
|
# ImageReader → FeatureExtractor
|
|
ok, msg = graph.can_connect(reader, "image", extractor, "image")
|
|
print("Reader → Extractor:", ok, msg)
|
|
|
|
# FeatureExtractor → FeatureWriter
|
|
ok, msg = graph.can_connect(extractor, "features", writer, "features")
|
|
print("Extractor → Writer:", ok, msg)
|