Spaces:
Runtime error
Runtime error
Update ctrnet_infer.py
Browse files- ctrnet_infer.py +2 -4
ctrnet_infer.py
CHANGED
|
@@ -148,14 +148,12 @@ class CTRNetInfer():
|
|
| 148 |
|
| 149 |
def __call__(self, ori_img):
|
| 150 |
ori_img_shape = ori_img.shape[:2]
|
| 151 |
-
bboxes = self.text_det(ori_img)[0].astype(np.int64)
|
| 152 |
|
| 153 |
# resize img 到512x512
|
| 154 |
resize_img = cv2.resize(ori_img, self.input_shape,
|
| 155 |
interpolation=cv2.INTER_LINEAR)
|
| 156 |
-
resize_bboxes = self.
|
| 157 |
-
ori_img_shape,
|
| 158 |
-
self.input_shape)
|
| 159 |
|
| 160 |
img, structure_im, gt_text, soft_mask = self.dataset(
|
| 161 |
resize_img, resize_bboxes)
|
|
|
|
| 148 |
|
| 149 |
def __call__(self, ori_img):
|
| 150 |
ori_img_shape = ori_img.shape[:2]
|
| 151 |
+
# bboxes = self.text_det(ori_img)[0].astype(np.int64)
|
| 152 |
|
| 153 |
# resize img 到512x512
|
| 154 |
resize_img = cv2.resize(ori_img, self.input_shape,
|
| 155 |
interpolation=cv2.INTER_LINEAR)
|
| 156 |
+
resize_bboxes = self.text_det(resize_img)[0].astype(np.int64)
|
|
|
|
|
|
|
| 157 |
|
| 158 |
img, structure_im, gt_text, soft_mask = self.dataset(
|
| 159 |
resize_img, resize_bboxes)
|