ashawkey commited on
Commit
9e50a29
1 Parent(s): 6f790c0

use issue#1 as gallery

Browse files
Files changed (5) hide show
  1. assets/gallery.md +0 -0
  2. encoding.py +0 -4
  3. main_nerf.py +1 -1
  4. nerf/provider.py +1 -1
  5. readme.md +1 -1
assets/gallery.md DELETED
File without changes
encoding.py CHANGED
@@ -26,10 +26,6 @@ def get_encoder(encoding, input_dim=3,
26
  elif encoding == 'tiledgrid':
27
  from gridencoder import GridEncoder
28
  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)
29
-
30
- elif encoding == 'ash':
31
- from ashencoder import AshEncoder
32
- encoder = AshEncoder(input_dim=input_dim, output_dim=16, log2_hashmap_size=log2_hashmap_size, resolution=desired_resolution)
33
 
34
  else:
35
  raise NotImplementedError('Unknown encoding mode, choose from [None, frequency, sphere_harmonics, hashgrid, tiledgrid]')
 
26
  elif encoding == 'tiledgrid':
27
  from gridencoder import GridEncoder
28
  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)
 
 
 
 
29
 
30
  else:
31
  raise NotImplementedError('Unknown encoding mode, choose from [None, frequency, sphere_harmonics, hashgrid, tiledgrid]')
main_nerf.py CHANGED
@@ -96,7 +96,7 @@ if __name__ == '__main__':
96
  else:
97
  test_loader = NeRFDataset(opt, device=device, type='test', H=opt.H, W=opt.W, size=100).dataloader()
98
  trainer.test(test_loader)
99
- trainer.save_mesh(resolution=256)
100
 
101
  else:
102
 
 
96
  else:
97
  test_loader = NeRFDataset(opt, device=device, type='test', H=opt.H, W=opt.W, size=100).dataloader()
98
  trainer.test(test_loader)
99
+ # trainer.save_mesh(resolution=256)
100
 
101
  else:
102
 
nerf/provider.py CHANGED
@@ -169,7 +169,7 @@ class NeRFDataset:
169
  else:
170
  # circle pose
171
  phi = (index[0] / self.size) * 2 * np.pi
172
- poses, dirs = circle_poses(self.device, return_dirs=self.opt.dir_text, radius=self.radius_range[1], theta=np.pi/2, phi=phi)
173
 
174
  # fixed focal
175
  fov = (self.fovy_range[1] + self.fovy_range[0]) / 2
 
169
  else:
170
  # circle pose
171
  phi = (index[0] / self.size) * 2 * np.pi
172
+ 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)
173
 
174
  # fixed focal
175
  fov = (self.fovy_range[1] + self.fovy_range[0]) / 2
readme.md CHANGED
@@ -8,7 +8,7 @@ Examples generated from text prompt `a DSLR photo of a pineapple` viewed with th
8
 
9
  https://user-images.githubusercontent.com/25863658/194241493-f3e68f78-aefe-479e-a4a8-001424a61b37.mp4
10
 
11
- ### [Gallery](assets/gallery.md) | [Update Logs](assets/update_logs.md)
12
 
13
  # Important Notice
14
  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.**
 
8
 
9
  https://user-images.githubusercontent.com/25863658/194241493-f3e68f78-aefe-479e-a4a8-001424a61b37.mp4
10
 
11
+ ### [Gallery](https://github.com/ashawkey/stable-dreamfusion/issues/1) | [Update Logs](assets/update_logs.md)
12
 
13
  # Important Notice
14
  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.**