如何将 laplacian 应用于文件夹中的每个图像

Question

我在一个文件夹中有多个图像，我想处理这些图像并对它们应用一些 OpenCV 函数。 我试图为文件夹中存在的每个图像找到一个拉普拉斯算子，但我可以一次处理一个。

import cv2
import numpy as np
from matplotlib import pyplot as plt

img = cv2.imread('xyz.jpg',0)

laplacian = cv2.Laplacian(img,cv2.CV_64F)
sobelx = cv2.Sobel(img,cv2.CV_64F,1,0,ksize=5)
sobely = cv2.Sobel(img,cv2.CV_64F,0,1,ksize=5)

plt.subplot(2,2,1),plt.imshow(img,cmap = 'gray')
plt.title('Original'), plt.xticks([]), plt.yticks([])

plt.subplot(2,2,2),plt.imshow(laplacian,cmap = 'gray')
plt.title('Laplacian'), plt.xticks([]), plt.yticks([])
plt.show()

Answer 1

可以将代码重构为一个函数，并在多个文件路径上循环该函数。

from pathlib import Path
import cv2
import numpy as np
from matplotlib import pyplot as plt


def process(filepath):
    img = cv2.imread(filepath, 0)

    laplacian = cv2.Laplacian(img,cv2.CV_64F)
    sobelx = cv2.Sobel(img,cv2.CV_64F,1,0,ksize=5)
    sobely = cv2.Sobel(img,cv2.CV_64F,0,1,ksize=5)

    plt.subplot(2,2,1),plt.imshow(img,cmap = 'gray')
    plt.title('Original'), plt.xticks([]), plt.yticks([])

    plt.subplot(2,2,2),plt.imshow(laplacian,cmap = 'gray')
    plt.title('Laplacian'), plt.xticks([]), plt.yticks([])
    plt.show()
    

for filepath in Path("path/to/images").glob("*.jpg"):
    print(f"Processing {filepath}")
    process(filepath)

Answer 2

实现这一目标的最简单方法是使用 for 循环

import os
from glob import glob

BASE_PATH = os.path.dirname(os.path.realpath(__file__))
IMGS_PATH = os.path.join(BASE_PATH, '*.jpg')

for path in glob(IMGS_PATH):
    img = cv2.imread(path)
    # processing

如果您有大量数据，这可能不是最有效的方法，在这种情况下，我可能会寻找一个像 Keras 一样进行并行图像加载的库。