caffe - 如何保存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