help

[WS517]

@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]

Could you print the whole error log?

[WS517]

@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'

Looking forward to your reply，thank you.