search

VinAIResearch / ISBNet

ISBNet: a 3D Point Cloud Instance Segmentation Network with Instance-aware Sampling and Box-aware Dynamic Convolution (CVPR 2023)
Apache License 2.0
104 stars 22 forks source link

help #57

Open WS517 opened 1 month ago

WS517 commented 1 month ago

@ngoductuanlhp ngoductuanlhp Hello Professor, I have trained and got the model, my input raw data contains xyzrgb semantic labels and instance labels, now I want to test it on unlabeled data, I ran the test.py program and reported the following error: Traceback (most recent call last). File “/home/wang_yongjun/zl/Code/ISBNet/tools/test.py”, line 200, in main() File “/home/wang_yongjun/zl/Code/ISBNet/tools/test.py”, line 160, in main sem_preds.append(res[“semantic_preds”]) Can you please tell me how I can solve this problem or what I should do for testing, looking forward to your reply, thanks a lot!

ngoductuanlhp commented 1 month ago

Could you print the whole error log?

WS517 commented 1 month ago

@ngoductuanlhp Thank you for your reply, the specific error is as follows: (isbnet) wang_yongjun@wang-yongjun:~$ /home/wang_yongjun/anaconda3/envs/isbnet/bin/python /home/wang_yongjun/zl/Code/ISBNet/tools/test.py 2024-05-31 22:22:17,293 - INFO - Load state dict from /home/wang_yongjun/zl/Code/ISBNet/stpls3d/isbnet_stpls3d/ws/best.pth 2024-05-31 22:22:17,487 - INFO - unexpected key in source state_dict: criterion.empty_weight 2024-05-31 22:22:17,488 - INFO - Load test dataset: 1 scans 2024-05-31 22:22:20,144 - INFO - Infer scene 1/1 Traceback (most recent call last): File "/home/wang_yongjun/zl/Code/ISBNet/tools/test.py", line 200, in main() File "/home/wang_yongjun/zl/Code/ISBNet/tools/test.py", line 160, in main sem_preds.append(res["semantic_preds"]) KeyError: 'semantic_preds' PixPin_2024-05-31_22-25-33 Looking forward to your reply，thank you.