Skip to content

Commit 934feff

Browse files
author
zylo117
committed
2 parents 1df4ed2 + dc738a4 commit 934feff

1 file changed

Lines changed: 10 additions & 0 deletions

File tree

efficientnet/utils_extra.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@ 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
<<<<<<< HEAD
3738
extra_h = (w - 1) * self.stride[1] - w + self.kernel_size[1]
3839
extra_v = (h - 1) * self.stride[0] - h + self.kernel_size[0]
@@ -43,6 +44,10 @@ def forward(self, x):
4344
extra_h = h_cover_len - w
4445
extra_v = v_cover_len - h
4546
>>>>>>> 21771ddb20591a2ff85cb2227f586e63f72c94f8
47+
=======
48+
extra_h = (math.ceil(w / self.stride[1]) - 1) * self.stride[1] - w + self.kernel_size[1]
49+
extra_v = (math.ceil(h / self.stride[0]) - 1) * self.stride[0] - h + self.kernel_size[0]
50+
>>>>>>> dc738a48daad9cc3530dd1beee5f4235b363687a
4651

4752
left = extra_h // 2
4853
right = extra_h - left
@@ -80,13 +85,18 @@ def __init__(self, *args, **kwargs):
8085
def forward(self, x):
8186
h, w = x.shape[-2:]
8287

88+
<<<<<<< HEAD
8389
<<<<<<< HEAD
8490
extra_h = (w - 1) * self.stride[1] - w + self.kernel_size[1]
8591
extra_v = (h - 1) * self.stride[0] - h + self.kernel_size[0]
8692
=======
8793
extra_h = (math.ceil(w / self.stride[1]) - 1) * self.stride[1] - w + self.kernel_size[1]
8894
extra_v = (math.ceil(h / self.stride[0]) - 1) * self.stride[0] - h + self.kernel_size[0]
8995
>>>>>>> 21771ddb20591a2ff85cb2227f586e63f72c94f8
96+
=======
97+
extra_h = (math.ceil(w / self.stride[1]) - 1) * self.stride[1] - w + self.kernel_size[1]
98+
extra_v = (math.ceil(h / self.stride[0]) - 1) * self.stride[0] - h + self.kernel_size[0]
99+
>>>>>>> dc738a48daad9cc3530dd1beee5f4235b363687a
90100

91101
left = extra_h // 2
92102
right = extra_h - left

0 commit comments

Comments
 (0)