tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 10:34:29 +08:00
Code

cleanup

Browse Source
This commit is contained in:
Ethan Weber 2023-01-27 23:22:56 +00:00
parent 7caafd6063
commit bb154c8801
3 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
nerfstudio/data/datamanagers/base_datamanager.py
View File

@ -45,11 +45,8 @@ from nerfstudio.data.dataparsers.friends_dataparser import FriendsDataParserConf
from nerfstudio.data.dataparsers.instant_ngp_dataparser import (
InstantNGPDataParserConfig,
)
<<<<<<< HEAD
from nerfstudio.data.dataparsers.mipnerf360_dataparser import Mipnerf360DataParserConfig
=======
from nerfstudio.data.dataparsers.minimal_dataparser import MinimalDataParserConfig
>>>>>>> main
from nerfstudio.data.dataparsers.mipnerf360_dataparser import Mipnerf360DataParserConfig
from nerfstudio.data.dataparsers.nerfstudio_dataparser import NerfstudioDataParserConfig
from nerfstudio.data.dataparsers.nuscenes_dataparser import NuScenesDataParserConfig
from nerfstudio.data.dataparsers.phototourism_dataparser import (

2
nerfstudio/data/pixel_samplers.py
View File

@ -234,4 +234,4 @@ class EquirectangularPixelSampler(PixelSampler): # pylint: disable=too-few-publ
* torch.tensor([num_images, image_height, image_width], device=device)
).long()
return indices
return indices

7
nerfstudio/engine/trainer.py
View File

@ -128,7 +128,12 @@ class Trainer:
self._check_viewer_warnings()
# set up writers/profilers if enabled
writer_log_path = self.base_dir / config.logging.relative_log_dir
writer.setup_event_writer(config.is_wandb_enabled(), config.is_tensorboard_enabled(), log_dir=writer_log_path, wandb_name=config.wandb_name)
writer.setup_event_writer(
config.is_wandb_enabled(),
config.is_tensorboard_enabled(),
log_dir=writer_log_path,
wandb_name=config.wandb_name,
)
writer.setup_local_writer(config.logging, max_iter=config.max_num_iterations, banner_messages=banner_messages)
writer.put_config(name="config", config_dict=dataclasses.asdict(config), step=0)
profiler.setup_profiler(config.logging)