Filter small labels

1 files changed, 3 insertions, 0 deletions

diff --git a/include/lib/seg/mask_data.h b/include/lib/seg/mask_data.h
index 34fd6f1..f2803e6 100644
--- a/include/lib/seg/mask_data.h
+++ b/include/lib/seg/mask_data.h
@@ -49,6 +49,9 @@ bool_t compare_uint16_t(uint16_t* s1, uint16_t* s2);
 // In-order traversal print pointer
 void print_in_order_uint16_t(struct AVLNode* root);
 
+// Check if uint16_t in AVLTree with uint16_t* data
+bool_t in_uint16_t_tree(struct AVLNode* root, uint16_t value);
+
 // Filter out small masks
 //  Assumption: Contiguous labeling
 struct AVLNode* get_small_labels(struct AVLNode* removal_tree, struct AVLNode* label_tree, size_t min_area, size_t min_perimeter);