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

fix ci export bug

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