From 91c4a6a90510a9efd492ba41dea3f93b1e58006e Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Fri, 25 Jun 2021 12:44:46 +0200 Subject: Add missing zero pad --- text_recognizer/networks/encoders/efficientnet/efficientnet.py | 1 + 1 file changed, 1 insertion(+) diff --git a/text_recognizer/networks/encoders/efficientnet/efficientnet.py b/text_recognizer/networks/encoders/efficientnet/efficientnet.py index b527d90..283b686 100644 --- a/text_recognizer/networks/encoders/efficientnet/efficientnet.py +++ b/text_recognizer/networks/encoders/efficientnet/efficientnet.py @@ -47,6 +47,7 @@ class EfficientNet(nn.Module): in_channels = 1 # BW out_channels = round_filters(32, self.arch) self._conv_stem = nn.Sequential( + nn.ZeroPad2d((0, 1, 0, 1)), nn.Conv2d( in_channels=in_channels, out_channels=out_channels, -- cgit v1.2.3-70-g09d2