search

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

Implementation of different kinds of Unet Models for Image Segmentation - Unet , RCNN-Unet, Attention Unet, RCNN-Attention Unet, Nested Unet
MIT License
1.87k stars 345 forks source link

How to test single image #25

Closed JasonChenhx closed 4 years ago

JasonChenhx commented 4 years ago

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

bigmb commented 4 years ago

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

JasonChenhx commented 4 years ago

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 4 years ago

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.

JasonChenhx commented 4 years ago

Thank you very much! Your answer is of great help to me! Thanks again!

------------------ 原始邮件 ------------------ 发件人: "Malav Bateriwala"<notifications@github.com>; 发送时间: 2020年1月10日(星期五) 下午2:54 收件人: "bigmb/Unet-Segmentation-Pytorch-Nest-of-Unets"<Unet-Segmentation-Pytorch-Nest-of-Unets@noreply.github.com>; 抄送: "JasonChenhx"<chx880324@163.com>;"Author"<author@noreply.github.com>; 主题: Re: [bigmb/Unet-Segmentation-Pytorch-Nest-of-Unets] How to test single image (#25)

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.

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.