Merge pull request #834 from tink2123/fix_srn

Fix srn
2020-09-24 15:56:15 +08:00 · 2020-09-24 15:56:15 +08:00 · 815f20b550
parent 5e25ed64a2 5f8c3e07d9
commit 815f20b550
1 changed files with 2 additions and 3 deletions
--- a/ppocr/modeling/backbones/rec_resnet_fpn.py
+++ b/ppocr/modeling/backbones/rec_resnet_fpn.py
@ -100,11 +100,10 @@ class ResNet():
                    conv = self.basic_block(
                        input=conv,
                        num_filters=num_filters[block],
-                        stride=stride,
-                        if_first=block == i == 0,
+                        stride=stride_list[block] if i == 0 else 1,
+                        is_first=block == i == 0,
                        name=conv_name)
                F.append(conv)
-
        base = F[-1]
        for i in [-2, -3]:
            b, c, w, h = F[i].shape