diff --git a/paddlex/repo_apis/PaddleOCR_api/configs/SLANet.yaml b/paddlex/repo_apis/PaddleOCR_api/configs/SLANet.yaml index a33d18ea84..3e2c300fde 100644 --- a/paddlex/repo_apis/PaddleOCR_api/configs/SLANet.yaml +++ b/paddlex/repo_apis/PaddleOCR_api/configs/SLANet.yaml @@ -97,7 +97,7 @@ Train: size: [488, 488] - ToCHWImage: - KeepKeys: - keep_keys: [ 'image', 'structure', 'bboxes', 'bbox_masks', 'shape' ] + keep_keys: [ 'image', 'structure', 'bboxes', 'bbox_masks', 'length', 'shape' ] loader: shuffle: True batch_size_per_card: 48 @@ -133,7 +133,7 @@ Eval: size: [488, 488] - ToCHWImage: - KeepKeys: - keep_keys: [ 'image', 'structure', 'bboxes', 'bbox_masks', 'shape' ] + keep_keys: [ 'image', 'structure', 'bboxes', 'bbox_masks', 'length', 'shape' ] loader: shuffle: False drop_last: False diff --git a/paddlex/repo_apis/PaddleOCR_api/configs/SLANet_plus.yaml b/paddlex/repo_apis/PaddleOCR_api/configs/SLANet_plus.yaml index a33d18ea84..3e2c300fde 100644 --- a/paddlex/repo_apis/PaddleOCR_api/configs/SLANet_plus.yaml +++ b/paddlex/repo_apis/PaddleOCR_api/configs/SLANet_plus.yaml @@ -97,7 +97,7 @@ Train: size: [488, 488] - ToCHWImage: - KeepKeys: - keep_keys: [ 'image', 'structure', 'bboxes', 'bbox_masks', 'shape' ] + keep_keys: [ 'image', 'structure', 'bboxes', 'bbox_masks', 'length', 'shape' ] loader: shuffle: True batch_size_per_card: 48 @@ -133,7 +133,7 @@ Eval: size: [488, 488] - ToCHWImage: - KeepKeys: - keep_keys: [ 'image', 'structure', 'bboxes', 'bbox_masks', 'shape' ] + keep_keys: [ 'image', 'structure', 'bboxes', 'bbox_masks', 'length', 'shape' ] loader: shuffle: False drop_last: False