Cyril666 commited on
Commit
682a278
·
1 Parent(s): d2fc4b7

First model version

Browse files
Files changed (1) hide show
  1. 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 = [], []