Update extract_color_mesh.py

tne-ai · Jun 22, 2020 · 34bb7a9 · 34bb7a9
1 parent 1829552
commit 34bb7a9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/extract_color_mesh.py b/extract_color_mesh.py
@@ -209,7 +209,7 @@ def f(models, embeddings, rays, N_samples, N_importance, chunk, white_back):
 
         # Step 2. project the vertices onto each training image to infer the color
         print('Fusing colors ...')
-        for idx in tqdm(range(len(dataset.image_paths[:1]))):
+        for idx in tqdm(range(len(dataset.image_paths))):
             ## read image of this pose
             image = cv2.imread(dataset.image_paths[idx])[:,:,::-1]
             image = cv2.resize(image, tuple(args.img_wh))
@@ -294,4 +294,4 @@ def f(models, embeddings, rays, N_samples, N_importance, chunk, white_back):
     PlyData([PlyElement.describe(vertex_all, 'vertex'), 
              PlyElement.describe(face, 'face')]).write(f'{args.scene_name}.ply')
 
-    print('Done!')
+    print('Done!')