提交 6e6d7743 编写于 作者: T Topdu

fix ci export bug

上级 1c5e7f25
......@@ -72,7 +72,6 @@ Train:
ignore_index: *ignore_index
- ABINetRecResizeImg:
image_shape: [3, 32, 128]
padding: False
- KeepKeys:
keep_keys: ['image', 'label', 'length'] # dataloader will return list in this order
loader:
......@@ -93,7 +92,6 @@ Eval:
ignore_index: *ignore_index
- ABINetRecResizeImg:
image_shape: [3, 32, 128]
padding: False
- KeepKeys:
keep_keys: ['image', 'label', 'length'] # dataloader will return list in this order
loader:
......
......@@ -84,7 +84,6 @@ Train:
channel_first: False
- CTCLabelEncode: # Class handling label
- SVTRRecResizeImg:
character_dict_path:
image_shape: [3, 64, 256]
padding: False
- KeepKeys:
......@@ -105,7 +104,6 @@ Eval:
channel_first: False
- CTCLabelEncode: # Class handling label
- SVTRRecResizeImg:
character_dict_path:
image_shape: [3, 64, 256]
padding: False
- KeepKeys:
......
......@@ -110,8 +110,6 @@ Train:
- RecAug: null
- CTCLabelEncode: null
- SVTRRecResizeImg:
character_dict_path: null
infer_mode: False
image_shape:
- 3
- 32
......@@ -139,8 +137,6 @@ Eval:
channel_first: false
- CTCLabelEncode: null
- SVTRRecResizeImg:
character_dict_path: null
infer_mode: False
image_shape:
- 3
- 32
......
......@@ -312,14 +312,8 @@ class GrayRecResizeImg(object):
class ABINetRecResizeImg(object):
def __init__(self,
image_shape,
infer_mode=False,
character_dict_path=None,
**kwargs):
def __init__(self, image_shape, **kwargs):
self.image_shape = image_shape
self.infer_mode = infer_mode
self.character_dict_path = character_dict_path
def __call__(self, data):
img = data['image']
......@@ -330,15 +324,8 @@ class ABINetRecResizeImg(object):
class SVTRRecResizeImg(object):
def __init__(self,
image_shape,
infer_mode=False,
character_dict_path='./ppocr/utils/ppocr_keys_v1.txt',
padding=True,
**kwargs):
def __init__(self, image_shape, padding=True, **kwargs):
self.image_shape = image_shape
self.infer_mode = infer_mode
self.character_dict_path = character_dict_path
self.padding = padding
def __call__(self, data):
......
......@@ -84,8 +84,7 @@ Train:
img_mode: BGR
channel_first: False
- CTCLabelEncode: # Class handling label
- RecResizeImg:
character_dict_path:
- SVTRRecResizeImg:
image_shape: [3, 64, 256]
padding: False
- KeepKeys:
......@@ -106,8 +105,7 @@ Eval:
img_mode: BGR
channel_first: False
- CTCLabelEncode: # Class handling label
- RecResizeImg:
character_dict_path:
- SVTRRecResizeImg:
image_shape: [3, 64, 256]
padding: False
- KeepKeys:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册