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: ''