PJLab-ADG
Error when using TTA strategy for test: box and score lengths are not equal
When I run test using the TTA strategy I get an error:
python test.py --cfg_file cfgs/det_model_cfgs/centerpoint_1sweep.yaml --ckpt <PATH_TO_CKPT> --set DATA_CONFIG.TTA True
Exception has occurred: ValueError
Length of boxes not equal to length of scores.
File "/workspace/DetZero/detection/detzero_det/utils/ensemble_utils/wbf_3d.py", line 16, in prefilter_boxes
raise ValueError("Length of boxes not equal to length of scores.")
File "/workspace/DetZero/detection/detzero_det/utils/ensemble_utils/wbf_3d.py", line 151, in weighted_boxes_fusion_3d
filtered_boxes = prefilter_boxes(boxes_list, scores_list, labels_list, weights, skip_box_thr)
File "/workspace/DetZero/detection/detzero_det/utils/ensemble_utils/ensemble.py", line 17, in wbf_online
boxes, scores, labels = weighted_boxes_fusion_3d(
File "/workspace/DetZero/detection/detzero_det/models/centerpoint.py", line 207, in test_time_augment
boxes, scores, labels = wbf_online(boxes, scores, labels)
File "/workspace/DetZero/detection/detzero_det/models/centerpoint.py", line 299, in post_processing
final_boxes, final_scores, final_labels = self.test_time_augment(batch_dict, pred_dicts)
File "/workspace/DetZero/detection/detzero_det/models/centerpoint.py", line 38, in forward
pred_dicts, recall_dicts = self.post_processing(batch_dict)
File "/workspace/DetZero/detection/tools/eval_utils.py", line 73, in eval_one_epoch
result = model(batch_dict)
File "/workspace/DetZero/detection/tools/test.py", line 68, in eval_single_ckpt
eval_utils.eval_one_epoch(
File "/workspace/DetZero/detection/tools/test.py", line 203, in main
eval_single_ckpt(model, test_loader, args, eval_output_dir, logger, epoch_id, dist_test=dist_test)
File "/workspace/DetZero/detection/tools/test.py", line 207, in
You can refer to this link： https://github.com/PJLab-ADG/DetZero/issues/34#issuecomment-1800922639
