zhuoxiao-chen / ReDB-DA-3Ddet

[ICCV 2023] Revisiting Domain-Adaptive 3D Object Detection by Reliable, Diverse and Class-balanced Pseudo-Labeling
Apache License 2.0
46 stars 8 forks source link

Question about SAMPLE_CLASSES: [ 'Vehicle' ] #7

Open fdy61 opened 2 months ago

fdy61 commented 2 months ago

When I run the command of self-training, an error occured in https://github.com/zhuoxiao-chen/ReDB-DA-3Ddet/blob/7d24bd0f1d0ea14efbaf539c4717ffd28cd30ef8/pcdet/utils/self_training_utils.py#L274,**KeyError: 'Vehicle'**,my command is: bash scripts/dist_train.sh 6 --cfg_file /home/fudongyi/ReDB/ReDB-DA-3Ddet/tools/cfgs/da-nuscenes-kitti_models/secondiou_ReDB/secondiou_ReDB.yaml --pretrained_model /disk1/Fudongyi/ReDB_DA/ckpt/checkpoint_epoch_50.pth --batch_size 64 I have noticed SAMPLE_CLASSES: [ 'Vehicle' ] in secondiou_ReDB.yaml. Is there any mistake within it?