Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

KeyError about run traiu.py script #19

Open
Linkersem opened this issue May 21, 2024 · 0 comments
Open

KeyError about run traiu.py script #19

Linkersem opened this issue May 21, 2024 · 0 comments

Comments

@Linkersem
Copy link

Hi , i try to run train.py script on my own dataset,but i meet a error like this,what should i do ?

[Config] merge from parent file: config/example/test/dataset.yml
[Config] merge from parent file: config/example/test/level_of_gaussian.yml
[Config] merge from parent file: config/example/test/stage_8_4.yml
Key is not in the template: split
[Config] replace key $root
[Config] replace key $scale3d
[Config] replace key $root
[Config] replace key $scale3d
[Config] replace key $PLYNAME
[Config] replace key $PLYNAME
[Config] replace key $PLYNAME
[Config] replace key $PLYNAME
[Config] replace key $xyz_scale
[Config] replace key $scale3d
[Config] replace key $PLYNAME
[Config] replace key $xyz_scale
[Config] replace key $max_steps
[Config] replace key $dataset
[Config] replace key $RGB_RENDER_L1_SSIM
[Config] replace key $NAIVE_STAGE
[Config] replace key $val_dataset
Using GPUs: 0
Write to output/example/test/log
Using /home/guoyl16/.cache/torch_extensions/py310_cu118 as PyTorch extensions root...
Detected CUDA files, patching ldflags
Emitting ninja build file /home/guoyl16/.cache/torch_extensions/py310_cu118/compute_radius/build.ninja...
Building extension module compute_radius...
Allowing ninja to set a default number of workers... (overridable by setting the environment variable MAX_JOBS=N)
ninja: no work to do.
Loading extension module compute_radius...
[/mnt/data/guoyl16/DATA/DATA_GS/object_pc_240307/sparse/0/sparse.npz] mean: -0.214, 0.565, -7.522
[/mnt/data/guoyl16/DATA/DATA_GS/object_pc_240307/sparse/0/sparse.npz] std: 8.602, 6.323, 15.863
[/mnt/data/guoyl16/DATA/DATA_GS/object_pc_240307/sparse/0/sparse.npz] sigma=1 15333/16862
bounds: [[-8.816, -5.758, -23.384], [8.388, 6.888, 8.341]]
[/mnt/data/guoyl16/DATA/DATA_GS/object_pc_240307/sparse/0/sparse.npz] sigma=2 15602/16862
bounds: [[-17.417, -12.081, -39.247], [16.990, 13.211, 24.204]]
[/mnt/data/guoyl16/DATA/DATA_GS/object_pc_240307/sparse/0/sparse.npz] sigma=3 15903/16862
bounds: [[-26.019, -18.404, -55.110], [25.591, 19.534, 40.067]]
[/mnt/data/guoyl16/DATA/DATA_GS/object_pc_240307/sparse/0/sparse.npz] z_min: -256.574, z_max: -2.182
[Load PLY] load from ply: /mnt/data/guoyl16/DATA/DATA_GS/object_pc_240307/sparse/0/sparse.npz
[Load PLY] min: [ -86.48607941 -8.61126238 -256.57448468], max: [ 41.38941455 117.52840973 -2.18179324]
[Load PLY] scale: 0.0050, 46.1572, mean = 0.2368
[GaussianPoint] scales: [0.00500.236846.1572]
[GaussianPoint] -> scales: [0.05920.13540.9473]

Code 211 files has been copied to output/example/test/log/code_backup_20240521-105923
[ImageDataset] set scales: [1, 2, 4, 8], crop size: [-1, -1]
[ImageDataset] cache dir: /mnt/data/guoyl16/DATA/DATA_GS/object_pc_240307/cache
[ImageDataset] offset: [ 0.06529621 0.01664672 -0.02070021], radius: 7.258950956096211
[ImageDataset] init dataset with 71 images
Base iteration: 100
[ImageDataset] set scales: [4], crop size: [-1, -1]
[ImageDataset] cache dir: /mnt/data/guoyl16/DATA/DATA_GS/object_pc_240307/cache_val
Loaded 71 cameras from /mnt/data/guoyl16/DATA/DATA_GS/object_pc_240307/sparse/0
y/8y01073
Traceback (most recent call last):
File "/mnt/data/guoyl16/project/LoG/apps/train.py", line 183, in
main()
File "/mnt/data/guoyl16/project/LoG/apps/train.py", line 163, in main
trainer = Trainer(cfg, model, renderer, logdir=outdir)
File "/mnt/data/guoyl16/project/LoG/LoG/utils/trainer.py", line 56, in init
self.check_val()
File "/mnt/data/guoyl16/project/LoG/LoG/utils/trainer.py", line 65, in check_val
dataset = load_object(self.cfg.val.dataset.module, self.cfg.val.dataset.args)
File "/mnt/data/guoyl16/project/LoG/LoG/utils/config.py", line 61, in load_object
obj = getattr(module, name)(**extra_args, **module_args)
File "/mnt/data/guoyl16/project/LoG/LoG/dataset/colmap.py", line 129, in init
cameras = self.check_cameras(scale3d=scale3d, scale_camera_K=scale_camera_K)
File "/mnt/data/guoyl16/project/LoG/LoG/dataset/image_base.py", line 73, in check_cameras
cameras_new[name] = cameras[name]
KeyError: 'y/8y01073'

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant