luni/reconstruction
1
0
Fork 0
Code Issues 6 Pull Requests Packages Projects 1 Releases Wiki Activity

change feature size based on 256x256 input

Browse Source
This commit is contained in:
luni
2023-01-18 23:19:53 +00:00
parent 1e1f9e2e09
commit f4341c5e6c
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/depth_estim.py
+2 -1
View File
@@ -26,10 +26,11 @@ class ViewEncoder(nn.Module):
x = self.act1(x) x = self.act1(x)
x = self.lin2(x) x = self.lin2(x)
x = self.act2(x) x = self.act2(x)
x = torch.ones((x.shape[0], self.features, 15, 20), device=x.get_device()) * x[:,:,None,None] x = torch.ones((x.shape[0], self.features, 8, 8), device=x.get_device()) * x[:,:,None,None]
return x return x
from torchvision.ops.misc import Conv2dNormActivation, Permute from torchvision.ops.misc import Conv2dNormActivation, Permute
class DumbReconstructor(nn.Module): class DumbReconstructor(nn.Module):