use issue#1 as gallery
Browse files- assets/gallery.md +0 -0
- encoding.py +0 -4
- main_nerf.py +1 -1
- nerf/provider.py +1 -1
- 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/
|
| 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](
|
| 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.**
|