4
0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-10 18:17:31 +08:00
Commit Graph

2063 Commits

Author SHA1 Message Date
hturki
1d070f5625
support generic TDataset in paralleldatamanager (#2884)
* support generic TDataset in paralleldatamanager

* run black

* run ruff

* run ruff
2024-02-07 16:50:15 -08:00
Frédéric Devernay
66e119485d
Remove useless call to .with_suffix(""), which causes a bug if filename contains dots (#2882)
* Remove useless call to .with_suffix(""), which causes a bug if filename contains dots
Fixes https://github.com/nerfstudio-project/nerfstudio/issues/2881

---------

Co-authored-by: Frédéric Devernay <f-dy@users.noreply.github.com>
2024-02-07 11:16:40 -08:00
Chung Min Kim
dc53517565
Add default depth output when no gaussians visible (#2874)
* Add default depth output when no gaussians visible

* Add accumulation to default output return

* Change to using new_ones/zeros
2024-02-07 00:21:21 +00:00
Brent Yi
6127ec0f72
Update viewer legacy docs (#2875) 2024-02-06 16:07:37 -08:00
Ethan Weber
af2072bf7c
Enable always returning depth for Gaussian Splatting (#2873)
enable always returning depth, if specified in the config
2024-02-06 10:48:20 +02:00
Congrong Xu
52cda21186
Add multi-processing support for cache_image. Fixed the issue of insufficient precision leading to a completely black image when converting from RGBA to RGB. (#2867) 2024-02-05 17:04:14 +00:00
Too Yuen Min
2046cfe9fb
fix: pixelated downscaled images (#2866) 2024-02-05 08:57:14 -08:00
Matthew Tancik
ffe3c31c13
v1.0.1 (#2865) 2024-02-01 17:00:02 -08:00
Brent Yi
6436742c34
Upper-bound torch version for dev installs only (#2864) 2024-02-02 00:19:54 +00:00
Piero Toffanin
c5a400c8c9
Fix ODM data processor coordinate system (#2847)
* Fix ODM data processor coordinate system
2024-02-01 10:34:19 -08:00
Simon Bethke
10c6f29451
Changed torch version to <2.2 (#2861)
* Changed torch version to <=2.2, as <=2.1 will uninstall torch 2.1.2+cu118 and leave users without cuda in torch

* changed <=2.2 to <2.2 to prevent the issue at to prevent the type error referenced here https://github.com/nerfstudio-project/nerfstudio/pull/2857

* Changed torch version to <=2.2, as <=2.1 will uninstall torch 2.1.2+cu118 and leave users without cuda in torch
2024-02-01 11:06:17 +00:00
Justin Kerr
0e01a90bba
Replace depth alpha mask logic with torch.where for gradients in splatfacto (#2856)
* replace alpha depth logic with torch.where for differentiability
2024-01-31 14:03:34 -08:00
J.Y
de36210ae5
Avoid OOM for caching undistorted images of splatfacto (#2855)
Co-authored-by: Justin Kerr <justin.g.kerr@gmail.com>
2024-01-31 13:41:20 -08:00
mattstrong-stanford
ebbebdab4c
Have predicted depth as expected_depth so gradients can be computed (#2852)
Update depth_nerfacto.py
2024-01-31 19:48:08 +00:00
Justin Kerr
17748a8309
limit torch to <=2.1 to avoid type errors (#2857) 2024-01-31 19:34:00 +00:00
Jonáš Kulhánek
3a90cb529f
Fix #2843 - safer typing (#2848) 2024-01-30 11:06:17 -08:00
Jonáš Kulhánek
4779f816d6
Add flag to disable apperance embedding (#2843) 2024-01-29 23:07:37 +00:00
Ethan Weber
c397588300
Mask flexibility for methods than need to sample in the masked region (#2647)
* minor changes to make masking more flexible

* formatting

* updating ignore mask

---------

Co-authored-by: Justin Kerr <justin.g.kerr@gmail.com>
2024-01-29 11:47:49 -08:00
J.Y
d8b517d6a5
Add average_init_density to improve robustness of nerfacto training (#2834) 2024-01-28 14:09:27 -08:00
Piero Toffanin
fb1fee1d8f
Add support for loading 3D points with ODM data processor (#2838)
* Add support for loading 3D points with OpenDroneMap data processor
2024-01-28 11:09:31 -08:00
psetinek
f9a5227828
External Method: seathru_nerf (#2829)
* add seathru_nerf as external method
2024-01-28 11:01:46 -08:00
Congrong Xu
3f23e6efb2
Fix the background color bug when exporting video with the "random" background option (#2831)
patch #2815
2024-01-27 11:27:49 -08:00
Brent Yi
1bc5b424b4
Update viewer docs (#2826)
* Update viewer docs

* Typo
2024-01-26 22:30:50 -08:00
Anthony Zhang
1aba4ea7a2
fixing colmap installation to use apt-get since gdown on the URL unable to download colmap (#2823)
fixing colmap installation to use apt-get since gdown on the URL was failing
2024-01-25 23:09:23 +00:00
Piotr Sokólski
67b1005784
Fix "mutable default" error in render.py (#2825)
This fixes 

```
ValueError: mutable default <class 'nerfstudio.data.scene_box.OrientedBox'> for field obb is not allowed: use default_factory
```
2024-01-25 15:04:14 -08:00
Justin Kerr
f2f4bf55b2
Add alpha rendering to splatfacto, fix depth bug (#2824)
* Add alpha rendering to splatfacto, output as accumulation
2024-01-25 12:51:16 -08:00
Brent Yi
3bade3a9dc
Remove viewer build badge (#2819) 2024-01-24 23:11:22 -08:00
Chung Min Kim
3db08019db
Use packaging for colmap version (#2814)
* Use packaging.version for colmap version

* Run ruff

* Change default_version to str type
2024-01-24 16:02:05 -08:00
Brent Yi
fb4f9f0060
Fix mask shape for Fisheye624 cameras (#2813) 2024-01-23 16:08:04 -08:00
hturki
365592036e
Have Mill19 dataset use official splits by default (#2745)
add splits to mill19
2024-01-23 23:51:58 +00:00
VasuAgrawal
7cc27da7ff
Make downloading more windows-friendly (#2810)
* Make downloading more windows-friendly

* Remove unnecessary f-strings
2024-01-23 15:37:05 -08:00
Ikko Eltociear Ashimine
0dfefbf922
Fix typo in parallel_datamanager.py (#2812)
occured -> occurred
2024-01-23 20:56:01 +02:00
Huang Binxiao
b669f51f02
Update splatfacto.py (#2804)
Shape error: The shape of batch['mask'] is [H, W, 1], not[H, W]
2024-01-23 10:16:22 -08:00
VasuAgrawal
5988c7ee7c
Add documentation on existing datasets (#2806) 2024-01-23 09:54:13 -08:00
J.Y
8952fb531e
important fix typo (#2800)
fix typo

Co-authored-by: Jianbo Ye <jianboye@amazon.com>
2024-01-19 21:15:00 -08:00
Brent Yi
19a21d7773
Bump version to 1.0.0 (#2799)
Update pyproject.toml
2024-01-19 18:15:02 -08:00
Congrong Xu
c83ed7dcaf
fix the bugs associated with blender datas (#2704)
* fix the bug when camera.distortion_params is None

* Handle background color override when using blender.

* fix bare except

* format

* Update background color override in Blender dataparser

* Add ability to download EyefulTower dataset

* wip before I copy linning's stuff in

* Generate per-resolution cameras.xml

* Generate transforms.json at download

* Fix a couple of quotes

* Use official EyefulTower splits for train and val

* Disable projectaria-tools on windows

* Fix extra imports

* Add a new nerfacto method tund for EyefulTower

* Split eyefultower download into a separate file

* Fix typo

* Add some fisheye support for eyeful data

* Reformatted imports to not be dumb

* Apparently this file was missed when formatting originally

* Added 1k resolution scenes

* revert method_configs.py to original values

* Also add 1k exrs

* Add option to modify bg color in gaussian splatting

* fix back the config, bg color should work now

* removed camera optimizer for gs to align with main

* Address feedback

* Revert changes to pyproject.toml, to be added in a later PR

* Oops, probably shouldn't have gotten rid of awscli ...

* adding support for bg color, tested and reformatted now

* formatted

* formatted

* changed glob variable name

* Refactor background color variable name

* prevent viser overriding

---------

Co-authored-by: Vasu Agrawal <vasuagrawal@meta.com>
Co-authored-by: Ethan Weber <ethanweber@berkeley.edu>
Co-authored-by: Brent Yi <yibrenth@gmail.com>
Co-authored-by: Justin Kerr <justin.g.kerr@gmail.com>
2024-01-20 02:10:06 +00:00
Brent Yi
ff4002d343
Clean up ns-train {method} --help for not-yet-installed external methods (#2760)
* Clean up `ns-train {method} --help` for not-yet-installed external methods

* Ruff

* Ruff

* add clearer print statement

* Types? Not sure if this fixes it

---------

Co-authored-by: Justin Kerr <justin.g.kerr@gmail.com>
2024-01-19 18:02:56 -08:00
Brent Yi
70d83d4201
Remove legacy viewer from GitHub actions (#2798)
Co-authored-by: Justin Kerr <justin.g.kerr@gmail.com>
2024-01-19 16:42:50 -08:00
Justin Kerr
616846b10e
Gracefuly handle old ViewerElements from legacy viewer (#2797)
* fix jittering in markdown in viewer beta

* Revert "fix jittering in markdown in viewer beta"

This reverts commit 70ade42e46.

* print correctly formatted url in banner for viewer beta

* warn when legacy viewerelements are used in new viewer

* Add shim for viewer_elements namespace, fixes https://github.com/ayaanzhaque/instruct-nerf2nerf/pull/88

* Fix typo

* suppress ruff error

---------

Co-authored-by: Brent Yi <yibrenth@gmail.com>
2024-01-20 00:37:40 +00:00
VasuAgrawal
248e99e7f4
Add preliminary EyefulTower dataset support (#2672)
* Add ability to download EyefulTower dataset

* wip before I copy linning's stuff in

* Generate per-resolution cameras.xml

* Generate transforms.json at download

* Fix a couple of quotes

* Use official EyefulTower splits for train and val

* Disable projectaria-tools on windows

* Fix extra imports

* Add a new nerfacto method tund for EyefulTower

* Split eyefultower download into a separate file

* Add some fisheye support for eyeful data

* Reformatted imports to not be dumb

* Apparently this file was missed when formatting originally

* Added 1k resolution scenes

* revert method_configs.py to original values

* Also add 1k exrs

* Address feedback

* Revert changes to pyproject.toml, to be added in a later PR

* Oops, probably shouldn't have gotten rid of awscli ...

* changed glob variable name

* Update tensor_dataclass.py

---------

Co-authored-by: Ethan Weber <ethanweber@berkeley.edu>
Co-authored-by: Brent Yi <yibrenth@gmail.com>
2024-01-20 00:22:49 +00:00
Justin Kerr
378dfc1661
Rename gaussian-splatting to splatfacto for clarity (#2796)
* method name->splatfacto

* change file names to splatfacto
2024-01-19 15:21:24 -08:00
VasuAgrawal
25d723c08b
Disable projectaria tools on Windows (#2794)
* Update pyproject.toml to work for windows again

* Actually pycolmap works now
2024-01-19 15:04:33 -08:00
Brent Yi
6389cacd19
Gaussian splatting support for Aria (#2785)
* Gaussian splatting support for Aria

* Respect masks in splatting loss function
2024-01-19 11:09:37 -08:00
Brent Yi
c27b5cb65f
Use github output format for ruff in CI (#2786)
* Use github output format for ruff in CI

* Fix

---------

Co-authored-by: Justin Kerr <justin.g.kerr@gmail.com>
2024-01-19 09:29:50 +00:00
Brent Yi
99073eb187
Don't throw away non-tensor objects in TensorDataclass (#2783)
* Don't throw away non-tensor objects in TensorDataclass

* Update test_tensor_dataclass.py

---------

Co-authored-by: Justin Kerr <justin.g.kerr@gmail.com>
2024-01-19 01:18:19 -08:00
Brent Yi
f55bd04ca6 Hotfix non-colmap nerfstudio datasets 2024-01-19 00:39:53 -08:00
Riley
288e1496f1
Switching the gaussian-splatting dataparser from ColmapDataParser to NerfStudioDataparser and updating ns-process-data (#2791)
* ns-process-data updated to output a .ply of sparse points from colmap

* docs for loading 3D points

* format fixing

* ruff formatting

* added read_points3D_text

* formatting

* added 3D points to test

* prompted user bool fix

* small fixes to pass test cases

* fix colors

* fixed coord system

* Tweak frames

* Comment wording

* ruff

* Comment intentional default change for load_3D_points

* Wording

* More wording

* Fix double/float issue

---------

Co-authored-by: Ethan Weber <ethanweber@berkeley.edu>
Co-authored-by: Brent Yi <yibrenth@gmail.com>
2024-01-19 08:00:45 +00:00
J.Y
25a00a09ab
Revisiting #1504 and fix colmap world coordinate transform (#2793)
fix colmap world coordinate transform
2024-01-19 05:08:19 +00:00
Brent Yi
46847ab56b
Bump tyro (#2792) 2024-01-18 20:19:28 -08:00