Files
nerfstudio/docs/tutorials/data/choosing_dataset.md
T
2022-06-28 16:51:08 -07:00

56 lines
1.5 KiB
Markdown

# Dataset configuration
#### Referring to a specific dataset
Here is how we refer to a dataset. We have a `dataset_inputs_train` and a `dataset_inputs_eval` config. These live under the `data` directory. You can see `configs/graph_default.yaml` for more details on what a full config would look like.
```yaml
data:
dataset_inputs_train:
data_directory: data/instant_ngp/fox
dataset_format: instant_ngp
downscale_factor: 1
alpha_color: white
dataset_inputs_eval:
data_directory: data/instant_ngp/fox
dataset_format: instant_ngp
downscale_factor: 1
alpha_color: white
```
#### Sanity check with a notebook
You can use `notebooks/visualize_dataset.ipynb` to see how your formatted dataset looks. This code will load a dataset from a yaml config, show the camera poses, and visualize rays with the specified collider in the config. TODO: add a better visualization with images too
![visualize_dataset](media/creating_dataset_visualize_dataset.png)
#### Supported datasets
We include support for a set of standard dataset formats.
###### Blender Synthetic
```{button-link} https://drive.google.com/drive/u/1/folders/128yBriW1IG_3NJ5Rp7APSTZsJqdJdfc1
:color: primary
:outline:
Download
```
```yaml
dataset_format: blender
```
Classic NeRF dataset including the "lego bulldozer" scene that was released with the original NeRF paper.
###### Instant NGP
```{button-link} https://github.com/NVlabs/instant-ngp#nerf-fox
:color: primary
:outline:
Download
```
```yaml
dataset_format: instant_ngp
```