tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 02:24:38 +08:00
Code

Fix skipping non-existent images (#832)

Browse Source 
Fix skipping non-existant images
This commit is contained in:
Kevin Chen 2022-10-26 13:27:08 -07:00 committed by GitHub
parent cc49356d97
commit 359f955f3f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
nerfstudio/data/dataparsers/nerfstudio_dataparser.py
View File

@ -17,7 +17,7 @@ from __future__ import annotations
import math
from dataclasses import dataclass, field
from pathlib import Path, PureWindowsPath
from pathlib import Path, PurePath
from typing import Optional, Type
import numpy as np
@ -80,12 +80,9 @@ class Nerfstudio(DataParser):
num_skipped_image_filenames = 0
for frame in meta["frames"]:
if "\\" in frame["file_path"]:
filepath = PureWindowsPath(frame["file_path"])
else:
filepath = Path(frame["file_path"])
filepath = PurePath(frame["file_path"])
fname = self._get_fname(filepath)
if not fname:
if not fname.exists():
num_skipped_image_filenames += 1
else:
image_filenames.append(fname)
@ -177,7 +174,7 @@ class Nerfstudio(DataParser):
)
return dataparser_outputs
def _get_fname(self, filepath):
def _get_fname(self, filepath: PurePath) -> Path:
"""Get the filename of the image file."""
if self.downscale_factor is None: