Open ZhaoLongjiea opened 1 month ago
when I start to train this model and print "bash script/train.sh bear 1", then it occurs:
(3DSegment) lzha0538@gpu1-3w-1:~/segment/gaussian-grouping$ bash script/train.sh bear 1 Optimizing output/bear Output folder: output/bear [31/07 17:19:21] Reading camera 96/96 [31/07 17:19:21] Loading Training Cameras [31/07 17:19:21] Loading Test Cameras [31/07 17:19:23] Number of points at initialisation : 63659 [31/07 17:19:23] Num classes: 256 [31/07 17:19:23] Training progress: 0%| | 0/30000 [00:00<?, ?it/s]Traceback (most recent call last): File "train.py", line 255, in training(lp.extract(args), op.extract(args), pp.extract(args), args.test_iterations, args.save_iterations, args.checkpoint_iterations, args.start_checkpoint, args.debug_from, args.use_wandb) File "train.py", line 83, in training render_pkg = render(viewpoint_cam, gaussians, pipe, background) File "/home/lzha0538/segment/gaussian-grouping/gaussian_renderer/init.py", line 95, in render cov3D_precomp = cov3D_precomp) File "/home/lzha0538/miniconda3/envs/3DSegment/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl return forward_call(*input, **kwargs) TypeError: forward() got an unexpected keyword argument 'sh_objs' Training progress: 0%| | 0/30000 [00:00<?, ?it/s] Looking for config file in output/bear/cfg_args Config file found: output/bear/cfg_args Rendering output/bear Traceback (most recent call last): File "render.py", line 168, in render_sets(model.extract(args), args.iteration, pipeline.extract(args), args.skip_train, args.skip_test) File "render.py", line 135, in render_sets scene = Scene(dataset, gaussians, load_iteration=iteration, shuffle=False) File "/home/lzha0538/segment/gaussian-grouping/scene/init.py", line 35, in init self.loaded_iter = searchForMaxIteration(os.path.join(self.model_path, "point_cloud")) File "/home/lzha0538/segment/gaussian-grouping/utils/system_utils.py", line 28, in searchForMaxIteration return max(saved_iters) ValueError: max() arg is an empty sequence
I found that the object_mask directory is essential for training. When I provide this directory, the error no longer occurs.
when I start to train this model and print "bash script/train.sh bear 1", then it occurs:
(3DSegment) lzha0538@gpu1-3w-1:~/segment/gaussian-grouping$ bash script/train.sh bear 1 Optimizing output/bear Output folder: output/bear [31/07 17:19:21] Reading camera 96/96 [31/07 17:19:21] Loading Training Cameras [31/07 17:19:21] Loading Test Cameras [31/07 17:19:23] Number of points at initialisation : 63659 [31/07 17:19:23] Num classes: 256 [31/07 17:19:23] Training progress: 0%| | 0/30000 [00:00<?, ?it/s]Traceback (most recent call last): File "train.py", line 255, in
training(lp.extract(args), op.extract(args), pp.extract(args), args.test_iterations, args.save_iterations, args.checkpoint_iterations, args.start_checkpoint, args.debug_from, args.use_wandb)
File "train.py", line 83, in training
render_pkg = render(viewpoint_cam, gaussians, pipe, background)
File "/home/lzha0538/segment/gaussian-grouping/gaussian_renderer/init.py", line 95, in render
cov3D_precomp = cov3D_precomp)
File "/home/lzha0538/miniconda3/envs/3DSegment/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl
return forward_call(*input, **kwargs)
TypeError: forward() got an unexpected keyword argument 'sh_objs'
Training progress: 0%| | 0/30000 [00:00<?, ?it/s]
Looking for config file in output/bear/cfg_args
Config file found: output/bear/cfg_args
Rendering output/bear
Traceback (most recent call last):
File "render.py", line 168, in
render_sets(model.extract(args), args.iteration, pipeline.extract(args), args.skip_train, args.skip_test)
File "render.py", line 135, in render_sets
scene = Scene(dataset, gaussians, load_iteration=iteration, shuffle=False)
File "/home/lzha0538/segment/gaussian-grouping/scene/init.py", line 35, in init
self.loaded_iter = searchForMaxIteration(os.path.join(self.model_path, "point_cloud"))
File "/home/lzha0538/segment/gaussian-grouping/utils/system_utils.py", line 28, in searchForMaxIteration
return max(saved_iters)
ValueError: max() arg is an empty sequence