Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

MSRA-TD500数据集的config文件能否公布下呢 #537

Closed
6098669 opened this issue Apr 5, 2022 · 3 comments
Closed

MSRA-TD500数据集的config文件能否公布下呢 #537

6098669 opened this issue Apr 5, 2022 · 3 comments
Labels
good first issue Good for newcomers

Comments

@6098669
Copy link

6098669 commented Apr 5, 2022

这个数据集太小了一直很难复现出ABCNetV2中的结果,跪求MSRA-TD500的config文件

@Yuliang-Liu
Copy link
Contributor

@6098669 Did you use the Chinese pretrained model to finetune on MSRA-TD500? The config for MSRA-TD500 as well as the models will be scheduled to release.

@ocrhei
Copy link

ocrhei commented Jun 2, 2022

同求MSRA-TD500数据集和config文件

@mxin262
Copy link

mxin262 commented Jun 6, 2022

I reproduced the results of the MSRD-TD500. You can reproduce it as follow.

(1) According to the paper, directly use the official Chinese pretrained model.
f7e12db1f1417e23244452fd0ee9b3e

(2) Change the config as follow.
e722f541a99184296e753b7bac8ded6
And you also need to change the config to use the training data of MSRA-TD500.

(3) Add the path of MSRA-TD500 here. The train&test data can be downloaded from BaiduNetDisk Passwd: b8ki

(4) Because the MSRA-TD500 does not involve recognition training, the following modifications are made here.
image
image

(5) Train the model.
OMP_NUM_THREADS=1 python tools/train_net.py --config-file configs/BAText/TotalText/v2_attn_R_50.yaml --num-gpus 2 MODEL.WEIGHTS model_v2_chn_pretrain.pth

(6) This is the result of my reproduction, which is consistent with the paper.
image

@Yuliang-Liu Yuliang-Liu added the good first issue Good for newcomers label Jun 19, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
good first issue Good for newcomers
Projects
None yet
Development

No branches or pull requests

4 participants