如何保存segnet的分割结果

Question

scipy.misc.toimage(rgb, cmin=0.0, cmax=255).save(IMAGE_FILE+'_segnet.png')

在文件test_segmentation_camvid中，我应该如何定义IMAGE_FILE并保存预测结果？

Answer 1

IMAGE_FILE只是一个变量，它可以是您可以执行的保存图像的名称

IMAGE_FILE="new_image"
scipy.misc.toimage(rgb, cmin=0.0, cmax=255).save(IMAGE_FILE+'_segnet.png')

并且将保存的图像将以new_image_segnet.png的名称命名