diff --git a/assets/gallery.md b/assets/gallery.md deleted file mode 100644 index e69de29b..00000000 diff --git a/encoding.py b/encoding.py index f20afd21..9b43f475 100644 --- 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 index a6dbf08f..59fd390d 100644 --- 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 index 9e2f6dd6..dca27fdc 100644 --- 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 index 3e26a36d..c95207ab 100644 --- 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.**