Skip to content

Commit f87446f

Browse files
added flowcharts into datamanagers.md
1 parent 1a92b5f commit f87446f

File tree

1 file changed

+26
-0
lines changed

1 file changed

+26
-0
lines changed

docs/developer_guides/pipelines/datamanagers.md

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -115,6 +115,32 @@ To train splatfacto with a large dataset that's unable to fit in memory, please
115115
ns-train splatfacto --data {PROCESSED_DATA_DIR} --pipeline.datamanager.cache-images disk
116116
```
117117

118+
Checkout these flowcharts for more customization on large datasets!
119+
120+
```{image} imgs/DatamanagerGuide-LargeNeRF-light.png
121+
:align: center
122+
:class: only-light
123+
:width: 600
124+
```
125+
126+
```{image} imgs/DatamanagerGuide-LargeNeRF-dark.png
127+
:align: center
128+
:class: only-dark
129+
:width: 600
130+
```
131+
132+
```{image} imgs/DatamanagerGuide-Large3DGS-light.png
133+
:align: center
134+
:class: only-light
135+
:width: 600
136+
```
137+
138+
```{image} imgs/DatamanagerGuide-Large3DGS-dark.png
139+
:align: center
140+
:class: only-dark
141+
:width: 600
142+
```
143+
118144
## Migrating Your DataManager to the new DataManager
119145
Many methods subclass a DataManager and add extra data to it. If you would like your custom datamanager to also support new parallel features, you can migrate any custom dataloading logic to the new `custom_ray_processor()` API. This function takes in a full training batch (either image or ray bundle) and allows the user to modify or add to it. Let's take a look at an example for the LERF method, which was built on Nerfstudio's VanillaDataManager. This API provides an interface to attach new information to the RayBundle (for ray based methods), Cameras object (for splatting based methods), or ground truth dictionary. It runs in a background process if disk caching is enabled, otherwise it runs in the main process.
120146

0 commit comments

Comments
 (0)