mirrors
/
mmocr
mirror of https://github.com/open-mmlab/mmocr.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[Config] Update rec configs (#1417)

pull/1434/head
Tong Gao 2022-10-09 12:43:45 +08:00 committed by GitHub
parent 3d015462e7
commit b26907e908
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
15 changed files with 18 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
configs/textrecog/abinet/_base_abinet-vision.py
View File

@ -46,7 +46,7 @@ train_pipeline = [
type='LoadImageFromFile',
file_client_args=file_client_args,
ignore_empty=True,
min_size=5),
min_size=2),
dict(type='LoadOCRAnnotations', with_text=True),
dict(type='Resize', scale=(128, 32)),
dict(

2
configs/textrecog/abinet/abinet_20e_st-an_mj.py
View File

@ -37,7 +37,7 @@ test_dataset = dict(
type='ConcatDataset', datasets=test_list, pipeline=_base_.test_pipeline)
train_dataloader = dict(
batch_size=192 * 4,
batch_size=192,
num_workers=32,
persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True),

2
configs/textrecog/crnn/_base_crnn_mini-vgg.py
View File

@ -25,7 +25,7 @@ train_pipeline = [
color_type='grayscale',
file_client_args=file_client_args,
ignore_empty=True,
min_size=5),
min_size=2),
dict(type='LoadOCRAnnotations', with_text=True),
dict(type='Resize', scale=(100, 32), keep_ratio=False),
dict(

2
configs/textrecog/crnn/crnn_mini-vgg_5e_mj.py
View File

@ -23,7 +23,7 @@ default_hooks = dict(logger=dict(type='LoggerHook', interval=50), )
train_dataloader = dict(
batch_size=64,
num_workers=8,
num_workers=24,
persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True),
dataset=dict(

2
configs/textrecog/master/_base_master_resnet31.py
View File

@ -79,7 +79,7 @@ train_pipeline = [
type='LoadImageFromFile',
file_client_args=file_client_args,
ignore_empty=True,
min_size=5),
min_size=2),
dict(type='LoadOCRAnnotations', with_text=True),
dict(
type='RescaleToHeight',

2
configs/textrecog/master/master_resnet31_12e_st_mj_sa.py
View File

@ -37,7 +37,7 @@ test_dataset = dict(
train_dataloader = dict(
batch_size=512,
num_workers=4,
num_workers=24,
persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True),
dataset=train_dataset)

2
configs/textrecog/nrtr/_base_nrtr_modality-transform.py
View File

@ -30,7 +30,7 @@ train_pipeline = [
type='LoadImageFromFile',
file_client_args=file_client_args,
ignore_empty=True,
min_size=5),
min_size=2),
dict(type='LoadOCRAnnotations', with_text=True),
dict(
type='RescaleToHeight',

2
configs/textrecog/nrtr/_base_nrtr_resnet31.py
View File

@ -36,7 +36,7 @@ train_pipeline = [
type='LoadImageFromFile',
file_client_args=file_client_args,
ignore_empty=True,
min_size=5),
min_size=2),
dict(type='LoadOCRAnnotations', with_text=True),
dict(
type='RescaleToHeight',

2
configs/textrecog/nrtr/nrtr_modality-transform_6e_st_mj.py
View File

@ -33,7 +33,7 @@ test_dataset = dict(
train_dataloader = dict(
batch_size=384,
num_workers=32,
num_workers=24,
persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True),
dataset=train_dataset)

2
configs/textrecog/robust_scanner/_base_robustscanner_resnet31.py
View File

@ -36,7 +36,7 @@ train_pipeline = [
type='LoadImageFromFile',
file_client_args=file_client_args,
ignore_empty=True,
min_size=5),
min_size=2),
dict(type='LoadOCRAnnotations', with_text=True),
dict(
type='RescaleToHeight',

4
configs/textrecog/robust_scanner/robustscanner_resnet31_5e_st-sub_mj-sub_sa_real.py
View File

@ -43,8 +43,8 @@ train_list = [
]
train_dataloader = dict(
batch_size=64,
num_workers=8,
batch_size=64 * 4,
num_workers=24,
persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True),
dataset=dict(type='ConcatDataset', datasets=train_list, verify_meta=False))

2
configs/textrecog/sar/_base_sar_resnet31_parallel-decoder.py
View File

@ -41,7 +41,7 @@ train_pipeline = [
type='LoadImageFromFile',
file_client_args=file_client_args,
ignore_empty=True,
min_size=5),
min_size=2),
dict(type='LoadOCRAnnotations', with_text=True),
dict(
type='RescaleToHeight',

4
configs/textrecog/sar/sar_resnet31_parallel-decoder_5e_st-sub_mj-sub_sa_real.py
View File

@ -43,8 +43,8 @@ train_list = [
]
train_dataloader = dict(
batch_size=64,
num_workers=8,
batch_size=64 * 6,
num_workers=24,
persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True),
dataset=dict(type='ConcatDataset', datasets=train_list, verify_meta=False))

2
configs/textrecog/satrn/_base_satrn_shallow.py
View File

@ -46,7 +46,7 @@ train_pipeline = [
type='LoadImageFromFile',
file_client_args=file_client_args,
ignore_empty=True,
min_size=5),
min_size=2),
dict(type='LoadOCRAnnotations', with_text=True),
dict(type='Resize', scale=(100, 32), keep_ratio=False),
dict(

4
configs/textrecog/satrn/satrn_shallow_5e_st_mj.py
View File

@ -28,8 +28,8 @@ test_dataset = dict(
optim_wrapper = dict(type='OptimWrapper', optimizer=dict(type='Adam', lr=3e-4))
train_dataloader = dict(
batch_size=64,
num_workers=8,
batch_size=128,
num_workers=24,
persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True),
dataset=train_dataset)