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

fix(extractor): enforce ndarray for bg masks

parent 060bca1d
No related branches found
No related tags found
No related merge requests found
...@@ -482,7 +482,7 @@ class Extractor(StepABC): ...@@ -482,7 +482,7 @@ class Extractor(StepABC):
# stored as an array arranged as (traps, channels, time points, X, Y, Z) # stored as an array arranged as (traps, channels, time points, X, Y, Z)
tiles = self.get_tiles(tp, tile_shape=tile_size, channels=tree_chs) tiles = self.get_tiles(tp, tile_shape=tile_size, channels=tree_chs)
# generate boolean masks for background as a list with one mask per trap # generate boolean masks for background as a list with one mask per trap
bgs = [] bgs = np.array([])
if self.params.sub_bg: if self.params.sub_bg:
# bgs = [ # bgs = [
# ~np.sum(m, axis=0).astype(bool) # ~np.sum(m, axis=0).astype(bool)
......
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