This commit is contained in:
69
benchmark/PaddleOCR_DBNet/config/icdar2015.yaml
Normal file
69
benchmark/PaddleOCR_DBNet/config/icdar2015.yaml
Normal file
@@ -0,0 +1,69 @@
|
||||
name: DBNet
|
||||
dataset:
|
||||
train:
|
||||
dataset:
|
||||
type: ICDAR2015Dataset # 数据集类型
|
||||
args:
|
||||
data_path: # 一个存放 img_path \t gt_path的文件
|
||||
- ''
|
||||
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
|
||||
thresh_min: 0.3
|
||||
thresh_max: 0.7
|
||||
- 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: ''
|
||||
validate:
|
||||
dataset:
|
||||
type: ICDAR2015Dataset
|
||||
args:
|
||||
data_path:
|
||||
- ''
|
||||
pre_processes:
|
||||
- type: ResizeShortSize
|
||||
args:
|
||||
short_size: 736
|
||||
resize_text_polys: false
|
||||
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: []
|
||||
ignore_tags: ['*', '###']
|
||||
loader:
|
||||
batch_size: 1
|
||||
shuffle: true
|
||||
num_workers: 0
|
||||
collate_fn: ICDARCollectFN
|
||||
Reference in New Issue
Block a user