| 12345678910111213141516171819202122232425262728293031323334353637383940 |
- name: DBNet
- dataset:
- train:
- dataset:
- type: SynthTextDataset # 数据集类型
- args:
- data_path: ''# SynthTextDataset 根目录
- pre_processes: # 数据的预处理过程，包含augment和标签制作
- - type: IaaAugment # 使用imgaug进行变换
- args:
- - {'type':Fliplr, 'args':{'p':0.5}}
- - {'type': Affine, 'args':{'rotate':[-10,10]}}
- - {'type':Resize,'args':{'size':[0.5,3]}}
- - type: EastRandomCropData
- args:
- size: [640,640]
- max_tries: 50
- keep_ratio: true
- - type: MakeBorderMap
- args:
- shrink_ratio: 0.4
- - type: MakeShrinkMap
- args:
- shrink_ratio: 0.4
- min_text_size: 8
- transforms: # 对图片进行的变换方式
- - type: ToTensor
- args: {}
- - type: Normalize
- args:
- mean: [0.485, 0.456, 0.406]
- std: [0.229, 0.224, 0.225]
- img_mode: RGB
- filter_keys: ['img_path','img_name','text_polys','texts','ignore_tags','shape'] # 返回数据之前，从数据字典里删除的key
- ignore_tags: ['*', '###']
- loader:
- batch_size: 1
- shuffle: true
- num_workers: 0
- collate_fn: ''
|
