Spaces:
Runtime error
Runtime error
File size: 1,738 Bytes
fc8c192 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 |
from __future__ import absolute_import, division, print_function, unicode_literals
from .ColorJitter import ColorJitter
from .copy_paste import CopyPaste
from .east_process import *
from .fce_aug import *
from .fce_targets import FCENetTargets
from .gen_table_mask import *
from .iaa_augment import IaaAugment
from .label_ops import *
from .make_border_map import MakeBorderMap
from .make_pse_gt import MakePseGt
from .make_shrink_map import MakeShrinkMap
from .operators import *
from .pg_process import *
from .randaugment import RandAugment
from .random_crop_data import EastRandomCropData, RandomCropImgMask
from .rec_img_aug import (
ClsResizeImg,
NRTRRecResizeImg,
PRENResizeImg,
RecAug,
RecConAug,
RecResizeImg,
SARRecResizeImg,
SRNRecResizeImg,
)
from .sast_process import *
from .ssl_img_aug import SSLRotateResize
from .vqa import *
def transform(data, ops=None):
"""transform"""
if ops is None:
ops = []
for op in ops:
data = op(data)
if data is None:
return None
return data
def create_operators(op_param_list, global_config=None):
"""
create operators based on the config
Args:
params(list): a dict list, used to create some operators
"""
assert isinstance(op_param_list, list), "operator config should be a list"
ops = []
for operator in op_param_list:
assert isinstance(operator, dict) and len(operator) == 1, "yaml format error"
op_name = list(operator)[0]
param = {} if operator[op_name] is None else operator[op_name]
if global_config is not None:
param.update(global_config)
op = eval(op_name)(**param)
ops.append(op)
return ops
|