Description
Hi, I have been facing this issue, i got the model trained but after that i am facing this.
CUDA_VISIBLE_DEVICES="$GPU" python fit.py --conf confs/train_person_1.conf \
--batch_size 1 --train_rotation True --fixed_images True \ --save_path $DATA_PATH/flame_fitting/$EXP_NAME_1/stage_1 \ --data_path $DATA_PATH \ --fitted_camera_path $EXP_PATH_1/cameras/30000_matrices.pkl
creating the SMPLX Decoder
Overwriting COLMAP cameras with optimized cameras
Traceback (most recent call last):
File "/home/lab/Desktop/GaussianHaircut/ext/NeuralHaircut/src/multiview_optimization/fit.py", line 82, in
main(
File "/home/lab/Desktop/GaussianHaircut/ext/NeuralHaircut/src/multiview_optimization/fit.py", line 24, in main
dataset = Multiview_dataset(data_path, fitted_camera_path, use_scale, fixed_images, **conf['dataset'], device=conf['general.device'], batch_size=batch_size)
File "/home/lab/Desktop/GaussianHaircut/ext/NeuralHaircut/src/multiview_optimization/dataset/dataset.py", line 162, in init
self.lmks = torch.from_numpy(np.stack([lmks[i] for i in self.good_views]))
File "<array_function internals>", line 200, in stack
File "/home/lab/miniconda3/envs/gaussian_splatting_hair/lib/python3.9/site-packages/numpy/core/shape_base.py", line 460, in stack
raise ValueError('need at least one array to stack')
ValueError: need at least one array to stack