Skip to content

Commit a60dc65

Browse files
authored
Replace deprecated ResizeBilinear for EAST (#479)
1 parent 027e50b commit a60dc65

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

mindocr/models/necks/fpn.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -280,17 +280,17 @@ def __init__(self, in_channels, out_channels=128):
280280
def construct(self, features):
281281
f1, f2, f3, f4 = features
282282

283-
out = ops.ResizeBilinear(f3.shape[2:], True)(f4)
283+
out = ops.ResizeBilinearV2(True)(f4, f3.shape[2:])
284284
out = self.concat((out, f3))
285285
out = self.relu1(self.bn1(self.conv1(out)))
286286
out = self.relu2(self.bn2(self.conv2(out)))
287287

288-
out = ops.ResizeBilinear(f2.shape[2:], True)(out)
288+
out = ops.ResizeBilinearV2(True)(out, f2.shape[2:])
289289
out = self.concat((out, f2))
290290
out = self.relu3(self.bn3(self.conv3(out)))
291291
out = self.relu4(self.bn4(self.conv4(out)))
292292

293-
out = ops.ResizeBilinear(f1.shape[2:], True)(out)
293+
out = ops.ResizeBilinearV2(True)(out, f1.shape[2:])
294294
out = self.concat((out, f1))
295295
out = self.relu5(self.bn5(self.conv5(out)))
296296
out = self.relu6(self.bn6(self.conv6(out)))

0 commit comments

Comments
 (0)