Python中的补丁提取

Question

我有一个图像，我想提取它的补丁，然后将每个补丁保存为该文件夹中的图像。这是我的第一次尝试：

from sklearn.feature_extraction import image
from sklearn.feature_extraction.image import extract_patches_2d
import os, sys
from PIL import Image



imgFile = Image.open('D1.gif')
window_shape = (10, 10)
B = extract_patches_2d(imgFile, window_shape)

print imgFile

但是我收到以下错误：

AttributeError：shape

我在网上搜索过，但找不到任何东西。如果有人能帮助我，我将非常感激。

提前致谢

Answer 1

根据documentation，extract_patches_2d的第一个参数是数组或形状。

首先应该从imgFile创建一个数组，以便获得像素，然后将该数组传递给函数。

import numpy
import PIL

# Convert Image to array
img = PIL.Image.open("foo.jpg").convert("L")
arr = numpy.array(img)