Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleOCR
提交
53c7c497
P
PaddleOCR
项目概览
PaddlePaddle
/
PaddleOCR
大约 1 年 前同步成功
通知
1528
Star
32962
Fork
6643
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
108
列表
看板
标记
里程碑
合并请求
7
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleOCR
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
108
Issue
108
列表
看板
标记
里程碑
合并请求
7
合并请求
7
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
53c7c497
编写于
12月 08, 2021
作者:
R
redearly123
提交者:
GitHub
12月 08, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Update base_model.py (#4728)
上级
552f5a22
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
3 addition
and
3 deletion
+3
-3
ppocr/modeling/architectures/base_model.py
ppocr/modeling/architectures/base_model.py
+3
-3
未找到文件。
ppocr/modeling/architectures/base_model.py
浏览文件 @
53c7c497
...
@@ -35,7 +35,7 @@ class BaseModel(nn.Layer):
...
@@ -35,7 +35,7 @@ class BaseModel(nn.Layer):
model_type
=
config
[
'model_type'
]
model_type
=
config
[
'model_type'
]
# build transfrom,
# build transfrom,
# for rec, transfrom can be TPS,None
# for rec, transfrom can be TPS,None
# for det and cls, transfrom shoul
e to
be None,
# for det and cls, transfrom shoul
d
be None,
# if you make model differently, you can use transfrom in det and cls
# if you make model differently, you can use transfrom in det and cls
if
'Transform'
not
in
config
or
config
[
'Transform'
]
is
None
:
if
'Transform'
not
in
config
or
config
[
'Transform'
]
is
None
:
self
.
use_transform
=
False
self
.
use_transform
=
False
...
@@ -45,7 +45,7 @@ class BaseModel(nn.Layer):
...
@@ -45,7 +45,7 @@ class BaseModel(nn.Layer):
self
.
transform
=
build_transform
(
config
[
'Transform'
])
self
.
transform
=
build_transform
(
config
[
'Transform'
])
in_channels
=
self
.
transform
.
out_channels
in_channels
=
self
.
transform
.
out_channels
# build backbone, backbone is need for del, rec and cls
# build backbone, backbone is need
ed
for del, rec and cls
config
[
"Backbone"
][
'in_channels'
]
=
in_channels
config
[
"Backbone"
][
'in_channels'
]
=
in_channels
self
.
backbone
=
build_backbone
(
config
[
"Backbone"
],
model_type
)
self
.
backbone
=
build_backbone
(
config
[
"Backbone"
],
model_type
)
in_channels
=
self
.
backbone
.
out_channels
in_channels
=
self
.
backbone
.
out_channels
...
@@ -62,7 +62,7 @@ class BaseModel(nn.Layer):
...
@@ -62,7 +62,7 @@ class BaseModel(nn.Layer):
self
.
neck
=
build_neck
(
config
[
'Neck'
])
self
.
neck
=
build_neck
(
config
[
'Neck'
])
in_channels
=
self
.
neck
.
out_channels
in_channels
=
self
.
neck
.
out_channels
# # build head, head is need for det, rec and cls
# # build head, head is need
ed
for det, rec and cls
config
[
"Head"
][
'in_channels'
]
=
in_channels
config
[
"Head"
][
'in_channels'
]
=
in_channels
self
.
head
=
build_head
(
config
[
"Head"
])
self
.
head
=
build_head
(
config
[
"Head"
])
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录