From f9dd8ddca8374e72508b1907f12e23bb2f3b412f Mon Sep 17 00:00:00 2001 From: anna-grim Date: Sun, 8 Feb 2026 23:00:51 +0000 Subject: [PATCH] refactor: input model to hgat --- src/neuron_proofreader/merge_proofreading/merge_dataloading.py | 2 +- src/neuron_proofreader/split_proofreading/split_inference.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/neuron_proofreader/merge_proofreading/merge_dataloading.py b/src/neuron_proofreader/merge_proofreading/merge_dataloading.py index eb54c20..0bf8656 100644 --- a/src/neuron_proofreader/merge_proofreading/merge_dataloading.py +++ b/src/neuron_proofreader/merge_proofreading/merge_dataloading.py @@ -15,7 +15,7 @@ from neuron_proofreader.utils import util -TEST_BRAIN = "653159" +TEST_BRAIN = "747807" # --- Load Skeletons --- diff --git a/src/neuron_proofreader/split_proofreading/split_inference.py b/src/neuron_proofreader/split_proofreading/split_inference.py index 9bcdf17..ec205fd 100644 --- a/src/neuron_proofreader/split_proofreading/split_inference.py +++ b/src/neuron_proofreader/split_proofreading/split_inference.py @@ -57,6 +57,7 @@ def __init__( img_path, model_path, output_dir, + model, config, log_preamble="", segmentation_path=None, @@ -92,7 +93,7 @@ def __init__( self.accepted_proposals = list() self.config = config self.img_path = img_path - self.model = VisionHGAT(config.ml.patch_shape) + self.model = model self.output_dir = output_dir self.soma_centroids = soma_centroids