Add All Folders
This commit is contained in:
@@ -0,0 +1,19 @@
|
||||
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)
|
||||
Reference in New Issue
Block a user