aboutsummaryrefslogtreecommitdiff
path: root/include/lib/seg/mask_data.h
diff options
context:
space:
mode:
Diffstat (limited to 'include/lib/seg/mask_data.h')
-rw-r--r--include/lib/seg/mask_data.h6
1 files changed, 6 insertions, 0 deletions
diff --git a/include/lib/seg/mask_data.h b/include/lib/seg/mask_data.h
index f2803e6..8e2ae3a 100644
--- a/include/lib/seg/mask_data.h
+++ b/include/lib/seg/mask_data.h
@@ -56,4 +56,10 @@ bool_t in_uint16_t_tree(struct AVLNode* root, uint16_t value);
// Assumption: Contiguous labeling
struct AVLNode* get_small_labels(struct AVLNode* removal_tree, struct AVLNode* label_tree, size_t min_area, size_t min_perimeter);
+// Get mask label data
+struct AVLNode* get_mask_data(uint16_t* masks, uint32_t width, uint32_t height);
+
+// Filter out small masks in mask
+void filter_small_masks(uint16_t* masks, uint32_t width, uint32_t height, size_t min_area, size_t min_perimeter);
+
#endif