diff --git a/src/aliby/baby_client.py b/src/aliby/baby_client.py index 5731967b3a5dd34da2457818de033ce6ceb6d68e..f13c5b027924c0f05a733edbc58a7d2f7ecb7280 100644 --- a/src/aliby/baby_client.py +++ b/src/aliby/baby_client.py @@ -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): diff --git a/src/aliby/tile/tiler.py b/src/aliby/tile/tiler.py index 91add3dafa0d6ed069f046e74ca3973c34540e04..db7b7a97ff2a86654a605cbfcd57991b89555beb 100644 --- a/src/aliby/tile/tiler.py +++ b/src/aliby/tile/tiler.py @@ -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.