如何使用cv2分割图像?

时间:2020-05-23 07:57:32

标签: python opencv

我有图像 paint.bmp 。我想将其拆分为32x32的块。这是我的代码:

import cv2

image = cv2.imread('paint.bmp')
img_h, img_w = image.shape[:2]
bl_w, bl_h = 32, 32

for i in range(int(img_h/bl_h)):
  for j in range(int(img_w/bl_w)):
    cropped = image[i*bl_h:bl_h, j*bl_w:bl_w]
    cv2.imwrite("Cropped image{}{}.bmp".format(str(i+1), str(j+1)), cropped)

第一个块正在保存,但是在程序崩溃后,显示以下内容:
cv2.error: OpenCV(4.2.0) C:\projects\opencv-python\opencv\modules\imgcodecs\src\loadsave.cpp:715: error (-215:Assertion failed) !_img.empty() in function 'cv::imwrite'

1 个答案:

答案 0 :(得分:3)

问题出在线
cropped = image[i*bl_h:bl_h, j*bl_w:bl_w]

在第一个块中,您像这样进行切片:image[0:32, 0:32],这很好用。但是,在第二个块中,您尝试使用image[32:32, 32:32],它将始终返回空数组。

有一种更简单的方法:

import cv2

image = cv2.imread('image.bmp')
img_h, img_w = image.shape[:2]
bl_w, bl_h = 32, 32

for i in range(int(img_h/bl_h)):
  for j in range(int(img_w/bl_w)):
    cropped = image[i*bl_h:(i+1)* bl_h, j*bl_w:(j+1)*bl_w]
    cv2.imwrite("Cropped image{}{}.bmp".format(str(i+1), str(j+1)), cropped)