矩阵转换图片

原创

于 2018-07-29 16:21:47 发布 · 2.2k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#matTojpg

时间：20180723-20180729

背景：基于上一篇tvnet博客里所输出的mat格式的图片转换为jpg或者其他格式的图片本文以jpg为例

import cv2
import scipy.io as scio
from PIL import Image
import numpy as np
import matplotlib.pyplot as plt

data = scio.loadmat('result.mat')
a=data['flow']
def MatrixToImage(data):
    data = data*255
    new_im = Image.fromarray(data.astype(np.uint8))
    return new_im

new_im = MatrixToImage(a)
for i in range(len(a[0][0])):
    plt.imshow(a[:][:][i],cmap=plt.cm.gray, interpolation='nearest')
    new_im.show()
    new_im=new_im.convert('RGB')
    new_im.save(str.format("a%d.jpg"%i)) # 保存图片

也可以在未生成mat格式的图片前直接在tvnetdemo里输出图片，将前面的demo改为demo2

import os
import cv2
import numpy as np
import tensorflow as tf
import scipy.io as sio
import matplotlib.pyplot as plt
from tvnet import TVNet

from PIL import Image

flags = tf.app.flags
flags.DEFINE_integer("scale", 5, " TVNet scale [3]")
flags.DEFINE_integer