bigmb/Unet-Segmentation-Pytorch-Nest-of-Unets

How to test single image

JasonChenhx opened this issue · 4 comments

I want to know the code of how to test single image.
Thanks

bigmb commented

You mean after the weights are trained?
I couldn't get your question properly.

You mean after the weights are trained?
I couldn't get your question properly.

I mean that I have got the model.pth after train, I want to test single image to see the performance of the result.
Thanks

bigmb commented

During the training its saved in the pred folder.

And after training if you just want to test a single file again. Then you will have to call line 408 (model loading ) and then set it to eval (line 412) .
After that follow the commands given in the line 290-306.