4
0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 02:24:38 +08:00
nerfstudio/nerfstudio
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
..
cameras Add support for Project Aria datasets. (#2617) 2023-12-12 22:18:13 +00:00
configs Converted many @dataclass assignments for python 3.11 compatibility (#2630) 2023-11-30 23:51:36 +00:00
data Add support for Project Aria datasets. (#2617) 2023-12-12 22:18:13 +00:00
engine Trainer can now always save dataparser transform (#2653) 2023-12-11 13:55:36 +00:00
exporter
field_components Encoder + MLP combo (#2063) 2023-12-01 03:16:42 +00:00
fields Encoder + MLP combo (#2063) 2023-12-01 03:16:42 +00:00
generative
model_components Remove an outdated comment in RayGenerator class (#2606) 2023-11-14 08:21:03 +00:00
models Converted many @dataclass assignments for python 3.11 compatibility (#2630) 2023-11-30 23:51:36 +00:00
pipelines Converted many @dataclass assignments for python 3.11 compatibility (#2630) 2023-11-30 23:51:36 +00:00
plugins
process_data Update dependencies for M1 Macs (#2665) 2023-12-11 18:08:17 +00:00
scripts Add support for Project Aria datasets. (#2617) 2023-12-12 22:18:13 +00:00
utils
viewer Update URLs for readme and image references (#2527) 2023-10-13 22:23:20 -07:00
viewer_beta Fix large dataset (> 512 images) training in viewer_beta (#2669) 2023-12-12 11:44:26 +00:00
__init__.py
py.typed