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

refractor(baby_client): remove hard-coded bf

parent d4d9dcf3
No related branches found
No related tags found
No related merge requests found
......@@ -143,7 +143,7 @@ class BabyRunner:
)
self.brain = BabyBrain(**self.model_config)
self.crawler = BabyCrawler(self.brain)
self.bf_channel = self.tiler.get_channel_index("Brightfield")
self.bf_channel = self.tiler.ref_channel_index
@classmethod
def from_tiler(cls, parameters: BabyParameters, tiler):
......
......@@ -548,6 +548,10 @@ class Tiler(ProcessABC):
)
return np.stack(res, axis=1)
@property
def ref_channel_index(self):
return self.get_channel_index(self.parameters.ref_channel)
def get_channel_index(self, item):
"""
Find index for channel using regex. Returns the first matched string.
......
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