如何在python中遍历多个图像？

Question

我正在使用python，从事图像处理工作。我使用opencv从图像中提取出唯一的绿色，并计算其像素为绿色和黑色，但只针对一张图像。就像下面的脚本一样，我如何应用脚本中的循环来处理几个(100+)图像。

import cv2
import numpy as np
## Read
img = cv2.imread("camera/2018-11-25_0116.jpg")
## convert to hsv
hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)


## mask of green (36,0,0) ~ (70, 255,255)
mask = cv2.inRange(hsv, (50, 50, 50), (70, 255,255))

## slice the green
imask = mask>0
green = np.zeros_like(img, np.uint8)
green[imask] = img[imask]

cv2.imwrite("2018-11-25_0116.jpg", green)

from PIL import Image
im = Image.open('2018-11-25_0116.jpg')

black = 0
green = 0

for pixel in im.getdata():
    if pixel == (0, 0, 0): # if your image is RGB (if RGBA, (0, 0, 0, 255) or so
        black += 1
    else:
        green += 1
print('black=' + str(black)+', green='+str(green))

Answer 1

os.walk应该可以解决你的问题。

import cv2
import numpy as np
## Read
for root, dirnames, filenames in os.walk(source):
    for filename in filenames:
            matches.append(os.path.join(root, filename))
for match in matches:
    img = cv2.imread(match)
## convert to hsv

像这样的东西就行了。别忘了定义源变量。这将是您的目录。