A question when i run train.py

Song-Jingyu / PointPainting

This repository is an open-source PointPainting package which is easy to understand, deploy and run!

MIT License

238 stars 45 forks source link

A question when i run train.py #16

Closed xing3507 closed 2 years ago

xing3507 commented 2 years ago

when i run python train.py --cfg_file cfgs/kitti_models/pointpillar_painted.yaml

it says that :

Traceback (most recent call last): File "train.py", line 198, in main() File "train.py", line 112, in main total_epochs=args.epochs File "/media/user/date/ZXH/PointPainting-main/detector/tools/pcdet/datasets/init.py", line 46, in build_dataloader dataset = all[dataset_cfg.DATASET]( KeyError: 'PaintedKittiDataset'

How can i fix it?? @Song-Jingyu 谢谢！！！

Song-Jingyu commented 2 years ago

I am not sure why this happen. But I would suggest remove and uninstall and re-download this repo to see if this would help.