Spaces:
Build error
Build error
First model version
Browse files- det_demo.py +1 -1
det_demo.py
CHANGED
@@ -499,7 +499,7 @@ class DetDemo(object):
|
|
499 |
cv2.imwrite('./demo/example_results.jpg', im_write[:, :, ::-1])
|
500 |
'''
|
501 |
|
502 |
-
return contours, np.array(masks.repeat(1,3,1,1)).astype(np.bool_).transpose(0,2,3,1), boxes
|
503 |
|
504 |
def process_char_mask(self, char_masks, boxes, threshold=192):
|
505 |
texts, rec_scores = [], []
|
|
|
499 |
cv2.imwrite('./demo/example_results.jpg', im_write[:, :, ::-1])
|
500 |
'''
|
501 |
|
502 |
+
return contours, np.array(masks.repeat(1,3,1,1)).astype(np.bool_).transpose(0,2,3,1), np.array(boxes).astype(int)
|
503 |
|
504 |
def process_char_mask(self, char_masks, boxes, threshold=192):
|
505 |
texts, rec_scores = [], []
|