search

LeeJunHyun / Image_Segmentation

Pytorch implementation of U-Net, R2U-Net, Attention U-Net, and Attention R2U-Net.
2.68k stars 595 forks source link

predict image #15

Closed vcvishal closed 5 years ago

vcvishal commented 5 years ago

how to get predicted segmented image from test data

LeeJunHyun commented 5 years ago

Hi, @vcvishal . There are code to obtain result images for validation data. You can modify this for test data.

https://github.com/LeeJunHyun/Image_Segmentation/blob/master/solver.py#L231

vcvishal commented 5 years ago

thank you

LeeJunHyun commented 5 years ago

If you don't have further issues, I will close the issue. I hope my answer was helpful :)

mjiangtao commented 5 months ago

每个纪元一个图像。有没有办法使用经过训练的模型来预测多个图像？

Do you solve the problem? Can you help me? Thank you.