def saveAllImages(rgb_array, directory, foldername):
"""
Saves unfiltered and filtered images to a file directory
:param rgb_array: image rgb array
:param directory: image directory name
:param foldername: name of the folder that will contain saved files
:return:
"""
rgb_array_red = rgb_array * 1
r_array = setImageColor(rgb_array_red, 'r')
rgb_array_green = rgb_array * 1
g_array = setImageColor(rgb_array_green, 'g')
rgb_array_blue = rgb_array * 1
b_array = setImageColor(rgb_array_blue, 'b')
trueimage = 'image.png'
redimage = 'red.png'
greenimage = 'green.png'
blueimage = 'blue.png'
plotimage = 'plot.png'
if not os.path.exists(os.path.join(directory, foldername)):
os.mkdir(os.path.join(directory, foldername))
misc.imsave(os.path.join(directory, foldername, trueimage), rgb_array)
misc.imsave(os.path.join(directory, foldername, redimage), r_array)
misc.imsave(os.path.join(directory, foldername, greenimage), g_array)
misc.imsave(os.path.join(directory, foldername, blueimage), b_array)
savePlot(rgb_array, os.path.join(directory, foldername, plotimage))
评论列表
文章目录