BenjiELCA commited on
Commit
144b4fe
·
1 Parent(s): 2b81555
Files changed (1) hide show
  1. 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)):