fixed normalization, can overfit on single image

2023-01-24 22:45:14 +00:00
parent a0f3e95255
commit d062c2068d
6 changed files with 147 additions and 72 deletions
@@ -108,6 +108,7 @@ class DepthEstimation(nn.Module):
        view_features = self.view_encoder(view)
        # x = self.flatten(x)
        x = self.decoder([torch.concat((x, view_features), axis=1), features])
+        x = torch.tanh(x[0]), x[1]
        # x = self.decoder(x)
        
        return x