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

1110 Commits

Author SHA1 Message Date
Jake Austin
2af29d00a5 Merge branch 'perf' of github.com:nerfstudio-project/nerfstudio into main 2022-11-30 02:02:04 +00:00
Ethan Weber
7346f97590
Quick fix to viewer OOM issue (#1053)
quick fix to viewer
2022-11-29 13:53:16 -08:00
Matthew Tancik
7280541350
Add per frame polycam intrinsics (#1050)
Add per frame polycam intrinsincs
2022-11-28 22:24:35 -08:00
Matthew Tancik
a1f20b3ed8
Add per frame intrinsic support to nerfstudio data (#1049) 2022-11-28 22:13:55 -08:00
Kevin Chen
87c3ead680
Add Support for Equirectangular Images (#966) 2022-11-28 21:29:02 -08:00
Julian Knodt
1d10e047c2
add render time in viewer (#1007)
* Add temporary render for viewer

* Add slider for render time
2022-11-28 17:17:54 -08:00
Matthew Tancik
666b43a577
Check is metashape xml is present (#1044) 2022-11-28 16:39:09 -08:00
Matthew Tancik
6c805b6d33
v0.1.11 (#1043) 2022-11-28 15:16:54 -08:00
Matthew Tancik
bce963facc
Allow users to skip using openrelay (#1042) 2022-11-28 14:55:43 -08:00
Frédéric Devernay
8fc7950719
Fix benchmarking scripts (#1005)
* launch_eval_blender.sh: fix script

- add shebang
- add -s option to launch a single job per GPU
- last GPU was ignored
- kill all subprocesses when script is terminated

* launch_train_blender.sh: fix script

- add -s option to launch a single job per GPU
- add -v option to use tensorboard instead of wandb
- set nerfacto options according to https://github.com/nerfstudio-project/nerfstudio/issues/806#issuecomment-1284327844
- use a single timestamp for all training jobs
- last GPU was ignored
- kill all subprocesses when script is terminated
- print the eval script command-line

* launch_eval_blender.sh: fix script

* Update benchmarking.md

* launch_train_blender.sh: add -s to eval command-line

* Update launch_train_blender.sh

* Update benchmarking.md
2022-11-28 13:25:14 -08:00
cnsumner
3d70b4e614
Add support for insta360 single-file format (#1024)
* Add support for insta360 single-file format

* Improve some variable names

* Add documentation for crop_percentage parameter

Co-authored-by: Matthew Tancik <tancik@berkeley.edu>
2022-11-28 13:00:10 -08:00
Nicolas Zunhammer
bed7ef94c5
Add Docker support (#1022)
* Added Dockerfile to build docker image including all dependencies and nerfstudio, added description on how to user docker.

* Move docker description to Installation.md.

Co-authored-by: Nicolas Zunhammer <n.zunhammer@tum.de>
2022-11-28 12:58:59 -08:00
Matthew Tancik
0a57c7c72b
Ignore non-image files in ns-process-data (#1040)
* Ignore non-image files in ns-process-data

* Update reqs

* bump ipython
2022-11-28 10:38:37 -08:00
David McAllister
a578ac8d15
Camera Label Hiding (#1031)
* ThreeJS group traversal

* Isolated hiding to camera labels

* Logic for hide scene addeD
2022-11-27 22:46:10 -08:00
Matthew Tancik
fff1fee686
Cleanup train cli logs (#1037) 2022-11-27 21:51:43 -08:00
David McAllister
f739786d3a
Pinched Looping Camera Path Fix (#1029)
Pinched looping path fix
2022-11-27 20:05:28 -08:00
Zak Murez
a7df91edc8
config nerfacto (#1020)
Co-authored-by: zmurez <zmurez@eng.ucsd.edu>
2022-11-27 15:56:06 -08:00
Jake Austin
d394950e76 Finished the performance optimizer 2022-11-27 06:29:51 +00:00
Alexander Kristoffersen
aac7aa4786
Typo in ProcessRecord3D error (#1027)
small error fix.
2022-11-26 13:16:27 -08:00
Ethan Weber
ee03a8f2c6
Switch colab pip install to git 2022-11-26 10:44:27 -08:00
Ethan Weber
b23d76cc47
Cameras TensorDataclass and Phototourism (#961)
This branch gives us the flexibility to use Cameras as a tensordataclass, giving arbitrary batch dimensions, slicing, etc all for free. It tests this with the phototourism dataset.

Tensordataclass changes:
Originally, the tensordataclass would look at tensors in an object and take everything in front of the last dimension to be the batch dimensions, but in the case of the camera_to_world matrices, everything except for the last TWO are the important dimensions, so we needed to have a way for the tensordataclass to acknowledge these specific fields where there are multiple data dimensions behind the batch dimensions. This is important for things like reshaping where we need to know which dimensions get reshaped and which dimensions get left alone.

Cameras changes:
Now subclass of tensordataclass, which required a change to the tensordataclass itself.
The possibility of extra batch dimensions is the hairy part that requires the majority of these changes. The extra batch dimensions aren't a hassle in every function except in ray generation. The ray generation is where this operates different relative to the old cameras. The reason for this is before, for each ray we wanted to generate, we could specify a single index into our batch of cameras, but if you have multiple camera dimensions, for each ray, you need to now specify one index per camera batch dimension, meaning our camera_indices we specify need to have one extra dimension at the end, a dimension of size Cameras.ndim (number of batch dimensions in the camera object).
I also added some neat logic in generate_rays where if you want your output raybundle to be batched, if you specify batched camera_indices, you will receive the same batch size for the output raybundle, meaning you can easily generate raybundles corresponding to things like image patches and whatnot.

Phototourism changes:
The difficulty here is that images in the phototourism dataset are of different sizes, which is difficult because we can no longer batch the image tensors. We also need to better utilize caching in the CacheDataloader since the vanilladatamanager defaults don't really use it as well as they could, and without properly utilizing it, the rays / sec are maybe 25% of the standard nerfstudio dataset speeds (which are 10% of the size of the phototourism datasets). To do this, we specify a collate function in the existing datamanager configs, and we make a subclass of the vanilladatamanagerconfig named VariableResDataManagerConfig which will have a custom collate function storing the images in a list, and using better defaults for caching in the dataloader, caching ~40 images at a time. The slowest part of using a list is pixel sampling, where we added a custom if case for if our pixel sampler receives a batch of images that are in a list, and the lack of ability to vectorize operations over a list means that caching in the dataloader is even more important.
2022-11-25 22:16:11 -08:00
Jake Austin
1eae0839b4
Updating the script for downloading data (#968)
* Updating the script for downloading data

* Making update to config structure

* Updating documentation and colab

* Adding back the save directory

* Pushing an all flag for nerfstudio

* forgot to save some previous edits
2022-11-25 21:40:13 -08:00
Matthew Tancik
2c64ba5b6a
Handle missing metashape data (#1015)
* Handle missing metashape data

* lint
2022-11-23 16:21:36 -08:00
Matthew Tancik
27ec772831
Improve webrtc stability (#1014)
debugging webrtc failed connections
2022-11-23 14:19:17 -08:00
Matthew Tancik
e4b5dea6e4
Fix dataparser filename bug (#1013) 2022-11-23 14:04:07 -08:00
Daniel Crispell
965dcb92cf
add support for per-image mask filenames in transforms.json (#957)
* add support for per-image mask filenames in transforms.json

* pass None instead of empty list to DataParserOutputs constructor

* look for downsampled masks in correct folder
2022-11-23 10:18:06 -08:00
Brent Yi
55db89d0c5
(minor) --trainer.load_dir => --trainer.load-dir in docs (#1010)
`--trainer.load_dir` => `--trainer.load-dir` in docs
2022-11-23 08:51:31 -08:00
dozeri83
0a8ebf93f1
implemented threadpool mechanics when loading images (#994)
Co-authored-by: dror <you@example.com>
Co-authored-by: Matthew Tancik <tancik@berkeley.edu>
2022-11-22 11:18:04 -08:00
Zak Murez
0264aacbe1
create rays on device when rendering (#993)
Co-authored-by: zmurez <zmurez@eng.ucsd.edu>
2022-11-22 11:07:36 -08:00
Matthew Tancik
a6d5b02446
Add metashape to docs (#1004) 2022-11-22 10:30:12 -08:00
Matthew Tancik
d97b834684
Simplify documentation (#1003) 2022-11-22 09:40:14 -08:00
Julian Knodt
0d3cfe2d38
bugfix: AABB normalized_positions to [-1, 1] (#998)
AABB normalized_positions to [-1, 1]

Previously, SceneBox.get_normalized_positions() returned a value in 0,1.
This was fine for InstantNGP and Nerfacto, which use these values directly in an MLP.
For TensoRF though, this was an error because grid sample expects values in the range [-1,1].
Thus, I think only 1/4th of the encoding planes were actually being used, and half of the
encoding lines were.
2022-11-21 12:15:01 -08:00
ksnzh
d767cd5011
fix typo error which should be self.cameras.times (#991) 2022-11-20 21:02:58 -08:00
Matthew Tancik
4065bb4733
Add Metashape support (#989)
Add metashape support
2022-11-20 14:55:41 -08:00
Ethan Weber
b829499c12
Computing and predicting normals (#905)
* staring normals PR

* code runs but visualization looks bad

* Camera and world coordinate conventions (#889)

* documented coordinate conventions without a figure for now

* another sentence

* update frames docs

* explain intrinsics and extrinsics

* more comments

* fix build

* Start viewer with more accurate speed calculation (#899)

Start viewer with more accurate speed calc

* Minor doc updates (#903)

* fix conflicts

* code runs but visualization looks bad

* normals implemented and vis fixed. still debugging quality

* minor name change

* comment

* fix black

* lint

* predicting normals

* adding detach

* predicted normals

* improved normals prediction performance

* cleaning code

* cleaning code

* cleaning code

* adding type to function param

* fixing tensorf forward func

* reverting tensorf changes

* fix style with tensorf field

* adding normals check to tensorffield

Co-authored-by: Matthew Tancik <tancik@berkeley.edu>
Co-authored-by: terrance <terrance.3512@gmail.com>
2022-11-20 13:22:37 -08:00
Nikhil Mohan
a82d19f549
Support downsample in dataset (#973)
* init

* tests passing

* pass through arguments

* verbose

* remove debug prints

* switch to bilinear

* update viewer

* make parser outputs private

* change name
2022-11-19 17:06:34 +00:00
Matthew Tancik
aaaa2fa393
Simplify readme (#982)
* Simplify readme

* Cleanup
2022-11-18 20:56:37 -08:00
Matthew Tancik
532f8067fd
Refactor process data (#976) 2022-11-18 09:19:44 -08:00
Matthew Tancik
1f85a31120
Update viewer sensitivity (#975) 2022-11-17 21:54:34 -08:00
Matthew Tancik
54d822afcc
v0.1.10 (#974) 2022-11-17 10:17:52 -08:00
Matthew Tancik
009e21be24
Fix loading in inference mode (#965) 2022-11-15 21:06:54 -08:00
Matthew Tancik
95a27ecba5
Add support for polycam (#958)
* Add support for polycam

* Modify blur score

* Support processing zip and add undistort fallback

* Created using Colaboratory

* Update docs

* Update demo.ipynb

* Address comments

* fix lint
2022-11-15 17:55:29 -08:00
Matthew Tancik
a7b2af185c
Cleanup configs (#963) 2022-11-15 16:56:56 -08:00
Alexander Kristoffersen
9ee6c31c02
Small Typos (#960)
small typos.
2022-11-15 14:59:31 -08:00
Julian Knodt
9ff167d705
Add temporal distortion and D-NeRF model (#825)
* Add D-NeRF temporal distortion and D-NeRF model

This allows for temporal distortion to be enabled on a variety of models through config
parameters, without an explicit model. Temporal distortion and MIP-NeRF are not currently
jointly runnable but can be changed in the future.

* Simplifying changes

* Simplifying changes

Co-authored-by: Matthew Tancik <tancik@berkeley.edu>
2022-11-15 14:29:29 -08:00
dependabot[bot]
8678f4c2e6
Bump loader-utils from 2.0.2 to 2.0.4 in /nerfstudio/viewer/app (#962)
Bumps [loader-utils](https://github.com/webpack/loader-utils) from 2.0.2 to 2.0.4.
- [Release notes](https://github.com/webpack/loader-utils/releases)
- [Changelog](https://github.com/webpack/loader-utils/blob/v2.0.4/CHANGELOG.md)
- [Commits](https://github.com/webpack/loader-utils/compare/v2.0.2...v2.0.4)

---
updated-dependencies:
- dependency-name: loader-utils
  dependency-type: indirect
...

Signed-off-by: dependabot[bot] <support@github.com>

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
2022-11-15 14:11:51 -08:00
Matthew Tancik
6baf31dc9a
Fix python 3.7 support (#959) 2022-11-15 09:51:23 -08:00
Nikhil Mohan
b5c34f0154
Add 'inference' mode to pipeline (#956)
* cleanup test mode

* cleanup eval loading

* change default

* cleanup

* cleanup

* comment
2022-11-14 16:56:36 +00:00
Matthew Tancik
b67f96fc56
Filter torchvision warnings (#954) 2022-11-12 18:23:24 -08:00
Matthew Tancik
5b0f1ffacb
Remove unimplemented interp render option (#953) 2022-11-12 17:18:11 -08:00