Skip to content
Snippets Groups Projects
Commit 52f24aea authored by Alán Muñoz's avatar Alán Muñoz
Browse files

feat(baby_client): assert model-tiler shape match

parent 563438cf
No related branches found
No related tags found
No related merge requests found
......@@ -139,6 +139,12 @@ class BabyRunner(StepABC):
if parameters is None
else parameters.model_config
)
tiler_z = self.tiler.shape[-3]
model_name = self.model_config["flattener_file"]
assert (
f"{tiler_z}z" in model_name
), f"Tiler z-stack ({tiler_z}) and Model shape ({model_name}) do not match "
self.brain = BabyBrain(**self.model_config)
self.crawler = BabyCrawler(self.brain)
self.bf_channel = self.tiler.ref_channel_index
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment