Commit Graph
1793 Commits
Author SHA1 Message Date
Brent Yi 3b6ca91dbc Viewer depth compositing for Gaussian Splatting models (#2744)
* Viewer depth compositing for Gaussian Splatting models

* Tune
2024-01-11 04:04:00 +00:00
Matthew Hoffman e7e713e900 Fix Python 3.11 dataclass field default_factory defaults (#2746) 2024-01-10 17:12:54 -08:00
3ec10f8ea7 Undistortion should not occur for already-undistorted images (#2726)
* undistortion should not occur if pinhole (smoothes the images), and accordingly fix rounding of downscaled image size

* rounding downscaling, more consistent with other resizing methods in public datasets

* reverting to cpu image caching

* re-allow camera unimplemented

* compatibility with mask/depth

* reverted scale factor rounding, corrected mask undistort condition

* moved distortion_params update inside PERSPECTIVE type if-clause

* bumping opencv to 4.8, to deal with undistortion bug with distortion parameters of zero

---------

Co-authored-by: liorf <liorf@amazon.com>
Co-authored-by: alshan <alshan@amazon.com>
Co-authored-by: Justin Kerr <justin.g.kerr@gmail.com>
2024-01-10 12:44:57 -08:00
hturki 4e8c65cfbf Add Mill 19 dataset (#2742)
* add Mill 19 dataset

* Address PR feedback
2024-01-10 16:08:12 +00:00
hturki 7edd768240 Add PyNeRF external method (#2734)
* add pynerf method

* fixes

* fixes

* format

* rename k-planes references
2024-01-10 00:58:02 +00:00
Matthew TancikandJustin Kerr 451a70c1f3 Migrate to viser viewer (#2515)
* fix jittering in markdown in viewer beta

* Revert "fix jittering in markdown in viewer beta"

This reverts commit 70ade42e46.

* Migrate to viser viewer

* Fix docs

* print correctly formatted url in banner for viewer beta

* add deprecation warnings to viewer_elements.py

* fix export string scientific notation bug, change .5 to 2.0 sec transition time

---------

Co-authored-by: Justin Kerr <justin.g.kerr@gmail.com>
2024-01-08 11:40:02 -08:00
Alexander KristoffersenandJustin Kerr ca9e6c1294 Fixed nerfstudio logo at the top of Colab notebook (#2733)
Fixed logo at the top of notebook

Co-authored-by: Justin Kerr <justin.g.kerr@gmail.com>
2024-01-08 18:39:41 +00:00
581ba95eab fixed not cache images for full image datamanager (#2692)
* rm no-cache option

---------

Co-authored-by: Matias Turkulainen <30566358+maturk@users.noreply.github.com>
Co-authored-by: maturk <matiasturkulainen@gmail.com>
Co-authored-by: Justin Kerr <justin.g.kerr@gmail.com>
2024-01-08 10:19:38 -08:00
J.YandMatthew Tancik 638180c90a Improvement to Gaussian Splatting workflow (#2709)
* Necessary changes to establish baseline as reported by papers

- Port multiple train/eval split strategies from nerfstudio parser for colmap parser

- A number of fixes to establish solid baseline

- Avoid empty frames where no gaussians are projected onto the image

- Fix random order of image filenames

* a few enhancement changes to bridge gap between Inria and NS

* Fix PLY exporter and implement an PLY renderer

* Add finite filter in ply exporter

* Major changes to improve nerfstudio-gs training

* separate sh_0 and sh_rest and use different learning rates, and remove sh_rest regularization.
* disable downscale in warm-up stages
* prune the original gaussians that splits.
* change color clamp_max mechanism to be applied per ray instead of per gaussian

* Enable exporter and gsplat_render to work with color only PLY

* Prepare PR with minor changes

* Fix pyright errors

* remove gsplat ply renderer

* Address github/pwais comments

* fix an evaluation metric bug

* Use 0.1 cull_alpha_threshold for fast training

* Add scale regularization as an option (off by default)

* add continue_cull_post_densification option and some refactor

---------

Co-authored-by: Matthew Tancik <matt@lumalabs.ai>
2024-01-08 06:11:03 +00:00
Wei DingandAlexander Kristoffersen 1e23781393 Add orthophoto rendering support. (#2648)
* add orthophoto CameraType.

* test case of orthophoto camera.

* Fix the rotation errors caused by the usage of left-handed system

* Optimized ortho-cam,  it can be used with other camera models now.

* fix TypeError of coords in `test_multi_camera_type()`

---------

Co-authored-by: Alexander Kristoffersen <akristoffersen@berkeley.edu>
2024-01-07 18:17:48 -08:00
Chung Min Kim cc61cafea4 Add visible as kwarg to ViewerElements (#2730)
* Add `visible` as kwarg to ViewerElements

* Formatting
2024-01-06 15:43:26 -08:00
Matias TurkulainenandJustin Kerr 7311ff69f6 allow render path to exist (#2710)
Co-authored-by: Justin Kerr <justin.g.kerr@gmail.com>
2024-01-05 15:47:13 -08:00
Ruilong Li(李瑞龙) 98a21263c1 Fix periodically stuck in parallel datamanager (#2713)
* Fix sticking in parallel datamanager
2024-01-05 15:21:50 -08:00
Cyrus Vachha a8e6f8fa3f Adding docs for Instruct-GS2GS (#2702)
* Adding docs for Instruct-GS2GS

Adding documentation for Instruct-GS2GS including its page and adding it to the list of external and 3rd party methods.

* fixing blank formatting dev checks

* updating subtitle

Updating the subtitle to specify gaussian splatting
2023-12-27 21:19:19 -08:00
Abhik Ahuja 38fc72729c Fix dycheck depth and rgb rescale (#2623)
Fix depth and rgb rescale

Not sure if depth is correct since I'm not sure what the 3rd dim is for, the code runs properly though.
2023-12-27 01:07:08 +00:00
Brent Yi 7a8e31d84e Improved render panel for beta viewer (#2667)
* Continuous velocities for render splines

* Configure ruff line length

* Improved render preview

* No longer need to update camera FOV in render preview

* Details

* Polishing

* Cleanup

* Details

* Address pyright errors

* type: ignore for broken (?) Aria stub
2023-12-20 17:20:41 -08:00
Hardik Dava 4a54763323 fixed color issue in ply file (#2693)
fixed color export
2023-12-19 09:20:17 -08:00
Zhuoyang Pan f7a49563a0 Fix a bug of random quaternions (#2687)
fix a bug for random quaternions
2023-12-18 17:52:14 -08:00
Otto SeiskariandMatias Turkulainen 626441e15e Add instructions for data collection and processing with Spectacular AI tools (#2668)
* Add instructions for using Spectacular AI tools

Adds easy support for new device types: OAK-D, RealSense and
Azure Kinect, as well as an alternative pipeline for iOS.

* Grammar fixes in docs/quickstart/custom_dataset.md (Spectacular AI section)

Co-authored-by: Matias Turkulainen <30566358+maturk@users.noreply.github.com>

* Fix section label in the Spectacular AI section of custom_dataset.md

* Update custom_dataset.md: typo fix.

"montions" -> "motions"

---------

Co-authored-by: Matias Turkulainen <30566358+maturk@users.noreply.github.com>
2023-12-15 18:37:27 +02:00
+5 ddedb920dc Gaussian splatting (#2521)
* Adds Gaussian Splatting integration
---------

Co-authored-by: Jake Austin jake-austin@berkeley.edu <jake-austin@berkeley.edu>
Co-authored-by: Brent Yi <yibrenth@gmail.com>
Co-authored-by: Zhuoyang <zhuoyangpan@outlook.com>
Co-authored-by: Vickie Ye <vye1116@gmail.com>
Co-authored-by: adamrashid96 <abrashid@berkeley.edu>
Co-authored-by: Paul Wais <paulwais@gmail.com>
Co-authored-by: Gina Wu <ginawu98@gmail.com>
Co-authored-by: Alexander Kristoffersen <akristoffersen@berkeley.edu>
Co-authored-by: machenmusik <machenmusik@users.noreply.github.com>
Co-authored-by: Riley <cardiacmangoes@gmail.com>
Co-authored-by: Gina Wu <42229107+ginazhouhuiwu@users.noreply.github.com>
Co-authored-by: maturk <matiasturkulainen@gmail.com>
Co-authored-by: Guangyun-Xu <38279883+Guangyun-Xu@users.noreply.github.com>
Co-authored-by: Zhuoyang Pan <102644383+Zhuoyang-Pan@users.noreply.github.com>
2023-12-14 22:14:09 -08:00
Rohan Mathur 88873ef3ab Added support for .tif and .tiff files in equirect utils (#2613)
* added tiff to equirect utils

* added tif files
2023-12-15 01:22:20 +00:00
Tobias FischerandMatias Turkulainen 622e26b511 [Fix] Pixel Sampler setup in VanillaDataManager for Cameras incl. metadata (#2678)
fix pixel sampler setup for cameras with metadata other than fisheye crop radius

Co-authored-by: Matias Turkulainen <30566358+maturk@users.noreply.github.com>
2023-12-14 10:07:41 -08:00
Boris Feld a894c6d938 Update first_nerf.md (#2677)
Update documentation to mention that Comet visualizer can be used without the viwer
2023-12-14 08:47:51 -08:00
Justin KerrandBrent Yi 17f785758b Parity of layout between viewer beta and legacy viewer (#2639)
* alter layout of beta viewer to match old viewer

---------

Co-authored-by: Brent Yi <yibrenth@gmail.com>
2023-12-13 11:17:36 -08:00
Brent Yi ae00061cf5 Fix custom data links in README (#2671) 2023-12-12 22:31:12 +00:00
Chris SweeneyandBrent Yi 6ccbc2de45 Add support for Project Aria datasets. (#2617)
* Add support for Project Aria datasets.

1) An export script for processing Project Aria datasets
2) Add support for Fisheye624 cameras
3) Add support for center-crop masking and sampling

Example commands for running nerfstudio on Project Aria data:

python scripts/datasets/process_project_aria.py --vrs_file=<path_to_vrs> --mps_data_dir=<path to MPS directory> --output_dir=<nerfstudio_data_dir>

ns-train nerfacto --data <nerfstudio_data_dir>  nerfstudio-data --orientation-method none

* Clean up type errors

* Fix formatting

* Fix linter errors, import, and use absolute file paths

* Add aria subcommand to ns-process-data

* Appease ruff

* Add Aria to docs

* Fix docs, add back import try/except

* nit: path.as_posix() -> str(path)

---------

Co-authored-by: Brent Yi <yibrenth@gmail.com>
2023-12-12 22:18:13 +00:00
Brent YiandGina Wu 01d57b90dc Fix large dataset (> 512 images) training in viewer_beta (#2669)
* Fix large dataset (> 512 images) training in viewer_beta (#2586)

* fix index bug in update_camera_poses of viewer

* Set Ruff line lenth

---------

Co-authored-by: Gina Wu <42229107+ginazhouhuiwu@users.noreply.github.com>
2023-12-12 11:44:26 +00:00
Mohit Motwani 281f81f1ec Update cameras.py - removed duplicate lines (#2661)
The self.times was being computed twice in cameras.py in the same function which was unnecessary
2023-12-12 11:13:50 +00:00
Brent Yi 73fc3dcd63 Update dependencies for M1 Macs (#2665)
* Update dependencies for M1 Macs

* import newrawpy as rawpy

* Comment

* Add back rawpy for Linux

* Run black
2023-12-11 18:08:17 +00:00
Paul WaisandBrent Yi 390b63373d Trainer can now always save dataparser transform (#2653)
* Trainer can now always save dataparser transform

* appease the linter

---------

Co-authored-by: Brent Yi <yibrenth@gmail.com>
2023-12-11 13:55:36 +00:00
Alexandru Kisandmfischer 1d9cc7dd8a Update hloc and add new matcher types (#2658)
* Update hloc and add new matcher types

* Update dockerfile comment to better reflect what the command is doing

---------

Co-authored-by: mfischer <michael.fischer@spleenlab.ai>
2023-12-08 08:58:51 -08:00
omahs 4c966b9bd4 Fix typos (#2655)
* fix typo

* fix typo

* fix typos

* fix typo
2023-12-08 11:27:05 +00:00
AdamRashid96 64f0b2547b Ns-process-data bug with dev version of colmap (#2651)
* only parse digit and decimal

* formatting
2023-12-06 22:56:39 -08:00
Ethan WeberandBrent Yi 4c627edbac Encoder + MLP combo (#2063)
* adding encoder + mlp combo

* update documentation

* minor fixes

* fixed issue when tcnn isn't installed

---------

Co-authored-by: Brent Yi <yibrenth@gmail.com>
2023-12-01 03:16:42 +00:00
0cb410001b Fixes bug in PairPixelSampler when working with masked dataset (#2368)
* Fixes bug in PairPixelSampler when working with masked dataset

* Formatting

* Black formatting now ok.

---------

Co-authored-by: AdamRashid96 <71362382+AdamRashid96@users.noreply.github.com>
Co-authored-by: Brent Yi <yibrenth@gmail.com>
2023-12-01 01:11:11 +00:00
JoseandBrent Yi 49503c2175 Mps fix (#2436)
* mps bugfix

* removed docs change

* Double quotes

---------

Co-authored-by: Brent Yi <yibrenth@gmail.com>
2023-12-01 00:04:01 +00:00
Jaggz H 43a7da36a5 Converted many @dataclass assignments for python 3.11 compatibility (#2630)
Converted many @dataclass assignments to field()s for python 3.11 compatibility
2023-11-30 23:51:36 +00:00
blacksino 747dd557f6 fix include_input for NeRFEncoding (#2642) 2023-11-30 23:07:26 +00:00
David Holtz 3d8af87290 Make nuScenes dataparser compatible with ParallelDataManager (#2635)
The ParallelDataManager (see #2092) makes pytorch crash if the cameras instance's fx, fy, cx or cy tensors
are loaded from a common shared tensor.

This PR fixes the issue by cloning the respective tensors before passing them to the Cameras(...) constructor.
2023-11-30 15:00:31 -08:00
5a772b177d Bugfix: shape confusion in resizing (#2565)
Co-authored-by: Justin Kerr <justin.g.kerr@gmail.com>
Co-authored-by: Brent Yi <yibrenth@gmail.com>
2023-11-30 05:57:43 +00:00
Justin KerrandBrent Yi 8b85c44b27 Patch spiral rendering w/ parallel datamanager (#2637)
* 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

* allow parallel datamanager in spiral rendering

* lint

* Update render.py

---------

Co-authored-by: Brent Yi <yibrenth@gmail.com>
2023-11-30 03:42:42 +00:00
Brent Yi c8c55521e8 Keyframe timing for beta viewer (#2638)
* Add keyframe timing to beta viewer

* Nits

* New keyframe timing logic

* Bump viser version
2023-11-30 00:52:37 +00:00
Chung Min KimandJustin Kerr 5865ecf323 Integrate scene click capability from original viewer (#2509)
* Integrate scene click capability from original viewer

* filter non-click ScenePointerEvents from ClickEvent

* lint

* Put click origin+direction in nerfstudio world scale

* Update docs

* Update doc heading

---------

Co-authored-by: Justin Kerr <justin.g.kerr@gmail.com>
2023-11-29 14:48:58 -08:00
Jonáš Kulhánek 38726da219 Fix tangential camera distortion (p1,p2 in opencv camera) (#2627) 2023-11-29 19:48:41 +00:00
Chung Min Kim c896ee4e0a Remove PCA colormap bug for float output maps (#2620)
Previously, "PCA" would be offered as a colormap option for float outputs (and throws "PCA is not a known colormap" error).
2023-11-26 22:11:05 -08:00
Chung Min Kim 43703fa05a Bugfix; scrape pipeline for viewercontrol, not trainer (#2621)
Scrape pipeline for viewercontrol, not trainer
2023-11-26 21:56:49 -08:00
Rohan MathurandJustin Kerr 9528a3f717 Added UI to load camera path from existing path for the dataset (#2530)
* added load camera path

* bugfixes

* fixed fov value in load camera path

---------

Co-authored-by: Justin Kerr <justin.g.kerr@gmail.com>
2023-11-16 04:09:48 +00:00
emma be37bcddea Remove an outdated comment in RayGenerator class (#2606)
clean up comment
2023-11-14 08:21:03 +00:00
Chris Heinrich b8736f704f Update LICENSE (#2597)
Filling in the license to what I think it should be. We're adding this to our list of third party libraries
2023-11-14 08:14:17 +00:00
Brent Yi 1f5f0c792b Make spherical harmonics test more forgiving (#2607) 2023-11-14 00:00:21 -08:00