open-mmlab / mmsegmentation

OpenMMLab Semantic Segmentation Toolbox and Benchmark.
https://mmsegmentation.readthedocs.io/en/main/
Apache License 2.0
8.21k stars 2.6k forks source link

How to save the "npy" segmentation results ? #2391

Open Yingyue-L opened 1 year ago

xiexinch commented 1 year ago

Hi @LYY-HH, Do you mean the segmentation masks? You might add argument --format-only when running with tools/test.py

https://mmsegmentation.readthedocs.io/en/latest/inference.html#test-a-dataset

https://github.com/open-mmlab/mmsegmentation/blob/master/tools/test.py#L35

Yingyue-L commented 1 year ago

Hi @LYY-HH, Do you mean the segmentation masks? You might add argument --format-only when running with tools/test.py

https://mmsegmentation.readthedocs.io/en/latest/inference.html#test-a-dataset

https://github.com/open-mmlab/mmsegmentation/blob/master/tools/test.py#L35

Thank for your reply. I'am using the 1.x branch which don't have the --format-only on the tools/test.py. Futhermore, I want the results that before the argmax which have the class channel.

switch959 commented 1 year ago

请问您解决这个问题了吗