Spaces:
Running
Running
Update segment_key.py
Browse files- segment_key.py +1 -0
segment_key.py
CHANGED
@@ -340,6 +340,7 @@ def show_kps(contour):
|
|
340 |
features += list2
|
341 |
|
342 |
black_image = np.zeros((300, 300), dtype=np.uint8)
|
|
|
343 |
cv2.drawContours(black_image, [features], -1, (255, 255, 255), 2)
|
344 |
return black_image
|
345 |
|
|
|
340 |
features += list2
|
341 |
|
342 |
black_image = np.zeros((300, 300), dtype=np.uint8)
|
343 |
+
features = np.array(features)
|
344 |
cv2.drawContours(black_image, [features], -1, (255, 255, 255), 2)
|
345 |
return black_image
|
346 |
|