How to test single image
JasonChenhx opened this issue · 4 comments
JasonChenhx commented
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.
JasonChenhx commented
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.
JasonChenhx commented
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.