Spaces:
Running
Running
- modules/eval.py +0 -6
modules/eval.py
CHANGED
@@ -424,20 +424,14 @@ def generate_data(image, boxes, labels, scores, keypoints, flow_links, best_poin
|
|
424 |
return data
|
425 |
|
426 |
def develop_prediction(boxes, labels, scores, keypoints, class_dict, correction=True):
|
427 |
-
# Regroup elements by pool
|
428 |
-
print('labels:', labels)
|
429 |
pool_dict, boxes, labels, scores, keypoints = regroup_elements_by_pool(boxes, labels, scores, keypoints, class_dict)
|
430 |
-
print('labels:', labels)
|
431 |
# Create links between elements
|
432 |
flow_links, best_points = create_links(keypoints, boxes, labels, class_dict)
|
433 |
-
print('labels:', labels)
|
434 |
#Correct the labels of some sequenceflow that cross multiple pool
|
435 |
if correction:
|
436 |
labels, flow_links = correction_labels(boxes, labels, class_dict, pool_dict, flow_links)
|
437 |
-
print('labels:', labels)
|
438 |
#give a link to event to allow the creation of the BPMN id with start, indermediate and end event
|
439 |
flow_links = give_link_to_element(flow_links, labels)
|
440 |
-
print('labels:', labels)
|
441 |
|
442 |
#change every datastore to dataobject [TO DO: change it to make the dataStore work]
|
443 |
for i in range(len(labels)):
|
|
|
424 |
return data
|
425 |
|
426 |
def develop_prediction(boxes, labels, scores, keypoints, class_dict, correction=True):
|
|
|
|
|
427 |
pool_dict, boxes, labels, scores, keypoints = regroup_elements_by_pool(boxes, labels, scores, keypoints, class_dict)
|
|
|
428 |
# Create links between elements
|
429 |
flow_links, best_points = create_links(keypoints, boxes, labels, class_dict)
|
|
|
430 |
#Correct the labels of some sequenceflow that cross multiple pool
|
431 |
if correction:
|
432 |
labels, flow_links = correction_labels(boxes, labels, class_dict, pool_dict, flow_links)
|
|
|
433 |
#give a link to event to allow the creation of the BPMN id with start, indermediate and end event
|
434 |
flow_links = give_link_to_element(flow_links, labels)
|
|
|
435 |
|
436 |
#change every datastore to dataobject [TO DO: change it to make the dataStore work]
|
437 |
for i in range(len(labels)):
|