tink/nerfstudio
4
0 Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-15 12:23:46 +08:00
Code Activity
1,804 Commits 85 Branches 40 Tags
a6a3dedc0e2abcd93466ab2b656c7855e33f036b
Commit Graph
21 Commits
Author SHA1 Message Date
Matias Turkulainen a6a3dedc0e camtoworld is already a tensor in sdfstudio_data (#2777) 
camtoworld is already a tensor
 2024-01-17 11:22:34 -08:00
Matthew Hoffman f0aadf9de8 Use ruff as linter and formatter (#2765) 
* Switch from black to ruff format

* Sort dependencies with ruff check

* Run ruff format
 2024-01-15 13:43:14 -08:00
Sasha SaxSasha SaxEthan WeberAdamRashid96AdamRashid96
 		6e37b2b542 Enable SparseNerf depth ranking loss to be used with GT/Sensor depth (#2312) 
* add ranking loss to depth-nerfacto with GT depth

* formating

* pair pixel sampler bug fix from pr #2315

---------

Co-authored-by: Sasha Sax <asax@stanford.edu>
Co-authored-by: Ethan Weber <ethanweber@berkeley.edu>
Co-authored-by: AdamRashid96 <abrashid@berkeley.edu>
Co-authored-by: AdamRashid96 <71362382+AdamRashid96@users.noreply.github.com>
 2023-08-26 02:56:52 -07:00
machenmusik cb2405d3ea When generating depth images, use torch_compile with ZoeDepth for compatibility (#2303) 
* Use torch_compile with Zoe for compatibility

torch_compile module support

Previously the compatibility utility torch.compile worked for functions, but not for modules as torch.compile was originally employed here.

* torch_compile -> Any
 2023-08-09 17:08:17 -07:00
Ethan WeberAdamRashid96Matthew Tancik
 		71c6b9cf3d Monodepth Regularization (#2284) 
* monodepth implemented

* device

* pair pixel sampler

* docstring

* removing unused imports

* fix bugs

* cache in cache dir, no longer edit json file

* moving to depth nerfacto

* fixed bug using median depth by accident

* removing depthdataset from nerfacto

* added pixelsamplerconfig

* added equirectangular warning back

* bug fix

* delete sample_config.py + small fixes

* rename to pixel_sampler

* cleaning

* test

* passing num_rays_per_batch properly

* fixed transforms typo

* renaming in nerfacto

* remove depth files from test data

* removed defaulting to base pixel sampler with mask

* adding patch_size back

---------

Co-authored-by: AdamRashid96 <abrashid@berkeley.edu>
Co-authored-by: Matthew Tancik <tancik@berkeley.edu>
 2023-08-01 18:52:47 -07:00
Nicholas PfaffandJonas Kulhanek 9f487a464f Implement nerfacto alpha transparency training (#2165) 
* Implement nerfacto alpha transparency training

* Stop adding all ones alpha channel and clean up blend_background

* Make background_color a required argument in blend_background

* Enable alpha transparent training by default

* Improve background_color typing

* Blend uniform background color for metric computation

* Fix typing error

* Clean up typing

* Fix background_color to outputs assignment

* Model outputs returns unblended RGB (black bg) for random bg

* Update renderers

* Fix comments

* Add support for transparent BG for other methods

* Minor refac

* Fix get_background_color

* Fix typing error for blend_background

* Fix comments

* Export utils support transparency

* Pylint fix

* Fix pcd export masking

* Filter pcd view directions based on opacity

* Clean up get_rgba_image

---------

Co-authored-by: Jonas Kulhanek <jonas.kulhanek@live.com>
 2023-07-18 12:54:52 +02:00
Jonáš Kulhánek 3b0f7581d8 Reduce GPU memory usage in CacheDataloader (#1730) 
* Reduce GPU memory usage in CacheDataloader

* Move also masks and seg.masks, depths to cpu

* Fix linter issues
 2023-06-07 14:14:24 -04:00
Matthew TancikandBrent Yi 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
Brent YiandMatt Tancik 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 814db32ec4 More typing fixes (#1941) 
* Fix more types

* Fix tests

* Minor cleaning

* Address comments

* Resolve linting errors

* Re-ignore pymeshlab type

* Add extra assert / error message to mipnerf model
 2023-05-17 07:17:40 -07:00
Brent YiandMatt Tancik 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
Ethan Weber 13e0fce122 Update copyright (#1917) 
* updated copyright

* including txt file

* updated black

* removing license

* fixing license
 2023-05-12 16:16:13 -07:00
Pablo Vela ddfa463f4d NeuS Implementation (#1444) 
* add initial method configs and data manager

* add neus scheduler

* add initial surface model and neus

* add Neus ray sampler

* add sdf field

* move scale/shift calc to math utils

* add debug and fix sdf_dataset typo

* update vscode neus debug

* fix dataset and get initial working version

* add magic number from original repo

* black fixes

* clean up get weights/transmittance from alphas

* clean up typos in sdf datamanager

* type annotations on neus scheduler and remove unused params

* clean up and annotate sdf field along w/ other fixes

* more clean up

* remove unnessary directions norm

* resolve comments

* update neus scheduler to follow refactor

* fix linter errors

* fix some typos

* remove get_weights_from_alpha

* more comment fixes

* more linter error fixes

* clean up and add comments

* more little fixes for types/comments

* more comments

* rename NeuS Scheduler to DelayedScheduler

* rename DelayedScheduler to CosineDecayScheduler
 2023-03-12 16:01:18 -04:00
Pablo Vela a37c73fb07 sdfstudio dataparser/dataset (#1381) 
* add downloads for sdfstudio datasets

* add sdfstudio parser to work with sdfstudio data

* fix licensing and doc strings

* fix linter errors

* move depths/normals to metadata, create sdfdataset

* remove extra scenebox parameters

* more linting errors fix

* fix wrong annotation

* add in missing depth/normal files

* minor fixes

---------
 2023-02-12 10:31:31 -08:00
Matthew Tancik 5b50740f45 Add cropping options to data processing (#1236) 2023-01-16 15:34:20 -08:00
Michal Pandy 3da877050d Depth supervision for Nerfacto (#1173) 
* Implement depth supervision

* Fixes

* Change normalize dtype

* refactor

* Change depth weight

* Small refactor

* Change docstring

* Removu unused import

* Add assumption to depth comment

* Implement URF loss

* Remove default depth loss type param

* URF fixes

* Fix depth loss scale

* Visualize gt depths

* Fixes & documentation

* Documentation and formatting

* Fix formatting problems

* Fix linter problems

* Fix tests

* Fix tests
 2023-01-02 10:13:38 +01:00
Martin Piala a5eea5f14d Camera export (#1191) 
* implement the camera exporter

* remove train_dataset from exporter

* fix linter errors

* add docstring

* correct comment

* fix linter errors

* make image_filenames a property of the dataset
 2022-12-31 22:38:43 -07:00
Matthias Blaicher 06642a5871 Support grayscale-only images in dataset (#1142) 
* Support grayscale-only images in dataset

* lint
 2022-12-22 11:16:03 -07: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
Brent YiandEthan Weber ad8599aad0 Add missing __init__.py files (#935) 
* Add missing __init__.py files

* fixing lint

Co-authored-by: Ethan Weber <eweb0124@gmail.com>
 2022-11-10 16:56:13 -08:00
Nikhil Mohan 1a24f3e58c Refactor Data Pipeline (#896) 
* add none check

* black formatting

* cleanup

* cleanup

* fix tests

* remove new line

* init

* toy example working

* semantic nerf working

* simplify pixel samplers

* cleanup + fix tests

* comments

* fix tests

* fix tests
 2022-11-06 07:40:43 -08:00