Laurent/REVA-QCAV
1
0
Fork 0
mirror of https://github.com/Laurent2916/REVA-QCAV.git synced 2024-09-19 19:45:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

diffX and diffY is now an int instead of a Tensor

Browse source 
Former-commit-id: 342d4e0a8ca19bbee77256caeda0ef6537d746f8
This commit is contained in:
HansBambel 2020-04-24 15:57:53 +02:00
parent c57180e90c
commit ef14c0e829
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
unet/unet_parts.py
View file

@ -57,8 +57,8 @@ class Up(nn.Module):
def forward(self, x1, x2): def forward(self, x1, x2):
x1 = self.up(x1) x1 = self.up(x1)
# input is CHW # input is CHW
diffY = torch.tensor([x2.size()[2] - x1.size()[2]]) diffY = x2.size()[2] - x1.size()[2]
diffX = torch.tensor([x2.size()[3] - x1.size()[3]]) diffX = x2.size()[3] - x1.size()[3]
x1 = F.pad(x1, [diffX // 2, diffX - diffX // 2, x1 = F.pad(x1, [diffX // 2, diffX - diffX // 2,
diffY // 2, diffY - diffY // 2]) diffY // 2, diffY - diffY // 2])