mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-14 03:44:02 +08:00
77 lines
2.2 KiB
Markdown
77 lines
2.2 KiB
Markdown
# MattPort
|
|
|
|
|
|
# Quickstart
|
|
## Installation: 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 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
|
|
```
|
|
|
|
# Logging and profiling features
|
|
We provide many logging functionalities for timing and/or tracking losses during training. All of these loggers are configurable via `configs/logging.yml`
|
|
|
|
1. **Writer**: Logs losses and generated images during training to a specified output stream. Specify the type of writer (Tensorboard, local directory, Weights and Biases), and how often to log in the config.
|
|
|
|
2. **StatsTracker**: Computes select statistics and prints to the terminal for local debugging. Specify the kinds of statistics to be logged, how often to log, and the maximum history buffer size in the config.
|
|
|
|
3. **Profiler**: Computes the average total time of execution for any function with the `@profiler.time_function` decorator. Prints out the full profile at the termination of training or the program.
|
|
|
|
|
|
# Setting up Jupyter
|
|
|
|
```
|
|
python -m jupyter lab build
|
|
bash environments/run_jupyter.sh
|
|
```
|
|
|
|
# Tooling
|
|
|
|
We use [autoenv](https://github.com/hyperupcall/autoenv) to make setting up the environment and environment variables easier. This will run the `.envrc` file upon entering the `/path/to/mattport` folder. It will also remove the set environment parameters upon leaving.
|
|
|
|
```
|
|
# Install direnv.
|
|
sudo apt install direnv
|
|
|
|
# Add the following line to the bottom of your ~/.bashrc file.
|
|
eval "$(direnv hook bash)"
|
|
|
|
# Populate your .envrc with commands you want to run. Then, run the following to allow updates.
|
|
cd /path/to/mattport
|
|
direnv allow .
|
|
```
|
|
|
|
<!-- Our PyTorch version uses cuda-11.3. -->
|
|
<!-- export PATH=/usr/local/cuda-11.3/bin:$PATH --> |