我有一个包含大量图像的文件夹,其中几张图像几乎是黑暗的,如下所示:Dark Images,而且很少有图像是这样的好图像:Good images
基本上我可以使用下面的代码识别较暗的图像,对于较暗的图像,np.mean(图像)低于0.1,对于良好的图像,它高于0.1范围:
from skimage import io, img_as_float
import numpy as np
image = io.imread('C:/Data/Testing/Image_0_5.jpg')
image = img_as_float(image)
print(np.mean(image))
但我想要的是传递具有所有图像的特定文件夹,以便我的代码可以解析所有图像,并列出具有暗图像的图像。需要帮助。
答案 0 :(得分:3)
感谢方向的人,感谢。
这是我的代码:
import matplotlib.image as mpimg
import os
def load_images(folder):
images = []
for filename in os.listdir(folder):
img = mpimg.imread(os.path.join(folder, filename))
img = img_as_float(img)
#print(np.mean(img))
if img is not None:
images.append(img)
if(np.mean(img) < 0.1):
print filename
load_images('C:/Data/Testing')
我已经实现了我想要的东西:)