correct dataAssociation

eval.py

@@ -274,6 +274,20 @@ def correction_labels(boxes, labels, class_dict, pool_dict, flow_links):
                         print('change the link from sequenceFlow to messageFlow')
                         labels[i]=list(class_dict.values()).index('messageFlow')
 
+    for i in range(len(labels)):
+        #check if dataAssociation is connected to a dataObject
+        if labels[i] == list(class_dict.values()).index('dataAssociation'):
+            id1, id2 = flow_links[i]
+            if (id1 and id2) is not None:
+                label1 = labels[id1]
+                label2 = labels[id2]
+                if label1 == list(class_dict.values()).index('dataObject') or label2 == list(class_dict.values()).index('dataObject'):
+                    continue
+                else:
+                    print('change the link from dataAssociation to messageFlow')
+                    labels[i]=list(class_dict.values()).index('messageFlow')
+                
+
     return labels, flow_links