You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
|
|
|
|
|
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: ''
|
