4
0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 10:34:29 +08:00
A collaboration friendly studio for NeRFs
Go to file
evonneng 609fcb04f6
Set up profiler and moved stats tracker to its own class; set up configurables for stats tracker and profiler (#42)
* added skeleton for terminal logger

* done skeleton terminal writer logic... todo: actual calls to the write classes

* moved lambda to top function

* added logic to support timing

* prettier printing

* prettier printing

* lint

* separated out timing and other stats variables. printing in dict format

* keeping logging history and wiping rest

* updated logic to calculate total times instead of average times

* made stats tracker configurable

* moved stats tracker and profiler to own class file; implemented basic functionality for profiler

* lint removing duplicate function
2022-04-28 15:04:00 -07:00
.github/workflows Rearrange nerf files 2022-04-21 14:51:51 -07:00
.vscode merged from master 2022-04-22 03:51:28 +00:00
configs Set up profiler and moved stats tracker to its own class; set up configurables for stats tracker and profiler (#42) 2022-04-28 15:04:00 -07:00
environment resizing image 2022-04-22 18:19:39 +00:00
external Initial ray generation and dataset code (#12) 2022-04-12 15:08:16 -07:00
mattport Set up profiler and moved stats tracker to its own class; set up configurables for stats tracker and profiler (#42) 2022-04-28 15:04:00 -07:00
notebooks pylint and test cases for dataloader 2022-04-23 19:45:53 +00:00
scripts added test set (using training data for now) 2022-04-22 03:45:06 +00:00
tests pylint and test cases for dataloader 2022-04-23 19:45:53 +00:00
viewer some more structure 2022-04-02 21:47:39 +00:00
.gitignore removing docker and fixing nfs issue 2022-04-17 16:36:09 +00:00
.gitmodules
pyproject.toml Image dataset implemented and ray generation working. (#23) 2022-04-17 09:26:44 -07:00
README.md resizing image 2022-04-22 18:19:39 +00:00
setup.py newline for linting 2022-04-02 20:25:37 +00:00

MattPort

Setup the environment

# Clone the repo
git clone --recurse-submodules git@github.com:ethanweber/mattport.git

# Create the python environment
conda create --name mattport python=3.8.13
conda activate mattport
pip install -r environment/requirements.txt

# Install helper library
cd external/goat
python setup.py develop
# then go back, i.e., `cd /path/to/mattport`

# Install mattport as a library
python setup.py develop

# Running the test cases
pytest tests

Getting the data

Download the original NeRF dataset and put it in the following format.

- data/
    - blender/
        - fern/
        - lego/

Training a model

# Run with the default config
python scripts/run_train.py
# Run with config changes
python scripts/run_train.py machine_config.num_gpus=1

Setting up Jupyter

python -m jupyter lab build
bash environments/run_jupyter.sh