diff --git a/configs/textdet/psenet/README.md b/configs/textdet/psenet/README.md index a4665074..d7da8af2 100644 --- a/configs/textdet/psenet/README.md +++ b/configs/textdet/psenet/README.md @@ -19,7 +19,7 @@ |Method | Backbone|Extra Data | Training set | Test set | #epochs | Test size|Recall|Precision|Hmean|Download| |:------:| :------:|:------:|:------:|:------:|:------:|:------:|:------:|:------:|:------:|:------:| -|[PSENet-4s](/configs/textdet/psenet/psenet_r50_fpnf_600e_ctw1500.py) |ResNet50 |-|CTW1500 Train|CTW1500 Test|600|1280|0.771|0.793|0.782|[model](https://download.openmmlab.com/mmocr/textdet/psenet/psenet_r50_fpnf_600e_ctw1500-ac38d587.pth) \| [config](https://download.openmmlab.com/mmocr/textdet/psenet/psenet_r50_fpnf_600e_ctw1500.py)| +|[PSENet-4s](/configs/textdet/psenet/psenet_r50_fpnf_600e_ctw1500.py) |ResNet50 |-|CTW1500 Train|CTW1500 Test|600|1280|0.728|0.849|0.784|[model](https://download.openmmlab.com/mmocr/textdet/psenet/psenet_r50_fpnf_600e_ctw1500_20210401-216fed50.pth) | [config](https://download.openmmlab.com/mmocr/textdet/psenet/psenet_r50_fpnf_600e_ctw1500_20210401.py) | [log](https://download.openmmlab.com/mmocr/textdet/psenet/20210401_215421.log.json)| ### ICDAR2015 diff --git a/configs/textdet/psenet/psenet_r50_fpnf_600e_ctw1500.py b/configs/textdet/psenet/psenet_r50_fpnf_600e_ctw1500.py index d0cc47f0..e1f5fd04 100644 --- a/configs/textdet/psenet/psenet_r50_fpnf_600e_ctw1500.py +++ b/configs/textdet/psenet/psenet_r50_fpnf_600e_ctw1500.py @@ -1,7 +1,12 @@ -_base_ = [ - '../../_base_/schedules/schedule_sgd_600e.py', - '../../_base_/default_runtime.py' -] +_base_ = ['../../_base_/default_runtime.py'] + +# optimizer +optimizer = dict(type='Adam', lr=1e-4) +optimizer_config = dict(grad_clip=None) +# learning policy +lr_config = dict(policy='step', step=[200, 400]) +total_epochs = 600 + model = dict( type='PSENet', pretrained='torchvision://resnet50',