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

fix : update ckpt path of psenet ()

pull/2/head
Hongbin Sun 2021-04-04 14:27:37 +08:00 committed by GitHub
parent d48f212baa
commit 12df13244d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 5 deletions

2
configs/textdet/psenet/README.md
View File

@ -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

13
configs/textdet/psenet/psenet_r50_fpnf_600e_ctw1500.py
View File

@ -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',