-
Notifications
You must be signed in to change notification settings - Fork 99
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
A error when running inference.py #129
Comments
我也遇到了类似的问题,请问您解决了吗,如果解决了可以告诉我是如何解决的吗? |
I have successfully train and test second_early_fusion.yaml and second_intermediate_fusion.yaml. However,I meet this error when I test config of second_late_fusion.
File "/graduation-project/OpenCOOD/opencood/data_utils/post_processor/voxel_postprocessor.py", line 263, in post_process
assert cav_id in output_dict
How could I solve this error? Any advice would be greatly appreciated!
(opencood) root@2406a1b90c16:/graduation-project/OpenCOOD# python opencood/tools/inference.py --model_dir /graduation-project/OpenCOOD/opencood/logs/second_late_fusion_low_res_2024_04_04_07_28_25 --fusion_method intermediate
Dataset Building
550 samples found.
/root/anaconda3/envs/opencood/lib/python3.7/site-packages/torch/utils/data/dataloader.py:560: UserWarning: This DataLoader will create 16 worker processes in total. Our suggested max number of worker in current system is 8, which is smaller than what this DataLoader is going to create. Please be aware that excessive worker creation might get DataLoader running slow or even freeze, lower the worker number to avoid potential slowness/freeze if necessary.
cpuset_checked))
Creating Model
Loading Model from checkpoint
resuming by loading epoch 40
0it [00:00, ?it/s]/root/anaconda3/envs/opencood/lib/python3.7/site-packages/torch/nn/functional.py:1960: UserWarning: nn.functional.sigmoid is deprecated. Use torch.sigmoid instead.
warnings.warn("nn.functional.sigmoid is deprecated. Use torch.sigmoid instead.")
0it [00:05, ?it/s]
Traceback (most recent call last):
File "opencood/tools/inference.py", line 208, in
main()
File "opencood/tools/inference.py", line 121, in main
opencood_dataset)
File "/graduation-project/OpenCOOD/opencood/tools/inference_utils.py", line 90, in inference_intermediate_fusion
return inference_early_fusion(batch_data, model, dataset)
File "/graduation-project/OpenCOOD/opencood/tools/inference_utils.py", line 68, in inference_early_fusion
output_dict)
File "/graduation-project/OpenCOOD/opencood/data_utils/datasets/late_fusion_dataset.py", line 270, in post_process
self.post_processor.post_process(data_dict, output_dict)
File "/graduation-project/OpenCOOD/opencood/data_utils/post_processor/voxel_postprocessor.py", line 263, in post_process
assert cav_id in output_dict
AssertionError
The text was updated successfully, but these errors were encountered: