mirror of https://github.com/open-mmlab/mmocr.git
[TODO] mv configs/base/models -> configs/textdet & textrecog
parent
d8c3aeff3a
commit
27261b2bce
|
@ -1,7 +1,6 @@
|
|||
_base_ = [
|
||||
'../../_base_/default_runtime.py',
|
||||
'dbnet_r18_fpnc.py', '../../_base_/default_runtime.py',
|
||||
'../../_base_/schedules/schedule_sgd_100k_iters.py',
|
||||
'../../_base_/det_models/dbnet_r18_fpnc.py',
|
||||
'../../_base_/det_datasets/synthtext.py',
|
||||
'../../_base_/det_pipelines/dbnet_pipeline.py'
|
||||
]
|
||||
|
|
|
@ -1,7 +1,7 @@
|
|||
_base_ = [
|
||||
'dbnet_r18_fpnc.py',
|
||||
'../../_base_/default_runtime.py',
|
||||
'../../_base_/schedules/schedule_sgd_1200e.py',
|
||||
'../../_base_/det_models/dbnet_r18_fpnc.py',
|
||||
]
|
||||
|
||||
default_hooks = dict(checkpoint=dict(type='CheckpointHook', interval=20), )
|
||||
|
|
|
@ -1,7 +1,6 @@
|
|||
_base_ = [
|
||||
'../../_base_/default_runtime.py',
|
||||
'dbnet_r50dcnv2_fpnc.py', '../../_base_/default_runtime.py',
|
||||
'../../_base_/schedules/schedule_sgd_100k_iters.py',
|
||||
'../../_base_/det_models/dbnet_r50dcnv2_fpnc.py',
|
||||
'../../_base_/det_datasets/synthtext.py',
|
||||
'../../_base_/det_pipelines/dbnet_pipeline.py'
|
||||
]
|
||||
|
|
|
@ -1,7 +1,7 @@
|
|||
_base_ = [
|
||||
'dbnet_r50dcnv2_fpnc.py',
|
||||
'../../_base_/default_runtime.py',
|
||||
'../../_base_/schedules/schedule_sgd_1200e.py',
|
||||
'../../_base_/det_models/dbnet_r50dcnv2_fpnc.py',
|
||||
]
|
||||
|
||||
default_hooks = dict(checkpoint=dict(type='CheckpointHook', interval=20), )
|
||||
|
|
|
@ -1,7 +1,6 @@
|
|||
_base_ = [
|
||||
'../../_base_/default_runtime.py',
|
||||
'dbnetpp_r50dcnv2_fpnc.py', '../../_base_/default_runtime.py',
|
||||
'../../_base_/schedules/schedule_sgd_100k_iters.py',
|
||||
'../../_base_/det_models/dbnetpp_r50dcnv2_fpnc.py',
|
||||
'../../_base_/det_datasets/synthtext.py',
|
||||
'../../_base_/det_pipelines/dbnet_pipeline.py'
|
||||
]
|
||||
|
|
|
@ -1,7 +1,6 @@
|
|||
_base_ = [
|
||||
'../../_base_/default_runtime.py',
|
||||
'dbnetpp_r50dcnv2_fpnc.py', '../../_base_/default_runtime.py',
|
||||
'../../_base_/schedules/schedule_sgd_1200e.py',
|
||||
'../../_base_/det_models/dbnetpp_r50dcnv2_fpnc.py',
|
||||
'../../_base_/det_datasets/icdar2015.py',
|
||||
'../../_base_/det_pipelines/dbnet_pipeline.py'
|
||||
]
|
||||
|
|
|
@ -1,7 +1,7 @@
|
|||
_base_ = [
|
||||
'drrg_r50_fpn_unet.py',
|
||||
'../../_base_/default_runtime.py',
|
||||
'../../_base_/schedules/schedule_sgd_1200e.py',
|
||||
'../../_base_/det_models/drrg_r50_fpn_unet.py',
|
||||
]
|
||||
|
||||
train_pipeline = [
|
||||
|
|
|
@ -1,7 +1,7 @@
|
|||
_base_ = [
|
||||
'fcenet_r50_fpn.py',
|
||||
'../../_base_/default_runtime.py',
|
||||
'../../_base_/schedules/schedule_sgd_1500e.py',
|
||||
'../../_base_/det_models/fcenet_r50_fpn.py',
|
||||
]
|
||||
|
||||
default_hooks = dict(
|
||||
|
|
|
@ -1,7 +1,7 @@
|
|||
_base_ = [
|
||||
'fcenet_r50_fpn.py',
|
||||
'../../_base_/default_runtime.py',
|
||||
'../../_base_/schedules/schedule_sgd_1500e.py',
|
||||
'../../_base_/det_models/fcenet_r50dcnv2_fpn.py',
|
||||
]
|
||||
|
||||
default_hooks = dict(
|
||||
|
|
|
@ -1,6 +1,6 @@
|
|||
_base_ = [
|
||||
'ocr_mask_rcnn_r50_fpn_ohem_poly.py',
|
||||
'../../_base_/default_runtime.py',
|
||||
'../../_base_/det_models/ocr_mask_rcnn_r50_fpn_ohem_poly.py',
|
||||
'../../_base_/schedules/schedule_sgd_160e.py',
|
||||
]
|
||||
|
||||
|
|
|
@ -1,6 +1,6 @@
|
|||
_base_ = [
|
||||
'ocr_mask_rcnn_r50_fpn_ohem_poly.py',
|
||||
'../../_base_/default_runtime.py',
|
||||
'../../_base_/det_models/ocr_mask_rcnn_r50_fpn_ohem_poly.py',
|
||||
'../../_base_/schedules/schedule_sgd_160e.py',
|
||||
]
|
||||
|
||||
|
|
|
@ -1,6 +1,6 @@
|
|||
_base_ = [
|
||||
'ocr_mask_rcnn_r50_fpn_ohem_poly.py',
|
||||
'../../_base_/default_runtime.py',
|
||||
'../../_base_/det_models/ocr_mask_rcnn_r50_fpn_ohem_poly.py',
|
||||
'../../_base_/schedules/schedule_sgd_160e.py',
|
||||
]
|
||||
|
||||
|
|
|
@ -1,7 +1,6 @@
|
|||
_base_ = [
|
||||
'../../_base_/default_runtime.py',
|
||||
'panet_r18_fpem_ffm.py', '../../_base_/default_runtime.py',
|
||||
'../../_base_/schedules/schedule_adam_600e.py',
|
||||
'../../_base_/det_models/panet_r18_fpem_ffm.py',
|
||||
'../../_base_/det_datasets/ctw1500.py',
|
||||
'../../_base_/det_pipelines/panet_pipeline.py'
|
||||
]
|
||||
|
|
|
@ -1,7 +1,7 @@
|
|||
_base_ = [
|
||||
'panet_r18_fpem_ffm.py',
|
||||
'../../_base_/default_runtime.py',
|
||||
'../../_base_/schedules/schedule_adam_600e.py',
|
||||
'../../_base_/det_models/panet_r18_fpem_ffm.py',
|
||||
]
|
||||
|
||||
train_pipeline = [
|
||||
|
|
|
@ -1,7 +1,6 @@
|
|||
_base_ = [
|
||||
'../../_base_/default_runtime.py',
|
||||
'panet_r50_fpem_ffm.py', '../../_base_/default_runtime.py',
|
||||
'../../_base_/schedules/schedule_adam_600e.py',
|
||||
'../../_base_/det_models/panet_r50_fpem_ffm.py',
|
||||
'../../_base_/det_datasets/icdar2017.py',
|
||||
'../../_base_/det_pipelines/panet_pipeline.py'
|
||||
]
|
||||
|
|
|
@ -1,8 +1,7 @@
|
|||
# TODO Train on ctw1500
|
||||
_base_ = [
|
||||
'../../_base_/default_runtime.py',
|
||||
'psenet_r50_fpnf.py', '../../_base_/default_runtime.py',
|
||||
'../../_base_/schedules/schedule_adam_step_600e.py',
|
||||
'../../_base_/det_models/psenet_r50_fpnf.py',
|
||||
'../../_base_/det_datasets/ctw1500.py',
|
||||
'../../_base_/det_pipelines/psenet_pipeline.py'
|
||||
]
|
||||
|
|
|
@ -1,7 +1,7 @@
|
|||
_base_ = [
|
||||
'psenet_r50_fpnf.py',
|
||||
'../../_base_/default_runtime.py',
|
||||
'../../_base_/schedules/schedule_adam_step_600e.py',
|
||||
'../../_base_/det_models/psenet_r50_fpnf.py',
|
||||
]
|
||||
|
||||
model = {{_base_.model_quad}}
|
||||
|
|
|
@ -1,8 +1,7 @@
|
|||
# TODO Train on ICDAR 2017
|
||||
_base_ = [
|
||||
'../../_base_/default_runtime.py',
|
||||
'psenet_r50_fpnf.py', '../../_base_/default_runtime.py',
|
||||
'../../_base_/schedules/schedule_sgd_600e.py',
|
||||
'../../_base_/det_models/psenet_r50_fpnf.py',
|
||||
'../../_base_/det_datasets/icdar2017.py',
|
||||
'../../_base_/det_pipelines/psenet_pipeline.py'
|
||||
]
|
||||
|
|
|
@ -1,7 +1,7 @@
|
|||
_base_ = [
|
||||
'textsnake_r50_fpn_unet.py',
|
||||
'../../_base_/default_runtime.py',
|
||||
'../../_base_/schedules/schedule_sgd_1200e.py',
|
||||
'../../_base_/det_models/textsnake_r50_fpn_unet.py',
|
||||
]
|
||||
|
||||
train_pipeline = [
|
||||
|
|
|
@ -1,3 +1,3 @@
|
|||
_base_ = ['../../_base_/recog_models/abinet.py', 'base.py']
|
||||
_base_ = ['abinet.py', 'base.py']
|
||||
|
||||
load_from = 'abinet_pretrain-1bed979b.pth'
|
||||
|
|
|
@ -1 +1 @@
|
|||
_base_ = ['../../_base_/recog_models/abinet_vision_only.py', 'base.py']
|
||||
_base_ = ['abinet_vision_only.py', 'base.py']
|
||||
|
|
|
@ -1,8 +1,8 @@
|
|||
# training schedule for 1x
|
||||
_base_ = [
|
||||
'crnn.py',
|
||||
'../../_base_/default_runtime.py',
|
||||
'../../_base_/schedules/schedule_adadelta_5e.py',
|
||||
'../../_base_/recog_models/crnn.py',
|
||||
]
|
||||
|
||||
default_hooks = dict(logger=dict(type='LoggerHook', interval=50), )
|
||||
|
|
|
@ -1,6 +1,6 @@
|
|||
_base_ = [
|
||||
'master.py',
|
||||
'../../_base_/default_runtime.py',
|
||||
'../../_base_/recog_models/master.py',
|
||||
'../../_base_/schedules/schedule_adam_step_12e.py',
|
||||
]
|
||||
|
||||
|
|
|
@ -1,6 +1,6 @@
|
|||
_base_ = [
|
||||
'master.py',
|
||||
'../../_base_/default_runtime.py',
|
||||
'../../_base_/recog_models/master.py',
|
||||
'../../_base_/schedules/schedule_adam_step_12e.py',
|
||||
]
|
||||
|
||||
|
|
|
@ -1,6 +1,5 @@
|
|||
_base_ = [
|
||||
'../../_base_/default_runtime.py',
|
||||
'../../_base_/recog_models/nrtr_modality_transform.py',
|
||||
'nrtr_modality_transform.py', '../../_base_/default_runtime.py',
|
||||
'../../_base_/schedules/schedule_adam_step_6e.py'
|
||||
]
|
||||
|
||||
|
|
|
@ -1,6 +1,5 @@
|
|||
_base_ = [
|
||||
'../../_base_/default_runtime.py',
|
||||
'../../_base_/recog_models/nrtr_modality_transform.py',
|
||||
'nrtr_modality_transform.py', '../../_base_/default_runtime.py',
|
||||
'../../_base_/schedules/schedule_adam_step_6e.py'
|
||||
]
|
||||
|
||||
|
|
|
@ -1,6 +1,5 @@
|
|||
_base_ = [
|
||||
'../../_base_/default_runtime.py',
|
||||
'../../_base_/recog_models/robust_scanner.py',
|
||||
'robust_scanner.py', '../../_base_/default_runtime.py',
|
||||
'../../_base_/schedules/schedule_adam_step_5e.py'
|
||||
]
|
||||
|
||||
|
|
|
@ -1,6 +1,6 @@
|
|||
_base_ = [
|
||||
'sar.py',
|
||||
'../../_base_/default_runtime.py',
|
||||
'../../_base_/recog_models/sar.py',
|
||||
'../../_base_/schedules/schedule_adam_step_5e.py',
|
||||
]
|
||||
|
||||
|
|
|
@ -1,5 +1,5 @@
|
|||
_base_ = [
|
||||
'../../_base_/default_runtime.py', '../../_base_/recog_models/sar.py',
|
||||
'sar.py', '../../_base_/default_runtime.py',
|
||||
'../../_base_/schedules/schedule_adam_step_5e.py',
|
||||
'../../_base_/recog_pipelines/sar_pipeline.py',
|
||||
'../../_base_/recog_datasets/toy_data.py'
|
||||
|
|
|
@ -1,6 +1,6 @@
|
|||
_base_ = [
|
||||
'sar.py',
|
||||
'../../_base_/default_runtime.py',
|
||||
'../../_base_/recog_models/sar.py',
|
||||
'../../_base_/schedules/schedule_adam_step_5e.py',
|
||||
]
|
||||
|
||||
|
|
|
@ -1,7 +1,7 @@
|
|||
_base_ = [
|
||||
'satrn.py'
|
||||
'../../_base_/default_runtime.py',
|
||||
'../../_base_/schedules/schedule_adam_step_5e.py',
|
||||
'../../_base_/recog_models/satrn.py'
|
||||
]
|
||||
|
||||
default_hooks = dict(logger=dict(type='LoggerHook', interval=50))
|
||||
|
|
Loading…
Reference in New Issue