Better implementation for label selection

UMEssen · Aug 4, 2023 · 1f5ae00 · 1f5ae00
1 parent 64ab05d
commit 1f5ae00
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/boa_contrast/features/feature_builder.py b/boa_contrast/features/feature_builder.py
@@ -81,9 +81,7 @@ def compute_features(
                 if isinstance(self.label_map[region], int):
                     region_mask = body_region_all == self.label_map[region]
                 else:
-                    region_mask = body_region_all == self.label_map[region][0]
-                    for num in self.label_map[region][1:]:
-                        region_mask += body_region_all == num
+                    region_mask = np.isin(body_region_all, self.label_map[region])
             if np.sum(region_mask) == 0:
                 continue