open-mmlab / mmsegmentation

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

Prediction inside mmsegmentation is different than model->forward #3699

Open ymutairi opened 3 weeks ago

ymutairi commented 3 weeks ago

Hi all,

I noticed if we use the mmsegmentation demo code to predict on image using for example psnet is different from if we just write a standard code to load the model and predict as the output size is downsized by 8 and resize function doesn’t simulate the output from mmsegmentation.

I do understand there is a preprocessing and post processing steps in mmsegmentation inference.

can we obtain similar result using libtorch for example?