Skip to content
Snippets Groups Projects
Commit 7b770b3f authored by pswain's avatar pswain
Browse files

feat(babysitter): added pixel_size to BabyParameters

parent 942aa01e
No related branches found
No related tags found
No related merge requests found
...@@ -15,24 +15,30 @@ class BabyParameters(ParametersABC): ...@@ -15,24 +15,30 @@ class BabyParameters(ParametersABC):
def __init__( def __init__(
self, self,
modelset_name, modelset_name,
pixel_size,
clogging_thresh, clogging_thresh,
min_bud_tps, min_bud_tps,
isbud_thresh, isbud_thresh,
session,
): ):
"""Initialise parameters for BABY.""" """Initialise parameters for BABY."""
self.modelset_name = modelset_name self.modelset_name = modelset_name
self.pixel_size = pixel_size
self.clogging_thresh = clogging_thresh self.clogging_thresh = clogging_thresh
self.min_bud_tps = min_bud_tps self.min_bud_tps = min_bud_tps
self.isbud_thresh = isbud_thresh self.isbud_thresh = isbud_thresh
self.session = session
@classmethod @classmethod
def default(cls, **kwargs): def default(cls, **kwargs):
"""Define default parameters; kwargs choose BABY model set.""" """Define default parameters; kwargs choose BABY model set."""
return cls( return cls(
modelset_name=get_modelset_name_from_params(**kwargs), modelset_name=get_modelset_name_from_params(**kwargs),
pixel_size=0.182,
clogging_thresh=0.75, clogging_thresh=0.75,
min_bud_tps=3, min_bud_tps=3,
isbud_thresh=0.5, isbud_thresh=0.5,
session=None,
) )
def update_baby_modelset(self, path: t.Union[str, Path, t.Dict[str, str]]): def update_baby_modelset(self, path: t.Union[str, Path, t.Dict[str, str]]):
...@@ -78,9 +84,11 @@ class BabyRunner(StepABC): ...@@ -78,9 +84,11 @@ class BabyRunner(StepABC):
else: else:
brain = modelsets.get( brain = modelsets.get(
modelset_name, modelset_name,
pixel_size=parameters.pixel_size,
clogging_thresh=parameters.clogging_thresh, clogging_thresh=parameters.clogging_thresh,
min_bud_tps=parameters.min_bud_tps, min_bud_tps=parameters.min_bud_tps,
isbud_thresh=parameters.isbud_thresh, isbud_thresh=parameters.isbud_thresh,
session=parameters.session,
) )
self.crawler = BabyCrawler(brain) self.crawler = BabyCrawler(brain)
self.brightfield_channel = self.tiler.ref_channel_index self.brightfield_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