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

[Config] Fix NRTR config (#1232)

pull/1247/head
Tong Gao 2022-08-01 15:34:41 +08:00 committed by GitHub
parent cdba3056c0
commit c9ec09d8f1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 19 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
configs/textrecog/nrtr/nrtr_modality_transform_academic.py
View File

@ -6,7 +6,7 @@ _base_ = [
]
# optimizer settings
optimizer = dict(type='Adam', lr=3e-4)
optim_wrapper = dict(type='OptimWrapper', optimizer=dict(type='Adam', lr=3e-4))
# dataset settings
train_list = {{_base_.train_list}}
@ -15,7 +15,11 @@ file_client_args = dict(backend='disk')
default_hooks = dict(logger=dict(type='LoggerHook', interval=50), )
train_pipeline = [
dict(type='LoadImageFromFile', file_client_args=file_client_args),
dict(
type='LoadImageFromFile',
file_client_args=file_client_args,
ignore_empty=True,
min_size=5),
dict(type='LoadOCRAnnotations', with_text=True),
dict(
type='RescaleToHeight',

10
configs/textrecog/nrtr/nrtr_r31_1by16_1by8_academic.py
View File

@ -6,7 +6,7 @@ _base_ = [
]
# optimizer settings
optimizer = dict(type='Adam', lr=3e-4)
optim_wrapper = dict(type='OptimWrapper', optimizer=dict(type='Adam', lr=3e-4))
# dataset settings
train_list = {{_base_.train_list}}
@ -15,7 +15,11 @@ file_client_args = dict(backend='disk')
default_hooks = dict(logger=dict(type='LoggerHook', interval=50))
train_pipeline = [
dict(type='LoadImageFromFile', file_client_args=file_client_args),
dict(
type='LoadImageFromFile',
file_client_args=file_client_args,
ignore_empty=True,
min_size=5),
dict(type='LoadOCRAnnotations', with_text=True),
dict(
type='RescaleToHeight',
@ -46,7 +50,7 @@ test_pipeline = [
train_dataloader = dict(
batch_size=384,
num_workers=32,
num_workers=24,
persistent_workers=True,
sampler=dict(type='DefaultSampler', shuffle=True),
dataset=dict(

8
configs/textrecog/nrtr/nrtr_r31_1by8_1by4_academic.py
View File

@ -6,7 +6,7 @@ _base_ = [
]
# optimizer settings
optimizer = dict(type='Adam', lr=3e-4)
optim_wrapper = dict(type='OptimWrapper', optimizer=dict(type='Adam', lr=3e-4))
# dataset settings
train_list = {{_base_.train_list}}
@ -17,7 +17,11 @@ default_hooks = dict(logger=dict(type='LoggerHook', interval=50), )
model = dict(backbone=dict(last_stage_pool=False))
train_pipeline = [
dict(type='LoadImageFromFile', file_client_args=file_client_args),
dict(
type='LoadImageFromFile',
file_client_args=file_client_args,
ignore_empty=True,
min_size=5),
dict(type='LoadOCRAnnotations', with_text=True),
dict(
type='RescaleToHeight',