HaochengWan / PVT

PVT: Point-Voxel Transformer for 3D Deep Learning
https://arxiv.org/abs/2108.06076
MIT License
87 stars 8 forks source link

got a KeyError, need help plz #3

Closed kellen5l closed 2 years ago

kellen5l commented 2 years ago

Hi, I use test command python main_cls.py --exp_name=cls --num_points=1024 --use_sgd=True --eval=True --model_path=checkpoints/cls/model.t7 --test_batch_size 32, but got a KeyError.Is the problem caused by different dataset files? need your help plz. 你好,我运行测试命令python main_cls.py --exp_name=cls --num_points=1024 --use_sgd=True --eval=True --model_path=checkpoints/cls/model.t7 --test_batch_size 32,但遇到了 KeyError,是数据集文件不同导致的问题吗?需要您的帮助。 Error :

...
File ".../PVT-master/data.py", line 71, in _get_item
    cls = self.classes[self.datapath[index][0]]
KeyError: 'data/modelnet40_normal_resampled/plydata'

content in modelnet40_test.txt:

data/modelnet40_normal_resampled/ply_data_test0.h5
data/modelnet40_normal_resampled/ply_data_test1.h5