Skip to content
Snippets Groups Projects
Unverified Commit 19d9fbfa authored by i-robot's avatar i-robot Committed by Gitee
Browse files

!2501 [crnn_seq2seq_ocr]set dataset prefetch size

Merge pull request !2501 from zhouneng/fix_issue_crnn_seq
parents 156279d8 c00e2252
No related branches found
No related tags found
No related merge requests found
......@@ -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)):
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment