Commit Graph
1618 Commits
Author SHA1 Message Date
Jonas Kulhanek 45c029d744 Add support for transparent BG for other methods 2023-07-17 13:39:55 +02:00
Jonas Kulhanek b994bf5f7f Fix comments 2023-07-17 13:21:03 +02:00
Jonas Kulhanek cf7b5a7c63 Update renderers 2023-07-17 13:13:48 +02:00
Jonas Kulhanek b6bbb15045 Model outputs returns unblended RGB (black bg) for random bg 2023-07-17 11:21:27 +02:00
Nicholas Pfaff edd594c850 Fix background_color to outputs assignment 2023-07-06 18:41:52 +01:00
Nicholas Pfaff 4c51bd4755 Clean up typing 2023-07-05 18:35:48 +01:00
Nicholas Pfaff 32900a6e5c Fix typing error 2023-07-05 18:30:24 +01:00
Nicholas Pfaff 324f1d4949 Blend uniform background color for metric computation 2023-07-05 08:35:42 +01:00
Nicholas Pfaff 119519d0c3 Improve background_color typing 2023-07-04 17:09:16 +01:00
Nicholas Pfaff bed684635e Enable alpha transparent training by default 2023-07-04 16:52:48 +01:00
Nicholas Pfaff c15fa1819d Make background_color a required argument in blend_background 2023-07-04 16:25:20 +01:00
Nicholas Pfaff e24f15baad Stop adding all ones alpha channel and clean up blend_background 2023-07-04 16:22:42 +01:00
Nicholas Pfaff 0bbaba6f1a Implement nerfacto alpha transparency training 2023-07-03 16:29:15 +01:00
Matthew Tancik 061d691d43 Default to tcnn (#2159) 2023-07-02 21:36:28 -07:00
Brent Yi d6bb8bd8e5 Fix use_appearance_embedding for instant-ngp (#1999)
* Attempt fixes for instant-ngp

* Run `ns-dev-test`

* Remove NGP field implementation

* Test nerfacto field instead of ngp field

* Black
2023-07-01 18:54:27 +00:00
Ilya Basharov 5c4b0a1dd6 add build-arg cuda version to dockerfile (#2139) 2023-07-01 17:06:05 +00:00
Milan van Wouden f4be2b729a Fix typo in nerfacto.py (#2149) 2023-06-30 08:44:53 +00:00
hturkiandMatthew Tancik 0719026c15 Add option to render evaluated images (#2135)
* Add option to render evaluated images

* add args

* fix formatting

* attempt to fix pyright

* attempt 2

---------

Co-authored-by: Matthew Tancik <matt@lumalabs.ai>
2023-06-29 23:26:56 +00:00
Jonáš KulhánekandMatthew Tancik b3df46314c Remove breakpoint from colmap_dataparser (#2141)
Co-authored-by: Matthew Tancik <matt@lumalabs.ai>
2023-06-29 17:03:29 +00:00
Konstantin Kozlovtsev 000272f895 Fix unnecessary encodings calculation in tensorf field (#2145)
Fixed unnecessary encodings calculation in tensorf field
2023-06-29 09:54:59 -07:00
machenmusikandMatias Turkulainen d78af5b569 allow ns-process-data to use CR2 raw images (#2119)
* allow ns-process-data to use CR2 raw images

* Update process_data_utils.py

* black wants an extra newline

---------

Co-authored-by: Matias Turkulainen <30566358+maturk@users.noreply.github.com>
2023-06-27 16:02:58 -04:00
Ethan WeberandMatias Turkulainen d9ee5752d2 Fix to fall back to torch if tcnn not found (#2124)
fix for falling back to torch if tcnn not found

Co-authored-by: Matias Turkulainen <30566358+maturk@users.noreply.github.com>
2023-06-27 09:33:28 -07:00
Matthew Tancik 6486c00d4d Update generfacto.md (#2132) 2023-06-25 23:48:33 -07:00
Jonáš Kulhánek aa6c052660 Jkulhanek/colmap dataparser (#2115)
* Add COLMAP dataparser implementation

* Fix downscale_images input

* Add help to COLMAP dataparser
2023-06-25 17:08:17 +00:00
Jonáš Kulhánek ac6a83cc53 Fix pickling of VanillaDataManager (#2129)
* Fix pickling of VanillaDataManager

* Fix vanilladatamanager init
2023-06-25 18:46:07 +02:00
Matthew Tancik 004d8ca9d2 Update nerfacto.md (#2126) 2023-06-23 17:56:11 -07:00
olegpokhilchenko ef1350bd36 Fix get_num_frames_in_video (#2090)
* Fix get_num_frames_in_video

* Make sure that there is a number

* Fix output[0]
2023-06-23 08:38:16 -07:00
Terrance WangandMatthew Tancik c023f82b11 Add examps to generfacto docs (#2120)
* add vids to generfacto docs

* update generfacto docs

---------

Co-authored-by: Matthew Tancik <tancik@berkeley.edu>
2023-06-22 21:24:29 +00:00
Matthew Tancik 7349f2320b v0.3.2 (#2118) v0.3.2 2023-06-22 09:25:36 -07:00
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
Jonáš Kulhánek 3214fcfeb5 Fix spherical harmonics #2081 (#2112) 2023-06-22 02:40:22 +00:00
Jonáš Kulhánek c68d622bbc Allow newer pycolmap for newer python (#2113) 2023-06-21 19:23:34 -07:00
b747db6877 Fixing the image_dir if using skip_image_processing (#2111)
Co-authored-by: khoa <tuankhoanguyen94@gmail.com>
Co-authored-by: Matthew Tancik <tancik@berkeley.edu>
2023-06-21 12:24:55 -04:00
machenmusik ccb8a25d4a Add option to process masks on GPU for speed at the expense of memory (#2110)
--pipeline.datamanager.masks-on-gpu True
2023-06-21 09:16:14 -07:00
Shiva Gantha 8cdec3e1d2 Update nerfacto.md (#2107) 2023-06-20 19:39:11 -07:00
Matias TurkulainenandMatthew Tancik e0e66ce9fb Add non-trainable camera indices to CameraOptimizer (#2080)
* add non-trainable indices to camera optimizer

* improve comment

* create on device

---------

Co-authored-by: Matthew Tancik <tancik@berkeley.edu>
2023-06-18 12:55:13 -04:00
ksnzhandZhang Jian c55f5d36b9 add gradient accumulation support (#2049)
* add gradient accumulation support

* fix 2 blank lines

* fix possibly unbound variable

* update gradient accumulation step with for loop

* add seert and ignore pyright check

---------

Co-authored-by: Zhang Jian <zhangjian49@lenovo.com>
2023-06-18 07:17:35 -07:00
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
Justin Kerr e4a5c38b28 Fix resolution rendering logic in statemachine for viewer beta (#2085)
* fix render statemachine logic for viewer beta
2023-06-17 21:15:33 -07:00
machenmusik 1066f8f433 don't force downscale factor to 1 for nerfacto-huge (#2088)
* per @kerrj, nerfacto-huge doesn't need downscale_factor=1
2023-06-16 23:49:56 -07:00
Matthew Tancik 7859f19f50 Fix data downscale paths (#2071)
* Fix data downscale paths

* ruff

* format

* Update process_data_utils.py
2023-06-16 11:40:12 -07:00
Justin Kerr f9d06d1f9c reduce nerfacto-huge memory (#2082)
reduce memory to 24G by reducing appearance embed dim
2023-06-15 16:33:30 -07:00
Justin Kerr 8fad92c39a nerfacto-huge (#2003)
An even larger version of nerfacto with scaled hashgrid resolution, MLP sizes, batch sizes, appearance embeddings, proposal sampler resolution, and more.
Also includes some minor tweaks to nerfacto-big configs
2023-06-15 11:05:53 -07:00
Matthew Tancik 26fc1c95dc Update data_utils doc (#2076) 2023-06-14 17:57:33 -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
Frédéric DevernayandFrédéric Devernay d18588d5d4 colormaps: add "gray" (#2058)
* colormaps: remove useless "pca" choice, add "gray"

- The "pca" choice didn't work when applied to a single-channel image (it crashed the viewer), and it's applied by default to any image of more than 3 channels, where the colormap choice is ignore anyway.
- Add the trivial "gray" colormap, useful to eg extract accumulation.

* colormaps.py: re-add "pca"

* control_panel: don't show the "pca" option either

---------

Co-authored-by: Frédéric Devernay <f-dy@users.noreply.github.com>
2023-06-11 18:41:23 -07:00
Weihang Li f2fdd10d9b fix for scannet dataset camera pose loading (#2056)
* fix scannet dataparser pose issue

* camera pose corrected
2023-06-10 09:07:57 -07:00
Frédéric DevernayandFrédéric Devernay e2e5637d05 nerfstudio dataparser: avoid loading images at inference time (#2051)
Also remove the image skipping logic, which seems to be a remnant of some old code (all images should exist)

Co-authored-by: Frédéric Devernay <f-dy@users.noreply.github.com>
2023-06-09 21:29:32 -04:00
Matias Turkulainen 88b253f506 Add kwargs to dataparser outputs (#2053)
add kwargs to dataparser outputs
2023-06-09 21:21:17 -04:00
henrypearce4D d01fef7650 Fix metashape multiple camera models error message (#2057)
Reverted back to the original error message with some extra clarification added.

cc #2011
2023-06-09 15:09:13 -07:00