Skip to content

Commit 1df4ed2

Browse files
author
zylo117
committed
2 parents c6a650f + 21771dd commit 1df4ed2

1 file changed

Lines changed: 13 additions & 0 deletions

File tree

efficientnet/utils_extra.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,8 +33,16 @@ def __init__(self, in_channels, out_channels, kernel_size, stride=1, bias=True,
3333
def forward(self, x):
3434
h, w = x.shape[-2:]
3535

36+
<<<<<<< HEAD
3637
extra_h = (w - 1) * self.stride[1] - w + self.kernel_size[1]
3738
extra_v = (h - 1) * self.stride[0] - h + self.kernel_size[0]
39+
=======
40+
extra_h = (math.ceil(w / self.stride[1]) - 1) * self.stride[1] - w + self.kernel_size[1]
41+
extra_v = (math.ceil(h / self.stride[0]) - 1) * self.stride[0] - h + self.kernel_size[0]
42+
43+
extra_h = h_cover_len - w
44+
extra_v = v_cover_len - h
45+
>>>>>>> 21771ddb20591a2ff85cb2227f586e63f72c94f8
3846

3947
left = extra_h // 2
4048
right = extra_h - left
@@ -72,8 +80,13 @@ def __init__(self, *args, **kwargs):
7280
def forward(self, x):
7381
h, w = x.shape[-2:]
7482

83+
<<<<<<< HEAD
7584
extra_h = (w - 1) * self.stride[1] - w + self.kernel_size[1]
7685
extra_v = (h - 1) * self.stride[0] - h + self.kernel_size[0]
86+
=======
87+
extra_h = (math.ceil(w / self.stride[1]) - 1) * self.stride[1] - w + self.kernel_size[1]
88+
extra_v = (math.ceil(h / self.stride[0]) - 1) * self.stride[0] - h + self.kernel_size[0]
89+
>>>>>>> 21771ddb20591a2ff85cb2227f586e63f72c94f8
7790

7891
left = extra_h // 2
7992
right = extra_h - left

0 commit comments

Comments
 (0)