bugfix: use hidden_dim_color instead of hidden_dim for color_net (ash…

…awkey#110)
niki-amini-naieni · Sep 20, 2022 · 2a5d4d0 · 2a5d4d0
1 parent 6d5016f
commit 2a5d4d0
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/dnerf/network.py b/dnerf/network.py
@@ -84,12 +84,12 @@ def __init__(self,
  if l == 0:
  in_dim = self.in_dim_dir + self.geo_feat_dim
  else:
- in_dim = hidden_dim
+ in_dim = hidden_dim_color
 
  if l == num_layers_color - 1:
  out_dim = 3 # 3 rgb
  else:
- out_dim = hidden_dim
+ out_dim = hidden_dim_color
 
  color_net.append(nn.Linear(in_dim, out_dim, bias=False))
 

diff --git a/dnerf/network_basis.py b/dnerf/network_basis.py
@@ -84,12 +84,12 @@ def __init__(self,
  if l == 0:
  in_dim = self.in_dim_dir + self.geo_feat_dim
  else:
- in_dim = hidden_dim
+ in_dim = hidden_dim_color
 
  if l == num_layers_color - 1:
  out_dim = 3 * self.color_basis_dim # 3 * CB rgb
  else:
- out_dim = hidden_dim
+ out_dim = hidden_dim_color
 
  color_net.append(nn.Linear(in_dim, out_dim, bias=False))
 

diff --git a/dnerf/network_hyper.py b/dnerf/network_hyper.py
@@ -82,12 +82,12 @@ def __init__(self,
  if l == 0:
  in_dim = self.in_dim_dir + self.geo_feat_dim
  else:
- in_dim = hidden_dim
+ in_dim = hidden_dim_color
 
  if l == num_layers_color - 1:
  out_dim = 3 # 3 rgb
  else:
- out_dim = hidden_dim
+ out_dim = hidden_dim_color
 
  color_net.append(nn.Linear(in_dim, out_dim, bias=False))
 

diff --git a/nerf/network.py b/nerf/network.py
@@ -56,12 +56,12 @@ def __init__(self,
  if l == 0:
  in_dim = self.in_dim_dir + self.geo_feat_dim
  else:
- in_dim = hidden_dim
+ in_dim = hidden_dim_color
 
  if l == num_layers_color - 1:
  out_dim = 3 # 3 rgb
  else:
- out_dim = hidden_dim
+ out_dim = hidden_dim_color
 
  color_net.append(nn.Linear(in_dim, out_dim, bias=False))