[已解决]如何读取文件夹里面的全部图片

飘香一剑

请问，如何用python读取文件夹里面的全部图片。然后储存到数组里面

最佳答案

月排行榜 / 总排行榜

My_A

2019-4-2 14:23:28

1. import os
   
2. 
   
3. path = "X:\\XXXXXXXXX"
   
4. file_list = []
   
5. 
   
6. def search(path):
   
7.     lst = os.listdir()
   
8.     for f in lst:
   
9.         f = os.path.join(path,f)
   
10.         if  os.path.isdir(f):
    
11.             search(f)
    
12.         elif  os.path.isfile(f)
    
13.             suffix = ['.jpg','.png','.bmp']
    
14.             for s in suffix:
    
15.                 if f.endwith(s):
    
16.                     file_list.append(f)

复制代码

跳转到最佳答案楼层

My_A

1. import os
   
2. 
   
3. path = "X:\\XXXXXXXXX"
   
4. file_list = []
   
5. 
   
6. def search(path):
   
7.     lst = os.listdir()
   
8.     for f in lst:
   
9.         f = os.path.join(path,f)
   
10.         if  os.path.isdir(f):
    
11.             search(f)
    
12.         elif  os.path.isfile(f)
    
13.             suffix = ['.jpg','.png','.bmp']
    
14.             for s in suffix:
    
15.                 if f.endwith(s):
    
16.                     file_list.append(f)

复制代码

wp231957

二进制读取到内存？？？

wongyusing

没记错的话，可以用os的isdir()方法,
定位到目标目录后用for循环遍历就可以生成列表了。

不过最好还是获取目录下的文件名就行了。
操作的时候再读取

飘香一剑

wongyusing 发表于 2019-4-2 14:31
没记错的话，可以用os的isdir()方法,
定位到目标目录后用for循环遍历就可以生成列表了。

我想全部读取出来，存储到矩阵里面。

塔利班

构造文件队列，阅读器，读取
你可以去百度下tensorflow的操作，b站也有视频

My_A

最后再  search() 就行了

飘香一剑

My_A 发表于 2019-4-2 15:07
最后再  search() 就行了

最后得到的是一个空列表

苏绛雪

import os
import cv2
from itertools import chain

array_of_img = [] # this if for store all of the image data
# this function is for read image,the input is directory name
def read_directory(directory_name):
    # this loop is for read each image in this foder,directory_name is the foder name with images.
    for filename in os.listdir(r"./"+directory_name):
        #print(filename) #just for test
        #img is used to store the image data
        img = cv2.imread(directory_name + "/" + filename)
        img=list(chain(*img))
        array_of_img.append(img)
     
        #print(img)
        #print(array_of_img)
        
        
read_directory("Images")

My_A

飘香一剑 发表于 2019-4-2 16:20
最后得到的是一个空列表

把path 的内容换为你要的文件夹的路径
把suffix 列表的内容换为你搜索的图片的后缀名

最后在 search(path)

My_A

如果要把图片中的二进制内容保存在列表中则可以加上：
from PIL import Image

pic_data = {}

for f in file_list:
        pic_data[f] = Image.open(f)