WIP

src/yolo/annotation_generator.py

@@ -11,6 +11,7 @@ import csv
 
 
 # Field class mapping for YOLO
+# Note: supplier_accounts is not a separate class - its matches are mapped to Bankgiro/Plusgiro
 FIELD_CLASSES = {
     'InvoiceNumber': 0,
     'InvoiceDate': 1,
@@ -19,6 +20,16 @@ FIELD_CLASSES = {
     'Bankgiro': 4,
     'Plusgiro': 5,
     'Amount': 6,
+    'supplier_organisation_number': 7,
+}
+
+# Fields that need matching but map to other YOLO classes
+# supplier_accounts matches are classified as Bankgiro or Plusgiro based on account type
+ACCOUNT_FIELD_MAPPING = {
+    'supplier_accounts': {
+        'BG': 'Bankgiro',  # BG:xxx -> Bankgiro class
+        'PG': 'Plusgiro',  # PG:xxx -> Plusgiro class
+    }
 }
 
 CLASS_NAMES = [
@@ -29,6 +40,7 @@ CLASS_NAMES = [
     'bankgiro',
     'plusgiro',
     'amount',
+    'supplier_org_number',
 ]