需要帮助寻找更好的图像区分方法

时间:2019-05-28 18:29:02

标签: python-3.x machine-learning jupyter-notebook

我正在尝试使用python区分大量图像。

我在Jupyter笔记本上运行python代码,并且尝试通过将图像转换为黑白并计数白色像素来区分图像。如果像素数超过5000,则将图片设置为两种图片之一。

这对两张图片效果很好,但对某些图片却无效。有些较小,有些较亮,但是它们都具有相同的样式。这是两张图片:

enter image description here enter image description here

这是我的代码:

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%。我该怎么办?

0 个答案:

没有答案