Skip to content

Commit

Permalink
Merge branch 'main' into justin/share_button
Browse files Browse the repository at this point in the history
  • Loading branch information
kerrj authored Nov 30, 2023
2 parents a30bba8 + 5a772b1 commit 9bcfd20
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 2 deletions.
2 changes: 1 addition & 1 deletion nerfstudio/data/dataparsers/dycheck_dataparser.py
Original file line number Diff line number Diff line change
Expand Up @@ -322,7 +322,7 @@ def process_frames(self, frame_names: List[str], time_ids: np.ndarray) -> Tuple[
for frame in frame_names:
cv2.imwrite(
str(self.data / f"rgb/{d}x/{frame}.png"),
cv2.resize(cv2.imread(str(self.data / f"rgb/1x/{frame}.png")), (h, w)),
cv2.resize(cv2.imread(str(self.data / f"rgb/1x/{frame}.png")), (w, h)),
)
CONSOLE.print("finished")

Expand Down
13 changes: 12 additions & 1 deletion nerfstudio/scripts/render.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,6 +61,10 @@
VanillaDataManager,
VanillaDataManagerConfig,
)
from nerfstudio.data.datamanagers.parallel_datamanager import ParallelDataManager
from nerfstudio.data.datamanagers.random_cameras_datamanager import (
RandomCamerasDataManager,
)
from nerfstudio.data.datasets.base_dataset import Dataset
from nerfstudio.data.scene_box import OrientedBox
from nerfstudio.data.utils.dataloaders import FixedIndicesEvalDataloader
Expand Down Expand Up @@ -666,7 +670,14 @@ def main(self) -> None:

install_checks.check_ffmpeg_installed()

assert isinstance(pipeline.datamanager, VanillaDataManager)
assert isinstance(
pipeline.datamanager,
(
VanillaDataManager,
ParallelDataManager,
RandomCamerasDataManager,
),
)
steps = int(self.frame_rate * self.seconds)
camera_start = pipeline.datamanager.eval_dataloader.get_camera(image_idx=0).flatten()
camera_path = get_spiral_path(camera_start, steps=steps, radius=self.radius)
Expand Down

0 comments on commit 9bcfd20

Please sign in to comment.