Upload 2 files

datatypes/config.py

@@ -0,0 +1,13 @@
+import os
+
+# config values 
+
+Config = {"table_detection_padding_pixel":10, 'table_recognition_padding_pixel':5,
+        'table_detection_threshold':0.7, 'table_recognition_threshold':0.8,
+        'table_padd': 20, 'row_padd':6, 'cell_padd':3,
+        }
+        
+tesseract_config = {'tesseractpath':'C://Program Files//Tesseract-OCR//tesseract.exe'}
+
+model_config = {'detection_model_path':'D:/Table-detection/models/detection-model',
+                'recognition_model_path':'D:/Table-detection/models/recognition-model'}

datatypes/datatypes.py

@@ -0,0 +1,53 @@
+import os
+from dataclasses import dataclass, field
+from enum import Enum
+from typing import List, Dict
+
+class DetectionLabels(Enum):
+    table = 0
+    table_column = 1
+    table_row = 2
+    table_column_header = 3
+    table_projected_row_header = 4
+    table_spanning_cell = 5
+
+class ExtractionContext(Enum):
+    document = 1
+    table = 2
+    row = 3
+
+@dataclass
+class Cell:
+    cellindex : int = 0
+    value : str = ''
+    prob : float = 0.5
+
+@dataclass
+class Row:
+    rowindex : int = 0
+    extracted_cells : List[Cell]= field(default_factory=lambda: [])
+
+@dataclass
+class TableRecognitionData:
+    scores : List = field(default_factory=lambda: [])
+    labels : List = field(default_factory=lambda: [])
+    boxes : List = field(default_factory=lambda: [])
+
+@dataclass
+class TableRecognitionOrdered:
+    recognized_row : List = field(default_factory=lambda: [])
+    recognized_column : List = field(default_factory=lambda: [])
+
+@dataclass
+class TableDetectionData:
+    detection_score : float = 0.0
+    detection_label : int = 0
+    detection_box : List = field(default_factory=lambda: [])
+    recognitiondata : TableRecognitionData = field(default_factory=lambda: [])
+    ordered_recognitiondata : List[TableRecognitionOrdered] = field(default_factory=lambda: [])
+    extracted_rows : List[Row] = field(default_factory=lambda: [])
+
+
+@dataclass
+class ImageData:
+    tables: List[TableDetectionData]