Skip to content

Commit

Permalink
[Dataset] Add config generators to all textdet and textrecog configs (#…
Browse files Browse the repository at this point in the history
  • Loading branch information
gaotongxiao committed Nov 23, 2022
1 parent 24aaec2 commit f6472ea
Show file tree
Hide file tree
Showing 10 changed files with 20 additions and 0 deletions.
2 changes: 2 additions & 0 deletions dataset_zoo/icdar2013/textdet.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,3 +55,5 @@
separator=' ',
mode='xyxy'),
dumper=dict(type='JsonDumper'))

config_generator = dict(type='TextDetConfigGenerator', data_root=data_root)
2 changes: 2 additions & 0 deletions dataset_zoo/icdar2015/textdet.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,3 +49,5 @@
parser=dict(type='ICDARTxtTextDetAnnParser', encoding='utf-8-sig'),
dumper=dict(type='JsonDumper'),
delete=['annotations', 'ic15_textdet_test_img', 'ic15_textdet_train_img'])

config_generator = dict(type='TextDetConfigGenerator', data_root=data_root)
2 changes: 2 additions & 0 deletions dataset_zoo/iiit5k/textrecog.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,3 +45,5 @@
format='img text'),
dumper=dict(type='JsonDumper'),
delete=['annotations', 'IIIT5K'])

config_generator = dict(type='TextRecogConfigGenerator', data_root=data_root)
2 changes: 2 additions & 0 deletions dataset_zoo/svt/textdet.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,3 +25,5 @@
parser=dict(type='SVTTextDetAnnParser', data_root=data_root),
dumper=dict(type='JsonDumper'),
delete=['annotations', 'svt'])

config_generator = dict(type='TextDetConfigGenerator', data_root=data_root)
2 changes: 2 additions & 0 deletions dataset_zoo/svt/textrecog.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
_base_ = ['textdet.py']

data_converter = dict(type='TextRecogCropConverter')

config_generator = dict(type='TextRecogConfigGenerator')
2 changes: 2 additions & 0 deletions dataset_zoo/svtp/textrecog.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,3 +25,5 @@
type='ICDARTxtTextRecogAnnParser', separator=' ', format='img text'),
dumper=dict(type='JsonDumper'),
delete=['svtp', 'annotations'])

config_generator = dict(type='TextRecogConfigGenerator', data_root=data_root)
2 changes: 2 additions & 0 deletions dataset_zoo/totaltext/textdet.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,3 +37,5 @@
parser=dict(type='TotaltextTextDetAnnParser', data_root=data_root),
dumper=dict(type='JsonDumper'),
delete=['totaltext', 'txt_format', 'annotations'])

config_generator = dict(type='TextDetConfigGenerator', data_root=data_root)
2 changes: 2 additions & 0 deletions dataset_zoo/totaltext/textrecog.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
_base_ = ['textdet.py']

data_converter = dict(type='TextRecogCropConverter')

config_generator = dict(type='TextRecogConfigGenerator')
3 changes: 3 additions & 0 deletions dataset_zoo/wildreceipt/textdet.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,3 +4,6 @@
type='TextDetDataConverter',
parser=dict(type='WildreceiptTextDetAnnParser'),
dumper=dict(type='JsonDumper'))

config_generator = dict(
type='TextRecogConfigGenerator', data_root=_base_.data_root)
1 change: 1 addition & 0 deletions dataset_zoo/wildreceipt/textrecog.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
_base_ = ['textdet.py']

data_converter = dict(type='TextRecogCropConverter')
config_generator = dict(type='TextRecogConfigGenerator')

0 comments on commit f6472ea

Please sign in to comment.