dvlab-research / 3DSSD

3DSSD: Point-based 3D Single Stage Object Detector (CVPR 2020)
MIT License
376 stars 68 forks source link

Error when preprocess test dataset #25

Open KaiWU5 opened 4 years ago

KaiWU5 commented 4 years ago

When I use the command "python lib/core/data_preprocessor.py --cfg configs/kitti/3dssd/3dssd.yaml --split testing --img_list test", I got the following error. AttributeError: 'KittiDataset' object has no attribute 'mixup_db_class'

With debugging, I found that mixup_db_class only will be activated by ['train', 'val', 'trainval'] modes. So May I ask for how to fix this and what is the intension of using mixup_db_class? Code reading is painful, please help out.

zhufeng888 commented 3 years ago

I have the same issue.

zhufeng888 commented 3 years ago

I have solved the problem to add if self.img_list in ['train', 'val', 'trainval']: before mixup_label_dict = dict([(cls, []) for cls in self.mixup_db_class])