Update segment_key.py

segment_key.py

@@ -325,23 +325,25 @@ def rotate_image(image):
 
 
 def show_kps(contour):
-    features = []
+    black_image = np.zeros((300, 300), dtype=np.uint8)
     list1 = range(0, 300)
     list2 = list(zip(get_features_right(contour), list1))
-    features += list2
+    list2 = np.array(list2)
+    cv2.drawContours(black_image, [list2], -1, (255, 255, 255), 2)
 
     list2 = list(zip(get_features_left(contour), list1))
-    features += list2
+    list2 = np.array(list2)
+    cv2.drawContours(black_image, [list2], -1, (255, 255, 255), 2)
 
     list2 = list(zip(list1, get_features_up(contour)))
-    features += list2
+    list2 = np.array(list2)
+    cv2.drawContours(black_image, [list2], -1, (255, 255, 255), 2)
 
-    list2 = list(zip(list1, get_features_down(contour)))
-    features += list2
 
-    black_image = np.zeros((300, 300), dtype=np.uint8)
-    features = np.array(features)
-    cv2.drawContours(black_image, [features], -1, (255, 255, 255), 2)
+    list2 = list(zip(list1, get_features_down(contour)))
+    list2 = np.array(list2)
+    cv2.drawContours(black_image, [list2], -1, (255, 255, 255), 2)
+    
     return black_image