fixed blender testskip

holzers · Mar 20, 2020 · 0247d6e · 0247d6e
1 parent e3b4749
commit 0247d6e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/config_lego.txt b/config_lego.txt
@@ -12,3 +12,5 @@ N_importance = 64
 use_viewdirs = True
 
 white_bkgd = True
+
+N_rand = 1024
diff --git a/load_blender.py b/load_blender.py
@@ -63,7 +63,7 @@ def load_blender_data(basedir, half_res=False, testskip=1):
             poses.append(np.array(frame['transform_matrix']))
         imgs = (np.array(imgs) / 255.).astype(np.float32) # keep all 4 channels (RGBA)
         poses = np.array(poses).astype(np.float32)
-        counts.append(counts[-1] + len(meta['frames']))
+        counts.append(counts[-1] + imgs.shape[0])
         all_imgs.append(imgs)
         all_poses.append(poses)
Original file line number	Diff line number	Diff line change
Expand Up		@@ -12,3 +12,5 @@ N_importance = 64
		use_viewdirs = True

		white_bkgd = True

		N_rand = 1024