在Python中将图像划分为矩形信息

Question

我有一系列图像，它们只不过是黑色背景中的一系列彩色矩形 示例：。我如何（在 Python 3.7 中使用 PIL）制作一个程序来识别这些矩形（xy 位置、颜色和大小）？效率并不是很重要，所以嵌套的fors 是可以接受的。到目前为止，我能够提出的每个算法都有很多缺陷，并且可能以多种不同的方式出错，或者只是太复杂了，根本不值得

Answer 1

您根本不需要编写任何代码，您可以使用 ImageMagick 来完成，它安装在大多数 Linux 发行版上，适用于 macOS 和 Windows。

只需在终端（Windows 上的命令提示符）中，您就可以运行：

magick convert image.png                            \
   -define connected-components:verbose=true        \
   -define connected-components:area-threshold=100  \
   -connected-components 4 -auto-level output.png

样本输出

Objects (id: bounding-box centroid area mean-color):
  0: 1200x714+0+0 651.2,369.3 703177 srgb(0,0,0)
  164: 1200x86+0+714 599.5,756.5 103200 srgb(255,21,0)
  2: 363x155+80+60 261.0,137.0 56265 srgb(255,255,255)
  26: 127x323+60+302 122.6,463.2 39668 srgb(255,255,255)
  54: 308x109+352+373 505.5,427.0 33572 srgb(255,255,255)
  1: 102x159+641+47 691.5,126.0 16218 srgb(255,255,255)
  53: 79x100+977+371 1016.0,420.5 7900 srgb(0,17,255)

因此，查看从0: 开始的行，有一个尺寸为 1200x714 的矩形，从 0,0（左上角）开始，颜色为黑色，即 srgb(0,0,0)。

查看下一行，有一个 1200x86 的矩形，从左上角向下 714 像素开始，颜色为红色，即 srgb(255,21,0)。

等等。

最后一行是一个 79x100 的矩形，位于 [977,31]，颜色为蓝色，即 srgb(0,17,255)。

Answer 2

我不知道您对 OpenCV 的了解程度。

如果你准备好使用 openCV，你可以使用 findContours 来获得想要的东西。

下面是代码：

    import cv2
    readImage= cv2.imread(r"<ImagePath>\oKvDi.png")
    img_gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

    #Code to Find edges of Square using Canny edge detection method and finding Contours and drawing in Back Line

    edges = cv2.Canny(img_gray, 0, 100)
    contours, hierarchy = cv2.findContours(edges, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)

    #Just for your reference to show all rectangles are found
    cv2.drawContours(readImage, contours, -1, (0, 0, 0), 5)
    cv2.imwrite(r"<ImageSavePath>/a.png",readImage)

希望这能解决您的问题