From 5b3cb257506f3fbe8c5ad3ea8aedb87d51e4c87c Mon Sep 17 00:00:00 2001
From: Christian C <cc@localhost>
Date: Wed, 5 Mar 2025 19:58:33 -0800
Subject: Filter small labels

---
 include/lib/seg/mask_data.h | 10 ++++++++++
 1 file changed, 10 insertions(+)

(limited to 'include/lib/seg')

diff --git a/include/lib/seg/mask_data.h b/include/lib/seg/mask_data.h
index 595ffd6..34fd6f1 100644
--- a/include/lib/seg/mask_data.h
+++ b/include/lib/seg/mask_data.h
@@ -43,4 +43,14 @@ struct AVLNode* increase_label_area_alloc(struct AVLNode* root, uint16_t label);
 //  Create an AVL node if it doesn't exist
 struct AVLNode* increase_label_perimeter_alloc(struct AVLNode* root, uint16_t label);
 
+// Comparison of uint16_ts
+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);
+
+// 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);
+
 #endif
-- 
cgit v1.2.1