Spaces:

dibahadie
/

KeychainSegmentation

Sleeping

dibahadie commited on Sep 15

Commit

8f38140

•

1 Parent(s): 4d0a681

Update segment_key.py

Files changed (1) hide show

segment_key.py CHANGED Viewed

@@ -291,7 +291,8 @@ def final_features(image_path):
     image = rotate_image(image)
     pil_image = pred_segmentation(image, pred_bbox(image_path))
     contour = get_kps_from_pil(pil_image)
-    return extract_features(contour)
 def predict_kps(image):

     image = rotate_image(image)
     pil_image = pred_segmentation(image, pred_bbox(image_path))
     contour = get_kps_from_pil(pil_image)
+    kp_image = show_kps(contour)
+    return kp_image, extract_features(contour)
 def predict_kps(image):