diff --git a/official/cv/crnn_seq2seq_ocr/src/dataset.py b/official/cv/crnn_seq2seq_ocr/src/dataset.py index f75db8f1f5d9db31f964ab0620a4bae3d7672122..40abc60fd46213668daf82805d98137dd89b3526 100644 --- a/official/cv/crnn_seq2seq_ocr/src/dataset.py +++ b/official/cv/crnn_seq2seq_ocr/src/dataset.py @@ -27,6 +27,9 @@ import mindspore.common.dtype as mstype from src.model_utils.config import config +de.config.set_prefetch_size(512) + + class AugmentationOps(): def __init__(self, min_area_ratio=0.8, aspect_ratio_range=(0.8, 1.2), brightness=32./255., contrast=0.5, saturation=0.5, hue=0.2, img_tile_shape=(150, 150)):