qianyu-dlut / MVANet

MIT License
109 stars 11 forks source link

Error in prediction #6

Open Kaustubh-cpu opened 4 months ago

Kaustubh-cpu commented 4 months ago

@qianyu-dlut

can you please suggest why i am getting the result in the below image when i am running the predict.py

4#Architecture#11#Pipe#7124946891_03a3a2a283_o

charchit7 commented 4 months ago

I also had the same issue. Could you please tell @Kaustubh-cpu if you resolved this?

come880412 commented 3 months ago

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

VMinB12 commented 1 month ago

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.