Commit Graph
2072 Commits
Author SHA1 Message Date
Gina Wu 02a3f7a806 ruff format import blocks 2024-11-06 16:05:42 -08:00
Gina Wu f51f355f79 error message and load checkpoint fixes 2024-11-06 15:56:07 -08:00
Gina Wu 2dc15df7ea Merge branch 'main' into gina/export 2024-11-05 15:25:41 -08:00
Gina Wu c0e57ab10a Merge branch 'gina/export' of https://github.com/nerfstudio-project/nerfstudio into gina/export 2024-11-05 15:24:45 -08:00
Gina Wu b838598fc9 nit 2024-11-05 15:23:24 -08:00
SebastiaanandJ.Y. cef95890f3 Restrict pymeshlab version on Windows (3461) (#3474)
Co-authored-by: J.Y. <132313008+jb-ye@users.noreply.github.com>
2024-10-24 19:12:25 +00:00
Justin Kerr fc4fc5cb15 Properly reset train state after render lock in viewer (#3486)
* properly reset train state after render locK

* finally block
2024-10-18 11:37:58 -07:00
Justin Kerr d2f2ad9474 update gsplat version to >=1.3.0 (#3478)
* update gsplat version

* 1.4.0
2024-10-16 12:32:06 -07:00
Simon Bethke adc743c3a0 tag exported ply files (#3472)
* tag exported ply files

* Added the version

* fix import order

* Added information about the up-axis
2024-10-13 18:48:52 -07:00
Brent Yi 67be06009e Add option to change .ply output filename for Gaussian exports (#3467)
Add option to configure .ply output filename for Gaussian exports
2024-10-09 22:22:52 +00:00
Nicholas Pfaff 8e27e5f54d Add option for refining the TSDF using the mesh's AABB (#3459) 2024-10-06 21:23:16 -07:00
Brent Yi 1bc1e1aaf6 Add max_frames + max_output_size arguments for Aria data processing (#3466)
* Add max_frames + max_image_size arguments for Aria data processing

* Change max_output_size
2024-10-06 21:22:16 -07:00
Nicholas PfaffandJ.Y. 866a069777 Keep depth_image on CPU (#3449)
Co-authored-by: J.Y. <132313008+jb-ye@users.noreply.github.com>
2024-10-01 19:39:26 +00:00
Nicholas Pfaff 334a1c8afe Make PixelSampler the depth-nerfacto default (#3454)
* Make PixelSampler the depth-nerfacto default

* Add note about PairPixelSampler and DepthLossType.SPARSENERF_RANKING
2024-10-01 16:59:31 +00:00
Nicholas Pfaff 56556d22fa Rejection sample mask for patch pixel sampler (#3457)
Use mask rejection sampling in PatchPixelSampler
2024-10-01 09:54:28 -07:00
Nicholas Pfaff 4c7297c75a Speed up PairPixelSampler (#3452) 2024-09-30 17:31:49 -07:00
Nicholas Pfaff 6da94443fd Change depth from float64 to float32 (#3453) 2024-09-28 16:51:05 -07:00
J.Y. 01d3f73e2c Fix #3417 (#3441) 2024-09-24 18:01:43 -07:00
Aayush GuptaandJ.Y. e6941aca47 Zipnerf point cloud and tsdf mesh export support (#3429)
Co-authored-by: J.Y. <132313008+jb-ye@users.noreply.github.com>
2024-09-24 15:28:18 -04:00
Aayush Gupta 516fd7c9ac Fix splatfacto crash in eval when using viewer (#3430) 2024-09-17 12:31:42 -07:00
3e638da635 Adding Random Seed for Frame Processing (#3416)
* Adding Random Seed for Frame Processing

* Added Unit Tests

* Updating Unit Tests for Ffmpeg

* Make Logs More Detailed

---------

Co-authored-by: Anthony-Tafoya <anthonytafoya@berkeley.edu>
Co-authored-by: J.Y. <132313008+jb-ye@users.noreply.github.com>
2024-09-17 15:32:09 +00:00
Nicholas Pfaff 27b8e14798 Fix crop obb masking in pointcloud generation (#3433) 2024-09-16 16:44:31 -04:00
Maximum Wilder-Smith 106d427e80 Minor update to generate splat files in inference mode (#3432)
changed gaussian splat export to inference mode
2024-09-16 13:57:21 -04:00
7a0b8cbc89 use gsplat strategy interface to simplify splatfacto (#3376)
* use gsplat strategy interface to simplify splatfacto

* change splatfacto-big densify_grad_thresh to 0.0005

* upgrade gsplat version

* filter low opacities when exporting

* remove num_sh_bases

---------

Co-authored-by: Jianbo Ye <jianboye@amazon.com>
Co-authored-by: Justin Kerr <justin.g.kerr@gmail.com>
2024-09-13 07:10:37 -04:00
Jonáš Kulhánek 194b5d4e46 Fix docker attestation fail on forked repo PRs (#3423) 2024-09-12 09:43:36 -04:00
Aayush Gupta e7b7dc9c6f Add option to start training paused (#3420) 2024-09-10 19:56:44 +00:00
Daniel Crispell e2a0915e56 bugfix: handle multiple camera types in a single batch (#3415) 2024-09-10 15:31:05 -04:00
Hardik Dava d67b281f7d RGBA renderer for splatfacto (#3307)
* Update base_model.py

* Update render.py

* Update render.py

* fixing error

* Update render.py

* Update render.py
2024-09-09 16:20:26 -04:00
Justin Kerr c976bf688f Merge branch 'main' into gina/export 2024-09-05 11:10:46 -07:00
f86dbe6d45 Update dockerfile and add docker build action (#3283)
* Use github docker registry

* Fix DDP train for GPU in exclusive mode

* Improve docker image - compile gsplat, decrease image size

* Drop unrelated change

* Add build docker image action

* Rename build docker image action

* nit

* Remove commented line from Dockerfile

* Fix dockerfile when explicit source is specified

* Fix failing dynamo build for torch.compile

* Lock dockerfile and tcnn versions

* Add `torch.cuda.is_available()` condition

* Drop set_cuda_device

* Fix build docker image github action

* Docker build save disk space

* Set MAX_JOBS to limit resource usage for docker build

* Try bumping `MAX_JOBS` 2 => 4

* Install fixed gsplat version from nerfstudio's pyproject.toml

* Update docs

* Finish docker build action

* Fix ignores push when PR

---------

Co-authored-by: Gina Wu <ginawu98@gmail.com>
Co-authored-by: Brent Yi <yibrenth@gmail.com>
2024-09-05 10:12:21 +02:00
Hoanh Le 96b7fe2230 update documentation for instant-ngp (#3406)
update documentation
2024-09-03 10:09:18 -04:00
a6214921f8 Add eval data to image_rename_map (#3396)
* Add eval data to image_rename_map

* Follow name convention

---------

Co-authored-by: Brent Yi <yibrenth@gmail.com>
Co-authored-by: J.Y. <132313008+jb-ye@users.noreply.github.com>
2024-09-03 01:35:54 +00:00
Hoanh Le d395aa2e53 change grid resolution type (#3400) 2024-08-30 17:32:42 +03:00
Gina Wu 61974c3ece Merge branch 'main' into gina/export 2024-08-29 18:11:43 -07:00
Gina WuandBrent Yi dff42c9a81 Bump viser to 0.2.7 (#3398)
Bump viser 0.2.7

Co-authored-by: Brent Yi <yibrenth@gmail.com>
2024-08-29 17:53:34 -07:00
Brent Yi 33dce833ee Suppress wandb type error (#3397) 2024-08-29 17:21:45 -07:00
Gina Wu 5f4ba8dd10 ruff again 2024-08-29 16:13:28 -07:00
Gina Wu 6b4afed0a9 ruff format exports 2024-08-29 16:08:35 -07:00
Gina Wu 8aa49c3699 ruff format 2024-08-29 16:05:21 -07:00
Gina Wu 58a4c3ad05 pyright 2024-08-29 15:30:30 -07:00
Gina Wu 2308450d39 pyright 2024-08-29 11:50:15 -07:00
Gina Wu e1ac0e56cd some pyright type fixes 2024-08-28 14:45:39 -07:00
J.Y.andJianbo Ye 6410080244 bump version to v1.1.4 (#3394)
bump version

Co-authored-by: Jianbo Ye <jianboye@amazon.com>
v1.1.4
2024-08-28 17:06:59 -04:00
96ca8b0ab7 Update exporter.py to export sh_degree 0 case #3371 (#3374)
* Update exporter.py for sh_degree 0

Change to write sh coefficients instead of color values

* Add flag for use_sh0_renderer

Add sh0 renderer case for model.config.sh_degree == 0

* fix ruff

* add warning if use_sh0_renderer is used when higher order of SH is available

* fix rgb export for color-only training

* use ply_color_mode

* better handling ply_color_mode=='rgb' when sh_degree>0

* clean RGB2SH

* fix issues

* update description

---------

Co-authored-by: bell-one <bellone1991@gmail.com>
Co-authored-by: Jianbo Ye <jianboye@amazon.com>
Co-authored-by: Brent Yi <yibrenth@gmail.com>
2024-08-28 19:41:25 +00:00
Hoanh Le 22dae34eeb add assertion to make sure positions and positions_flat are not empty (#3387) 2024-08-28 12:37:54 -07:00
Gina Wu 12c454a074 RUFF FORMAT 2024-08-26 20:43:45 -07:00
Gina Wu 2ff7da4888 ruff format 2024-08-26 20:40:41 -07:00
Gina Wu f5f9fdb31b ruff format 2024-08-26 20:33:45 -07:00
Gina Wu 24235109e6 ruff format 2024-08-26 20:33:38 -07:00
Gina Wu 15a1fbbe5a Merge branch 'main' into gina/export 2024-08-26 20:26:52 -07:00