ROI后对图像进行排序（Python，OpenCV）

Question

我接近OpenCV世界的新手。我正在开发一个项目，需要（现在）检测图像中的数字，选择并保存。

这是我使用的代码：

# Importing modules

import cv2
import numpy as np


# Read the input image 
im = cv2.imread('C:\\Users\\User\\Desktop\\test.png')

# Convert to grayscale and apply Gaussian filtering
im_gray = cv2.cvtColor(im, cv2.COLOR_BGR2GRAY)
im_gray = cv2.GaussianBlur(im_gray, (5, 5), 0)

# Threshold the image
ret, im_th = cv2.threshold(im_gray, 90, 255, cv2.THRESH_BINARY_INV)

# Find contours in the image
image, ctrs, hier = cv2.findContours(im_th.copy(), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

# Bounding rectangle for a set of points
i = 0

#rects = [cv2.boundingRect(ctr) for ctr in ctrs]
#rects.sort()

for ctr in ctrs:
    x, y, w, h = cv2.boundingRect(ctrs[i])

    # Getting ROI
    roi = im[y:y+h, x:x+w]

    #cv2.imshow('roi',roi)
    #cv2.waitKey()

    i += 1

    cv2.imwrite('C:\\Users\\User\\Desktop\\crop\\' + str(i) + '.jpg', roi)

#print(rects)    
print("OK - NO ERRORS")

它工作了一半。问题是原始图像（下图）没有排序输出数字（图像格式，需要这样）。

这是输出：

代码有什么问题？

另外，您可以注意rects变量。我用它来做一些调试，我注意到一件有趣的事情：如果我对它进行排序，在控制台中，图像数组顺序是正确的。

有没有办法按原始顺序对图像进行排序？

我也看到了very similar post，但我无法理解解决方案。

非常感谢。

Answer 1

由于ROI可以在二维空间中展开，因此没有自然顺序。

如果您想按x坐标排序，可以这样做：

sorted_ctrs = sorted(ctrs, key=lambda ctr: cv2.boundingRect(ctr)[0])

然后循环sorted_ctrs而不是ctrs。

编辑：更确切地说：

import cv2
import numpy as np

# Read the input image
im = cv2.imread('JnUpW.png')

# Convert to grayscale and apply Gaussian filtering
im_gray = cv2.cvtColor(im, cv2.COLOR_BGR2GRAY)
im_gray = cv2.GaussianBlur(im_gray, (5, 5), 0)

# Threshold the image
ret, im_th = cv2.threshold(im_gray, 90, 255, cv2.THRESH_BINARY_INV)

# Find contours in the image
image, ctrs, hier = cv2.findContours(im_th.copy(), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

# Sort the bounding boxes
sorted_ctrs = sorted(ctrs, key=lambda ctr: cv2.boundingRect(ctr)[0])

for i, ctr in enumerate(sorted_ctrs):
    # Get bounding box
    x, y, w, h = cv2.boundingRect(ctr)

    # Getting ROI
    roi = im[y:y+h, x:x+w]

    # Write to disk
    cv2.imwrite(str(i) +  '.jpg', roi)

#print(rects)
print("OK - NO ERRORS")