读取目录中的 JPG 文件？

Question

我正在尝试浏览目录中的不同图像文件。我也在使用 Jupyter 运行我的 python 代码。但是我不断收到此错误。下面是我的代码和我收到的错误。

代码：

import os
import os.path
for img in os.listdir('test_images'):
    if img.endswith("jpg"):
        scriptpath = os.path.dirname(img)
        print(os.path.join('test_images', img))
        # Read in the image
        image = os.path.join(scriptpath, img)
        image = mpimg.imread(image)
        # Grab the x and y size and make a copy of the image
        ysize = image.shape[0]
        xsize = image.shape[1]
        color_select = np.copy(image)
        # Define color selection criteria 
        red_threshold = 200
        green_threshold = 200
        blue_threshold = 200

        rgb_threshold = [red_threshold, green_threshold, blue_threshold]

        # Do a boolean or with the "|" character to identify
        # pixels below the thresholds
        thresholds = (image[:,:,0] < rgb_threshold[0]) \
                    | (image[:,:,1] < rgb_threshold[1]) \
                    | (image[:,:,2] < rgb_threshold[2])
        color_select[thresholds] = [red_threshold,green_threshold,blue_threshold]
        plt.imshow(color_select)
        # Display the image                 
        plt.imshow(color_select)
        continue
    else:
        continue

输出：

test_images/solidWhiteCurve.jpg

错误：

FileNotFoundError                         Traceback (most recent call last)
<ipython-input-3-6edf7c0860b7> in <module>()
      7         # Read in the image
      8         image = os.path.join(scriptpath, img)
----> 9         image = mpimg.imread(image)
     10         # Grab the x and y size and make a copy of the image
     11         ysize = image.shape[0]

/Users/steveburgos/anaconda/envs/carnd-term1/lib/python3.5/site-packages/matplotlib/image.py in imread(fname, format)
   1225 
   1226     if ext not in handlers:
-> 1227         im = pilread(fname)
   1228         if im is None:
   1229             raise ValueError('Only know how to handle extensions: %s; '

/Users/steveburgos/anaconda/envs/carnd-term1/lib/python3.5/site-packages/matplotlib/image.py in pilread(fname)
   1203         except ImportError:
   1204             return None
-> 1205         with Image.open(fname) as image:
   1206             return pil_to_array(image)
   1207 

/Users/steveburgos/anaconda/envs/carnd-term1/lib/python3.5/site-packages/PIL/Image.py in open(fp, mode)
   2310 
   2311     if filename:
-> 2312         fp = builtins.open(filename, "rb")
   2313 
   2314     try:

FileNotFoundError: [Errno 2] No such file or directory: 'solidWhiteCurve.jpg'

Answer 1

您的代码中有路径不匹配，您的错误清楚地表明了这一点（找不到文件）。当你这样做时：

for img in os.listdir('test_images'):

您正在列出当前目录中的 test_images 目录。 img 将包含 file1.ext、file2.ext 等形式的值，因为 os.listdir() 仅列出其中的文件和目录的名称，因此当您调用：

scriptpath = os.path.dirname(img)

你基本上什么也得不到，因为 img 不包含任何路径信息。所以，最后，当你这样做时：

image = os.path.join(scriptpath, img)

您在技术上只传递了文件名，因为 scriptpath 是空的。由于您的图片位于 test_images 子目录中，而不是您的工作目录中，因此您通常会收到找不到文件的消息。

有几种方法可以解决这个问题，最简单的方法是在变量中声明一个查找目录并在需要时使用它，例如：

target_path = "test_images"
# ...
for img in os.listdir(target_path):
# ...
    image = os.path.join(target_path, img)

Answer 2

在我看来，我更喜欢使用 glob 库而不是 return 目录中的文件列表。 import glob print (glob.glob("/home/peter/pictures/*.png")

return秒： ['/home/peter/pictures/pic1.png', '/home/peter/pictures/pic2.png', '/home/peter/pictures/pic3.png', ...ect]

如果您想继续您的方法，我相信您没有提供正确的文件夹目录路径。想一想，程序怎么知道test_images所在的位置。

Answer 3

像grillo opinion一样使用glob，或者你可以使用下面的lambda语句

files = os.listdir(file_dir)
img_files = list(filter(lambda x: '.jpg' in x, files))

读取目录中的 JPG 文件？

Read JPG files in a directory?

python

os.path

jupyter

jupyter-notebook