IanTaehoonYoo / semantic-segmentation-pytorch

Pytorch implementation of FCN, UNet, PSPNet, and various encoder models.
MIT License
86 stars 19 forks source link

How can I make a "result image" from trained model ? #1

Closed bemoregt closed 4 years ago

bemoregt commented 4 years ago

Hi, @IanTaehoonYoo

How can I make a "result image" from trained model ? I can't found that code...

Thanks in advance.

Best, @bemoregt.

IanTaehoonYoo commented 4 years ago

Hi, @bemoregt

You can find the code in the Trainer class, and the line is 160. Also, the result images will be saved in the runs folder if you run this project.

If you have any questions, please let me know.

Thank you, Best Regards,

@IanTaehoonYoo