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

请问能否公布一下不采用预训练的config文件么 #18

Open
921162820 opened this issue Apr 21, 2023 · 11 comments
Open

请问能否公布一下不采用预训练的config文件么 #18

921162820 opened this issue Apr 21, 2023 · 11 comments

Comments

@921162820
Copy link

比如说CTW1500或Totaltext直接训练的训练参数如何设置呢。

我已经知道了怎么训练自己的数据集,就是通过textsnake中的代码获取两条边,但算力有限,想知道不预训练该怎么跑呢。

@ymy-k
Copy link
Owner

ymy-k commented Apr 21, 2023

直接用pretrain的学习率设置就行,iteration和学习率下降的节点需要调整,因为没预训练的时候置信度不会很高,可以把INFERENCE_TH_TEST往下调

@dhddxdhd
Copy link

作者您好,我之前用pretrain的学习率跑了一下totaltext的代码,标签是用的TESTR论文的,没有用您的,因为您的那个太大了,也没用旋转数据增强,结果最高只有84.3。 参数的话,我把INFERENCE_TH_TEST调成了0.35,STEPS改成了100000,MAX_ITER是160000,请问哪里出问题了呢,为什么跑不到您实验中所说的85.几呢。 恳请作者告知一下,非常感谢,因为跑这个需要的卡太多了,跑一次太不容易了

@ymy-k
Copy link
Owner

ymy-k commented Apr 24, 2023

我用的INFERENCE_TH_TEST是0.3

@dhddxdhd
Copy link

我用的INFERENCE_TH_TEST是0.3

那请问step跟max_iter该设置成多少呢,我再跑一下试试

@ymy-k
Copy link
Owner

ymy-k commented Apr 24, 2023

max_iter我只用了120K,step应该是100K

@dhddxdhd
Copy link

作者您好,我仅调了SYEPS,MAX_ITER, INFERNEC_TH_TEST,发现还是只有84,请问还有其他需要改动的么,我刚入门,就想先复现好精度后再试试怎么改进。

@dhddxdhd
Copy link

或者我跑一下您发布的旋转数据集试试,请问旋转数据集该如何调整MAX_ITER和STEPS呢,恳请作者指点一下,呜呜呜两三天才跑一次实验太慢了。

@xieliaolieji
Copy link

同问,请问采用作者您提供的旋转的totaltext数据集,不预训练的话需要把Max_iter和Steps等参数调整为多少呢,我之前跑过一次hmean最高只有85.8。

@ymy-k
Copy link
Owner

ymy-k commented May 11, 2023

论文里有提到200K。step160K,85.8也还正常,不同显卡对deformable attention似乎有比较明显的影响

@xieliaolieji
Copy link

论文里有提到200K。step160K,85.8也还正常,不同显卡对deformable attention似乎有比较明显的影响

噢噢,感谢指点,我没有训那么多次,因为我觉得毕竟数据集都扩充了10倍了嘛,那我换A100再试一下,感谢感谢。

@lerndeep
Copy link

@921162820
How to prepare data for training, data example:
x1,y1,x2,y2,x3,y3,x4,y4,text
x1,y1,x2,y2,x3,y3,x4,y4,text
x1,y1,x2,y2,x3,y3,x4,y4,text
Each annotation text file contains bounding box coordinates with corresponding text.
could you please share your code?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

5 participants