Train and eval data splits (Nerfbusters) (#2207)

* nerfbusters dataparser

* nerfbuster dataparser and visualization of eval views

* resolved some errors from  pyright

* enable ns-process-data to take two videos as input and use one for eval and the other for train

* resolve some pyright issues

* instead of data being a tuple, I added eval_data as another argument

* implemented train eval split for all cases

* implemented train eval split for all cases

* reverting equi changes

* reverting reality capture changes

* raise valueerror

* revert comment

* added split override code back to dataparser

* reverted one change

* using image border now

* adding docs

---------

Co-authored-by: Ethan Weber <eweb0124@gmail.com>
This commit is contained in:
Frederik WarburgandEthan Weber authored and GitHub committed 2023-08-04 10:11:55 -04:00
1 parent 388fcb09e1
commit 9b03299d40
14 files changed
+288 -83

No files matched your search

+2 -8
View File
@@ -11,14 +11,8 @@ from PIL import Image
from nerfstudio.data.dataparsers.nerfstudio_dataparser import NerfstudioDataParserConfig
from nerfstudio.data.utils.colmap_parsing_utils import Camera
from nerfstudio.data.utils.colmap_parsing_utils import Image as ColmapImage
from nerfstudio.data.utils.colmap_parsing_utils import (
qvec2rotmat,
write_cameras_binary,
write_images_binary,
)
from nerfstudio.process_data.images_to_nerfstudio_dataset import (
ImagesToNerfstudioDataset,
)
from nerfstudio.data.utils.colmap_parsing_utils import qvec2rotmat, write_cameras_binary, write_images_binary
from nerfstudio.process_data.images_to_nerfstudio_dataset import ImagesToNerfstudioDataset
def random_quaternion(num_poses: int):