Spaces:
Sleeping
Sleeping
add container in webpage
Browse files- 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 |
-
|
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,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 |
|