我正在尝试使用python区分大量图像。
我在Jupyter笔记本上运行python代码,并且尝试通过将图像转换为黑白并计数白色像素来区分图像。如果像素数超过5000,则将图片设置为两种图片之一。
这对两张图片效果很好,但对某些图片却无效。有些较小,有些较亮,但是它们都具有相同的样式。这是两张图片:
这是我的代码:
for index,element in data.iterrows():
image_file = Image.open('images_training/'+ str(element[0])
image_file = image_file.convert('1') # convert image to black
pil_img = image_file.convert('RGB')
open_cv_image = np.array(pil_img)
# Convert RGB to BGR
img = open_cv_image[:, :, ::-1].copy()
height,width,channels = img.shape
#crop image
img = img.copy()
x = width//2-(200//2)
y = height//2-(200//2)
crop_img = (img[y:y+200, x:x+200])
white_pixels.append([element[1],(np.sum(crop_img == 255)/(np.sum(crop_img != 255)))])
我正在尝试寻找另一种方法,以便在检测到的图像的好坏之间获得更好的比例。目前,有60%的图像可以使用此功能,但我希望比例超过85%。我该怎么办?