forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix srn fpn

This commit is contained in:
tink2123 2020-09-24 15:11:31 +08:00
parent 5e25ed64a2
commit a5fbd902fe
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ppocr/modeling/backbones/rec_resnet_fpn.py
View File

@ -100,11 +100,11 @@ class ResNet():
conv = self.basic_block( conv = self.basic_block(
input=conv, input=conv,
num_filters=num_filters[block], num_filters=num_filters[block],
stride=stride, stride=stride_list[block] if i == 0 else 1,
if_first=block == i == 0, is_first=block == i == 0,
name=conv_name) name=conv_name)
F.append(conv) F.append(conv)
print("F:", F)
base = F[-1] base = F[-1]
for i in [-2, -3]: for i in [-2, -3]:
b, c, w, h = F[i].shape b, c, w, h = F[i].shape