41 lines
1.3 KiB
YAML
41 lines
1.3 KiB
YAML
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: ''
|