BenjiELCA commited on
Commit
fc2d014
·
1 Parent(s): cc79c19

add container in webpage

Browse files
Files changed (1) hide show
  1. modules/eval.py +2 -3
modules/eval.py CHANGED
@@ -435,9 +435,9 @@ def generate_data(image, boxes, labels, scores, keypoints, flow_links, best_poin
435
  return data
436
 
437
  def develop_prediction(boxes, labels, scores, keypoints, class_dict, correction=True):
438
- print("Lenghts 1", len(boxes), len(labels), len(scores), len(keypoints))
439
  pool_dict, boxes, labels, scores, keypoints = regroup_elements_by_pool(boxes, labels, scores, keypoints, class_dict)
440
- print("Lenghts 2", len(boxes), len(labels), len(scores), len(keypoints))
441
  # Create links between elements
442
  flow_links, best_points = create_links(keypoints, boxes, labels, class_dict)
443
 
@@ -454,7 +454,6 @@ def develop_prediction(boxes, labels, scores, keypoints, class_dict, correction=
454
  labels[i] = list(class_dict.values()).index('dataObject')
455
 
456
  boxes,labels,scores,keypoints,flow_links,best_points,pool_dict = last_correction(boxes,labels,scores,keypoints,flow_links,best_points, pool_dict)
457
- print("Lenghts 3", len(boxes), len(labels), len(scores), len(keypoints))
458
 
459
  return boxes, labels, scores, keypoints, flow_links, best_points, pool_dict
460
 
 
435
  return data
436
 
437
  def develop_prediction(boxes, labels, scores, keypoints, class_dict, correction=True):
438
+
439
  pool_dict, boxes, labels, scores, keypoints = regroup_elements_by_pool(boxes, labels, scores, keypoints, class_dict)
440
+
441
  # Create links between elements
442
  flow_links, best_points = create_links(keypoints, boxes, labels, class_dict)
443
 
 
454
  labels[i] = list(class_dict.values()).index('dataObject')
455
 
456
  boxes,labels,scores,keypoints,flow_links,best_points,pool_dict = last_correction(boxes,labels,scores,keypoints,flow_links,best_points, pool_dict)
 
457
 
458
  return boxes, labels, scores, keypoints, flow_links, best_points, pool_dict
459