WHU-USI3DV / Mobile-Seed

[IEEE RAL'24 & IROS'24] Mobile-Seed: Joint Semantic Segmentation and Boundary Detection for Mobile Robots
https://whu-usi3dv.github.io/Mobile-Seed/
BSD 2-Clause "Simplified" License
118 stars 2 forks source link

Question about evaluating the pre-trained model for cityscapes dataset #9

Closed Chenneyi closed 2 months ago

Chenneyi commented 3 months ago

Hello, when I evaluated the pre-trained model for cityscapes dataset, using the command: bash tools/dist_test.sh ./configs/Mobile_Seed/MS_tiny_cityscapes.py ckpt/MS_tiny_cityscapes.pth 1 --eval mIoU, the error occurred as follows: image Have you ever come across this problem? How did you solve it?

martin-liao commented 3 months ago

Thank you for reporting this error! I have fixed the code for multi-gpu test, please pull the latest code and try again! tip: if you are testing with a single GPU, you can also change the --launcher pytorch to --launcher none for single-GPU testing.

martin-liao commented 3 months ago

The evaluation results on the Cityscapes dataset should be close to this: Screenshot from 2024-06-08 10-40-46

Chenneyi commented 3 months ago

Thank you for reporting this error! I have fixed the code for multi-gpu test, please pull the latest code and try again! tip: if you are testing with a single GPU, you can also change the --launcher pytorch to --launcher none for single-GPU testing.

That works, thank you so much!!!!!!!!

martin-liao commented 2 months ago

Since the problem has been resolved, I will close the issue.