mirrors
/
PaddleOCR
mirror of https://github.com/PaddlePaddle/PaddleOCR.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #5954 from WenmuZhou/whl 

rm to tensor in  fce head
pull/5982/head^2
zhoujun 2022-04-13 23:41:41 +08:00 committed by GitHub
parent a2050ad123 0f5fb9bd35
commit b8e7c6cef7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ppocr/modeling/heads/det_fce_head.py
View File

@ -63,7 +63,7 @@ class FCEHead(nn.Layer):
weight_attr=ParamAttr( weight_attr=ParamAttr(
name='cls_weights', name='cls_weights',
initializer=Normal( initializer=Normal(
mean=paddle.to_tensor(0.), std=paddle.to_tensor(0.01))), mean=0., std=0.01)),
bias_attr=True) bias_attr=True)
self.out_conv_reg = nn.Conv2D( self.out_conv_reg = nn.Conv2D(
in_channels=self.in_channels, in_channels=self.in_channels,
@ -75,7 +75,7 @@ class FCEHead(nn.Layer):
weight_attr=ParamAttr( weight_attr=ParamAttr(
name='reg_weights', name='reg_weights',
initializer=Normal( initializer=Normal(
mean=paddle.to_tensor(0.), std=paddle.to_tensor(0.01))), mean=0., std=0.01)),
bias_attr=True) bias_attr=True)
def forward(self, feats, targets=None): def forward(self, feats, targets=None):