Spaces:
Build error
Build error
First model version
Browse files- tools/demo.py +1 -1
tools/demo.py
CHANGED
@@ -367,7 +367,7 @@ class TextDemo(object):
|
|
367 |
def mask_to_contours(self, mask, img_shape):
|
368 |
e = mask[0, :, :]
|
369 |
|
370 |
-
|
371 |
|
372 |
if len(countours) == 0:
|
373 |
return np.zeros((1, 8))
|
|
|
367 |
def mask_to_contours(self, mask, img_shape):
|
368 |
e = mask[0, :, :]
|
369 |
|
370 |
+
countours, hier = cv2.findContours(e.clone().numpy(), cv2.RETR_CCOMP, cv2.CHAIN_APPROX_NONE) # Aarlog
|
371 |
|
372 |
if len(countours) == 0:
|
373 |
return np.zeros((1, 8))
|