yujunjun/PaddleOCR
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2025-12-27 23:18:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
LDOUBLEV 2022-04-26 17:56:54 +08:00
parent 62a01f7e30
commit 10c5773b9b
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ppocr/modeling/necks/db_fpn.py
View File

@ -120,6 +120,7 @@ class CALayer(nn.Layer):
def __init__(self, in_channels, out_channels, kernel_size, shortcut=True):
super(CALayer, self).__init__()
weight_attr = paddle.nn.initializer.KaimingUniform()
self.out_channels = out_channels
self.in_conv = nn.Conv2D(
in_channels=in_channels,
out_channels=self.out_channels,