otthorn/nerf_plus_plus
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

minor fix

Browse source
This commit is contained in:
Kai Zhang 2020-10-15 23:30:57 -04:00 committed by GitHub
parent 19fbf43032
commit 77d5146ca4
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
nerf_network.py
View file

@ -111,7 +111,7 @@ class MLPNet(nn.Module):
for i in range(1): for i in range(1):
rgb_layers.append(nn.Linear(dim, W // 2)) rgb_layers.append(nn.Linear(dim, W // 2))
rgb_layers.append(nn.ReLU()) rgb_layers.append(nn.ReLU())
dim = W dim = W // 2
rgb_layers.append(nn.Linear(dim, 3)) rgb_layers.append(nn.Linear(dim, 3))
rgb_layers.append(nn.Sigmoid()) # rgb values are normalized to [0, 1] rgb_layers.append(nn.Sigmoid()) # rgb values are normalized to [0, 1]
self.rgb_layers = nn.Sequential(*rgb_layers) self.rgb_layers = nn.Sequential(*rgb_layers)