用python将文件夹中多个图片转化为一维数组

刚开始学习lstm,要用lstm做图像预测,考虑到lstm的结构,初步想先把图片转成一维的数组,在输入到lstm网络中训练。琢磨了好几天终于是转成了

 

试了很多办法, io模块读取文件夹下多个图片很方便,但是好像不能分离三通道(我没查到,不知道到底行不行)。opencv分离三通道方便,但是读取多个图片感觉很麻烦。我就把两个结合了一下。

 

下面是代码,朋友们有更好的方法的话,欢迎交流哈哈。

 

用python将文件夹中多个图片转化为一维数组

用python将文件夹中多个图片转化为一维数组