Open Kaustubh-cpu opened 4 months ago
I also had the same issue. Could you please tell @Kaustubh-cpu if you resolved this?
Perhaps you could try applying min-max normalization before saving the figure. This will set the minimum value to zero and the maximum value to one.
Like here: https://github.com/qianyu-dlut/MVANet/blob/ff270a6682c9b5bf3ff73c588b0ef0291de49fed/test.py#L50
Please make sure you are using the trained model weights from section III. Testing, rather than the pretrained model from section II. Training. The pretrained model is the base SwinTransformer model which has not been trained for this DIS task.
@qianyu-dlut
can you please suggest why i am getting the result in the below image when i am running the predict.py