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

add text detection algorithm DRRG #7832

Merged
merged 6 commits into from
Oct 13, 2022

Conversation

zhiminzhang0830
Copy link
Contributor

No description provided.

@paddle-bot
Copy link

paddle-bot bot commented Oct 8, 2022

Thanks for your contribution!

Backbone:
name: ResNet_vd
layers: 50
pretrained_model: ./pretrain_models/ResNet50_vd_ssld_pretrained.pdparams
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

这个字段不要加在模型里,用Global.pretrained_model字段进行预训练模型的设置

if pretrained_model is not None:
self.load_pretrained_params(pretrained_model)

def load_pretrained_params(self, path):
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

这个删掉吧,用之前的预训练模型加载方式

return x


class FPN_UNet(nn.Layer):
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

这个fpn看看能不能用已有的fpn模块替代

@WenmuZhou
Copy link
Collaborator

WenmuZhou commented Oct 8, 2022

还需要修改infer_det.py,做训练引擎的预测支持,以及添加文档

@WenmuZhou WenmuZhou merged commit ce72023 into PaddlePaddle:dygraph Oct 13, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants