use issue#1 as gallery

lab-yue · Oct 6, 2022 · 9e50a29 · 9e50a29
1 parent 6f790c0
commit 9e50a29
Show file tree

Hide file tree

Showing 5 changed files with 3 additions and 7 deletions.
diff --git a/assets/gallery.md b/assets/gallery.md
diff --git a/encoding.py b/encoding.py
@@ -26,10 +26,6 @@ def get_encoder(encoding, input_dim=3,
     elif encoding == 'tiledgrid':
         from gridencoder import GridEncoder
         encoder = GridEncoder(input_dim=input_dim, num_levels=num_levels, level_dim=level_dim, base_resolution=base_resolution, log2_hashmap_size=log2_hashmap_size, desired_resolution=desired_resolution, gridtype='tiled', align_corners=align_corners)
-
-    elif encoding == 'ash':
-        from ashencoder import AshEncoder
-        encoder = AshEncoder(input_dim=input_dim, output_dim=16, log2_hashmap_size=log2_hashmap_size, resolution=desired_resolution)
 
     else:
         raise NotImplementedError('Unknown encoding mode, choose from [None, frequency, sphere_harmonics, hashgrid, tiledgrid]')

diff --git a/main_nerf.py b/main_nerf.py
@@ -96,7 +96,7 @@
         else:
             test_loader = NeRFDataset(opt, device=device, type='test', H=opt.H, W=opt.W, size=100).dataloader()
             trainer.test(test_loader)
-            trainer.save_mesh(resolution=256)
+            # trainer.save_mesh(resolution=256)
 
     else:
 

diff --git a/nerf/provider.py b/nerf/provider.py
@@ -169,7 +169,7 @@ def collate(self, index):
         else:
             # circle pose
             phi = (index[0] / self.size) * 2 * np.pi
-            poses, dirs = circle_poses(self.device, return_dirs=self.opt.dir_text, radius=self.radius_range[1], theta=np.pi/2, phi=phi)
+            poses, dirs = circle_poses(self.device, return_dirs=self.opt.dir_text, radius=self.radius_range[1] * 1.2, theta=np.pi/3, phi=phi)
 
             # fixed focal
             fov = (self.fovy_range[1] + self.fovy_range[0]) / 2

diff --git a/readme.md b/readme.md
@@ -8,7 +8,7 @@ Examples generated from text prompt `a DSLR photo of a pineapple` viewed with th
 
 https://user-images.githubusercontent.com/25863658/194241493-f3e68f78-aefe-479e-a4a8-001424a61b37.mp4
 
-### [Gallery](assets/gallery.md) | [Update Logs](assets/update_logs.md)
+### [Gallery](https://github.com/ashawkey/stable-dreamfusion/issues/1) | [Update Logs](assets/update_logs.md)
 
 # Important Notice
 This project is a **work-in-progress**, and contains lots of differences from the paper. Also, many features are still not implemented now. **The current generation quality cannot match the results from the original paper, and still fail badly for many prompts.**