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

1794 Commits

Author SHA1 Message Date
Brent Yi
55b7a2c8ef
Update docs to recommend torch 2.1 (#2748) 2024-01-11 15:12:41 -08:00
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
liorfritz
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 Tancik
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 Kristoffersen
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
Hardik Dava
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.Y
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 Ding
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 Turkulainen
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 Seiskari
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
Justin Kerr
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 Fischer
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 Kerr
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 Sweeney
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 Yi
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 Wais
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 Kis
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 Weber
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
Panteleris Paschalis
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
Jose
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
Yosshi999
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 Kerr
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 Kim
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 Mathur
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