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

290 Commits

Author SHA1 Message Date
jlightfoot
f22e9f4db9
Fix quickstart doc for remote host viewing setup (#2101)
Fix remote viewing setup doc

SSH command should specific localport:remotehost:remoteport rather than localport:localhost:remoteport in -L switch.  Changed description for clarity.
2023-06-22 16:22:45 +00:00
Shiva Gantha
8cdec3e1d2
Update nerfacto.md (#2107) 2023-06-20 19:39:11 -07:00
Terrance Wang
b17268e83b
[WIP] dreamfusion implementation (#1008)
* starting the framework for dreamfusion

* Removed view direction from RGB to calculate albedo

* Started diffusion

* Added dreamfusion to CLI, removed direction embeddings

* Normals in dreamfusion

* Lambertian shading added

* Train normals in Dreamfusion

* Stable diffusion from Stable Dreamfusion

* Fix normal orientation loss

* Configurable light direction sampling

* adding density initalization at origin

* V1 datamanager

* Don't train normals for first 1k iters

* SDS loss added to dreamfusion

* Dreamfusion pipeline

* Background MLP and SphereCollider

* Normals prediction training fixed

* Sigmoid on background, prompt in config

* Shading hyperparameter adjustements

* Shading hyperparameter tuning

* Location based prompting, density initialization fades, LR change

* Cleaned imports, more config options

* Tuning, more config changes

* remove unused sampler, normalize predicted normals from network

* change position encoding to direction encoding for background color mlp

* Updating dreamfusion to use custom trainer, fixed bug, multi-gpu support (put diffusion on one specific device), merged with main

* Hard coding less, offloading to config

* Appeasing the linter

* First pass opacity / sparsity losses

* Small patches and configs

* Accidentally pushed logs

* Removing logs for real this time

* Transmittance loss bug fix

* 12 GB Dreamfusion!

* Lint

* Remove unecissary collider changes

* Start SD cleanup

* Cleanup SD code

* adding overhead location prompting

* Use lambertian renderer

* turn grad accum off by default, decrease background mlp size, add overhead text embedding

* Cleanup

* Fix normals vis

* change uniform sampler to prediction network, add option to jit unet for faster sds_loss

* Use shaders

* cleaning up jit code

* add warning if using non jit unet

* add proposal network optimizer to config

* speed up render and cleanup code

* Clean up datamanager

* Add distortion loss

* fix wanb

* adding interpolated location sampling

* Update configs

* interpolated location prompting option, lowering init density, transmittance loss interpolation flipped

* rename alphas mult to alphas loss mult

* Mess around with normals

* add option to use random bg color + bg mlp

* Remove Dreamfusion Trainer (#1423)

* Dreamfusion trainer no longer needed

* Added typing for grad scaler

* SDS return loss

* Fix CFG

* Cleanup

* Update configs

* Fix opacity loss, use unform sampling first round of prop

* add prompt edditing to viewer

* Choose your own diffusion

* remove dreamfusion pipeline file

* merging + remove dreamfusion pipeline file

* Update configs

* add stable diff utils

* cleanup

* cleanup

* Fix test

* positional prompting bugfix

* Warmups

* tune

* lint

* Remove dataparser

* merging + text embedding bugfix

* Set data path to launch viewer for DreamFusion (#1776)

* Set data path to launch viewer for DreamFusion

* Apply formatter

* Fallback to 'base_dir' if data manager has no path

* Make if statement non-inline

* Reformat

---------

Co-authored-by: Matthew Tancik <mattt4001@gmail.com>

* add deepfloyd option first try

* merging

* merging from main

* merging

* removing comments, cleaning code, making stable diff utils usable for deepfloyd, remove deepfloyd text encoder after use

* undo chagnes in data folder

* cleaning code, removing unneccesary changes from main, removed normals prediction network

* remove old scripts files, rename generative pipeline flag to no_input_data, revert to default histogram padding in proposal sampler

* changing pipeline flag

* move no input data flag from pipeline to datamanager

* undo config and merge in deepfloyd from main

* merging grad scaler pipeline addition

* merging + setting proposal net scheduler

* revert nerfacto config change

* typing + linting

* rename dreamfusion datamanager to random camras datamanager, check datamanager typing in pipeline and trainer

* fix test import

* fix test import

* changing datamanager check in trainer, add typing in random cameras datamanager

* add Tensor to randomcamerasdatamanager

* remove old comments

* resolve tyro warning

* update fields to use tcnn

* lint

* add packages for deepfloyd

* add versions

* docs first draft

* rename to generfacto, clean docs

* fix import error

* minor changes

* fix orientation loss bug, retune params

* cleanup

* typing

* trying docs fix

* fixing generative package imports

* fix pyright

* docs fix

* merge w main

* docs minor change

* tuning + bugfixes

* remove torch.backends

---------

Co-authored-by: Abhik Ahuja <ahujaabhik@berkeley.edu>
Co-authored-by: David McAllister <davidrmc2@gmail.com>
Co-authored-by: Jake Austin <jake-austin@berkeley.edu>
Co-authored-by: Matthew Tancik <tancik@berkeley.edu>
Co-authored-by: Matt Tancik <mattt4001@gmail.com>
Co-authored-by: SeungWoo Yoo <dreamy1534@gmail.com>
Co-authored-by: ayaanzhaque <ayaanzhaque@gmail.com>
Co-authored-by: Ayaan Haque <ayaanzhaque@berkeley.edu>
2023-06-17 23:34:04 -07:00
LSong
10dab142d0
Move out NeRFPlayer: clean files and add method link (#2072)
* clean files and add method link

* fix black

* update
2023-06-14 13:51:09 -07:00
Sebastiaan
893730fa05
installation.md: Use https instead of ssh for git clone. (#2030)
* installation.md: Use https instead of ssh for git clone.

Https git clone can be run by anyone directly, without account, whereas ssh git on Github requires having a Github account set up.

* Add .git

---------

Co-authored-by: Matthew Tancik <mattt4001@gmail.com>
Co-authored-by: Matthew Tancik <tancik@berkeley.edu>
2023-06-04 19:23:12 -07:00
Cyrus Vachha
6993f75dc8
Render Omni-directional stereo video (360 3D VR) (#1986)
* Render Omni-directional stereo (360 3D VR)

Adding the omni-directional stereo camera model to render 3D 360 video and images for VR. This includes the camera models for the left and right eyes and additions to the render script to stack the left and right renders for the final render.
I have also updated the documentation on the custom_dataset page to have information on ODS but let me know if there is a better place to put it.

* Updated ODS camera model and render paths

Updates based on suggested code changes including adding a helper function for ODS and improving path creation.
I also added a part where if a user tries to render a video with an output path that doesn't have an extension it will automatically add .mp4 to the end. Otherwise this usually results in an error when there is no file extension when rendering a video.

* Updated formatting for render and cam path scripts

Updated python formatting for render and camera_paths script

* Adding type checking for ODS function and reformatting camera type code

Added the type checking and doc string for the ODS camera rays function and reformatted the cam_type ray if statements.
2023-05-26 18:01:13 -07:00
Ethan Weber
e94d9031ab
SIGGRAPH 2023 Citation Update (#1997)
* siggraph 2023 update

* update citation

* update citation
2023-05-24 18:16:01 -07:00
Matthew Tancik
76165fcd02
Fix custom dataparser docs (#1991)
Update new_methods.md
2023-05-24 11:18:35 -07:00
Sophus
3a31da36f7
fix visualize encoding bug (#1984) 2023-05-24 10:01:22 -07:00
Milan van Wouden
d413ca4204
fix large number of typos in repository (#1977)
* fix large number of typos in repository

* fix typos in tests
2023-05-23 10:41:49 -07:00
Matthew Tancik
a90602060b
Update installation docs (#1970)
* Update installation docs

* Update instructions

* Make tcnn consistent
2023-05-22 17:39:32 -07:00
Justin Kerr
0ee63cb581
Expose viewer controls to pipeline (#1804)
* Increase max_res of first proposal network

* start on ViewerCamera

* crop controls exposed

* lint

* fix viewer crash when invisible element has its value set

* fov, look_at, and camera type

* stuff

* set update interval correctly in cameracontrols, smooth transitions work now

* demo with depth zooming

* depth min

* WIP adding click message

* put the clock back in viewerwindow

* use setlookat when pos and target are defined at the same time

* remove demo from nerfacto

* Add click commands as rays (origin, direction)

* short doc

* Fix double event for doubleclick

* add ViewerClick, cleanup docs

* add docs

* cleanup

* remove scripts

* remove nerfacto changes

* js compile things

* update viewer version

* fix mutable state thing

* docs improvements

* add kwarg value for enableTransition

* fix

* lint

* path in sync-defs

* Misc doc edits requested in pr

* Add click description

* WIP remove redefinition of banner_height, refer to SCSS vars

* add docstring

* radix

* radix

* speed

---------

Co-authored-by: Chung Min Kim <chungmin99@gmail.com>
2023-05-18 17:36:00 -07:00
Jonáš Kulhánek
9584fa9b39
Add pre-commit (#1949)
* Add pre-commit

* Add pre-commit to docs

---------

Co-authored-by: Brent Yi <yibrenth@gmail.com>
2023-05-18 07:13:58 -07:00
Matthew Tancik
2b0831b6d4
Final typing fixes + add Pyright to CI (#1944)
* knocking out a couple

* Fix more errors

* Add pyright

* Remove some unnecessary cv2 type ignores, warning on missing imports
(for missing tinycudann)

* Minor annotation fixes

* Update docs

* Drop `None` from temporal distortion output

---------

Co-authored-by: Brent Yi <yibrenth@gmail.com>
2023-05-18 02:25:34 -07:00
Jonáš Kulhánek
e898f562e6
Switching to Torch 2.0 by default. (#1922)
* Update requirements for torch 2.0

* Update docs

* Fix missing _LRScheduler import

* Fix pylint issues

* Update docs, drop 1.13

* Fix pip postfix version

* fix docs

* Allow python 1.13 (for now)

* Minor fix for torch.compile

* Switch to torch.compile

* ruff

* Fix test

---------

Co-authored-by: Matt Tancik <mattt4001@gmail.com>
Co-authored-by: Matthew Tancik <tancik@berkeley.edu>
2023-05-17 16:03:17 -07:00
Giodiro
be291970db
Documentation for k-planes add-on (#1935)
* Add documentation for kplanes add-on

* Fix all integration spots with nerfstudio

---------

Co-authored-by: gmeanti <giacomo.meanti@gmail.com>
2023-05-17 11:48:08 -07:00
Brent Yi
d542646c72
Switch to ruff (#1939)
* `isort` -> `ruff`

* `pylint` -> `ruff`

* Remove line

* More minor cleanup

* Better documentation for rules we are enabling

* Fix gitignore, run black

* vscode settings

---------

Co-authored-by: Matt Tancik <mattt4001@gmail.com>
2023-05-17 10:08:14 -07:00
Brent Yi
6ba2a18b2b
torchtyping => jaxtyping (#1838)
* Run `ruff --fix`

* Add jaxtyping migration script

* Run migration scripts + ruff + black

* Merge remote-tracking branch 'origin/main' into brent/jaxtyping

* black

* Fix lint

* fix lint

* Cleanup and more specific types

* Fixed annotation errors

* Fix more errors

* Fix annotation

* Delete _torchtyping_to_jaxtyping.py

---------

Co-authored-by: Matt Tancik <mattt4001@gmail.com>
2023-05-16 15:57:47 -07:00
Jonáš Kulhánek
9b6290ea43
Discover external methods from CLI (#1934)
* Discover external methods from CLI

* Fix external method registration

* External method add docstrings

* Update docs: register custom method in external_methods.py
2023-05-16 22:15:47 +02:00
Matthew Tancik
6c3106004d
Update contributing docs (#1926) 2023-05-16 00:06:43 -07:00
Matt Tancik
9932b0fecf fix doc error 2023-05-15 15:01:39 -07:00
J.Y
7851dfa9db
Fix bug related to distributed data parallel training (#1856)
* Fix bug related to distributed data parallel training

* fix format by black

* Add nerfacto-big multi-gpu commands

* cleaned up docs

* reverted default dataset to poster

---------

Co-authored-by: Ethan Weber <eweb0124@gmail.com>
2023-05-15 14:23:29 -07:00
Matthew Tancik
415a037880
Update method docs (#1925)
* Update method docs

* Cleanup
2023-05-15 13:47:40 -07:00
wuzirui
bc110d7bd9
extract dataparser configs to a stand-alone file, support custom dataparsers (#1909)
* extract dataparser configs to a stand-alone file, support custom dataparsers

* add license

* fix pylint issues

* fix type error issue

* update naming; add documentations;

* print f string

* fix pylint issue

---------

Co-authored-by: Matthew Tancik <tancik@berkeley.edu>
2023-05-12 15:11:46 -07:00
Ayaan Haque
fe4586839a
Add INGP Docs (#1774)
* initial draft

* updates

* added details on sampling

* updates

* final notes

---------

Co-authored-by: Ruilong Li(李瑞龙) <ruilongli94@gmail.com>
2023-05-12 14:51:00 -07:00
Matthew Tancik
72a14b0b22
Move scripts folder into nerfstudio (#1897)
move scripts folder into nerfstudio
2023-05-09 16:44:21 -07:00
Matthew Tancik
df514b121c
Update docs for ns-viewer (#1892) 2023-05-09 10:07:11 -07:00
Brent Yi
63c6303736
Bump minimum Python version to 3.8 (#1876)
* Bump minimum Python version to 3.8

* typing_extensions.Literal -> typing.Literal

* ns-dev-test

* typing_extensions.get_args => typing.get_args

* Format with older version of black

* Bump `black`

* Run black
2023-05-06 13:29:48 -07:00
Jonáš Kulhánek
1147fe4d5c
Make VanillaDataManager generic (#1868)
* Make VanillaDataManager generic

* Remove removed classes from docs

* Simplify generic datamanager

* Simplify code

* Add serialization tests
2023-05-06 09:54:52 -07:00
Jonáš Kulhánek
971c9eb307
Enable method config registration from env variable (#1869) 2023-05-04 20:53:52 +02:00
Jonáš Kulhánek
61bad4dab3
Fix flag name in tetra-nerf (#1870)
Fix flag name for tetra-nerf
2023-05-04 09:22:32 -07:00
Jonáš Kulhánek
686e995dd6
Tetra-NeRF docs - fix versions and table formatting. (#1864)
* Add Tetra-NeRF to the docs

* Fix versions in tetranerf docs
2023-05-03 14:50:30 -07:00
Jonáš Kulhánek
f4064aa593
Add Tetra-NeRF to the docs (#1862)
* Add Tetra-NeRF to the docs

* Add memory requirements + different model configurations
2023-05-03 22:18:07 +02:00
Matthew Tancik
610fe8bbd4
Update logging docs (#1863) 2023-05-03 10:00:46 -07:00
Milan van Wouden
28c84d116b
Fix typos in Ray Samplers docs (#1851) 2023-05-02 09:29:47 -07:00
Frédéric Devernay
9a0ecb6fd7
Support equirectangular sensor with metashape (#1841)
* Document how to use equirectangular sensor with metashape

* equirectangular sensor is actually supported by nerfstudio!

* Update custom_dataset.md

* equirectangular: fix COLMAP workflow

* metashape_utils.py: black

---------

Co-authored-by: Frédéric Devernay <f-dy@users.noreply.github.com>
2023-05-01 14:44:00 -07:00
Matthew Tancik
9a61851bcf
Update lerf docs (#1782) 2023-04-20 10:06:13 -07:00
Rohan Mathur
bf3ae6ac5c
Added docs wording change and also --crop-bottom (#1767)
* added docs wording change and also --crop-bottom

* fixed linting
2023-04-20 09:58:02 -07:00
Chung Min Kim
1d92cea2e3
Add documentation for LERF (#1771)
* Initial docs for lerf

* Add memory requirements for lerf-big and lerf

* Include LERF paper dataset info, include bibtex
2023-04-19 14:40:10 -07:00
Brent Yi
63919c0db2
Docs fixes for viewer elements (#1770) 2023-04-18 19:15:29 -07:00
Jonáš Kulhánek
ea14152ae6
Add PyTorch profiler to support tracing of forward/backward calls (#1761)
* Add PyTorch profiler support

* Add support for wrapped code, fix linting errors

* Update docs
2023-04-18 20:29:09 +02:00
cvachha
7c4e139f2e
Adding Blender tutorial video and documentation on NeRF in NeRF composite (#1745)
* Added Blender tutorial video and documentation on NeRF in NeRF compositing

Updated Blender Add-on documentation to include a video tutorial and added documentation on compositing NeRF objects in NeRF scenes.

* Updating Blender documentation

Removing additional images and resizing images on the documentation.
2023-04-16 10:13:15 -07:00
LownyCGI
16725ae8fa
Update sdfstudio.md (#1744)
Add supported method
2023-04-15 09:16:39 -07:00
Brent Yi
2ddb2631b5
Improve viewer element types (#1743) 2023-04-15 01:14:38 -07:00
Matthew Tancik
a9eea19880
Update method docs (#1742)
* Update method docs

* typo
2023-04-14 15:02:45 -07:00
Brent Yi
6a7c8820b8
Viewer backend refactor (#1702)
* Copy in `viser` communication infrastructure

* Update background image

* Add build directory to fix viser HTTP server
May want to remove these later

* Copy in `viser` communication infrastructure

* Update background image

* Pass dataset vis using viser

* initial impl of ViewerParameter

* idk what i did but brent and matt know

* Suppress eslint typescript warnings

* implement ViewerElement base class, then ViewerButton, ViewerSlider, etc

* client -> server message setup

* multi folder stuff

* Stream controls over viser

* import

* add types to viewer_param.py

* Move another message to viser

* Update folder color

* Use non throttled messages

* multiple folders works

* Cleanup, more porting

* starting impl of control panel in viser

* More porting

* scenebox to viser

* Cleanup

* Fix render updates

* More cleanup, docs, typing

* start control_panel.py

* set_hidden logic but not merged in

* control_panel replicates behavior of existing control panel (visually), except for setting the list of output options. not yet integrated with viewer_utils.py for logic flow. also implemented set_disabled and set_hidden in javscript/ViewerElement side

* Remove some original leva components

* add some train util logic, attempt set_value, doesn't work on javascript side fully yet (value updates but not slider visually

* bye zmq

* missing call to decoder

* 'gui_set' works properly, control panel now replicates behavior of original

* remove print

* Fix status viewer message

* More cleanup

* support set_options in dropdown

* hiding buttons works correctly

* fix set_options, switch from cur_value to get_value()

* Fix slowdown

* output options

* colormap logic converted

* add handle to cb_hook

* fix some lambda bugs

* cleanup

* Save checkpoint logic readded

* Cleanup and minor fixes

* snappier ui

* Refactoring

* Fix state update bugs

* Bug fixes

* Write less to redux

* added comments to reducer code

* Render state machine (#1727)

* state machine works, but train step throttling not implemented

* more step logic, cleanup

* remove random try catcH

* mutex logic for render

* Update lock logic

* types and docs

* added some state machine logic to help rerenders

* merge

---------

Co-authored-by: Justin <justin.g.kerr@gmail.com>

* Refactor and cleanup

* More cleanup and docs

* docstring to gui_utils

* docstrings, rename viewer_param.py to viewer_elements.py

* Fix viewer, remove bridge stuff

* Render on connect

* Fix lint errors

* More eslint fixes

* render_freq logic

* better rerender logic for crop

* More cleanup

* Purge

* Cleanup

* Lint fixes

* Fix cache

* better parsing logic

* Fix infinite test

* More cleanup

* typo

* Update webdocs

* Update docs

* rerender on custom hooks

* add more docs

* Use shared utilities in `viser.infra`

* rework callback to use elements instead of handles

* Fix ESlint errors

* change naming of handle to element, add setter for value

* cleanup

---------

Co-authored-by: Matt Tancik <mattt4001@gmail.com>
Co-authored-by: Justin <justin.g.kerr@gmail.com>
Co-authored-by: Ethan Weber <eweb0124@gmail.com>
Co-authored-by: Matthew Tancik <tancik@berkeley.edu>
2023-04-14 14:53:10 -07:00
Matthew Tancik
e5e67a7a5a
Update blender web docs (#1740)
Update blender docs
2023-04-14 11:42:19 -07:00
Ayaan Haque
b85cb9f979
update in2n docs based (#1721)
* base changes

* done for now, just need assets

* fix

* add few changes

* remove unused files

* fixse
2023-04-11 13:05:51 -07:00
Matthew Tancik
2c6a328780
Add instruct-nerf2nerf to docs (#1711)
* base changes

* done for now, just need assets

* fix

* Update docs

---------

Co-authored-by: ayaanzhaque <ayaanzhaque@gmail.com>
2023-04-10 11:26:21 -07:00
Matthew Tancik
2e777e4577
Update dataparser docs (#1698)
* Update dataparser docs

* Add missing files
2023-04-04 18:30:52 -07:00