您好,这段代码旨在存储用open cv绘制的矩形的坐标,并将结果编译为单个图像。

import numpy as np
import cv2

im = cv2.imread('1.jpg')
im3 = im.copy()

gray = cv2.cvtColor(im,cv2.COLOR_BGR2GRAY)
blur = cv2.GaussianBlur(gray,(5,5),0)
thresh = cv2.adaptiveThreshold(blur,255,1,1,11,2)



contours,hierarchy = cv2.findContours(thresh,cv2.RETR_LIST,cv2.CHAIN_APPROX_SIMPLE)


squares = []

for cnt in contours:
    if cv2.contourArea(cnt)>50:
        [x,y,w,h] = cv2.boundingRect(cnt)

        if  h>28 and h<34:
            rect = (cv2.rectangle(im,(x,y),(x+w,y+h),(255,255,255),3))
            squares.append(cv2.boundingRect(cnt))
            cv2.imwrite('norm1.jpg',im)

crop_img = [[255 for x in xrange(377)] for x in xrange(377) ]

for s in squares:
    s = squares[0]
    x = s[0]
    y = s[1]
    w = s[2]
    h = s[3]
    img = im[y:y+h,x:x+w]
    for col in range(y,y+h):
        for row in range(x,x+w):
            if img[col - y][row - x].tolist() == [0,0,0]:
                crop_img[col][row] = [0,0,0]



cv2.imwrite("cropped.jpg", np.array(crop_img))

但是,它抛出此错误信息
File "C:\Users\Program\Desktop\new  1.py", line 43, in <module>
    cv2.imwrite("cropped.jpg", np.array(crop_img))
ValueError: setting an array element with a sequence

我已经读到这可能是由“不均匀”矩阵引起的,但是经过几轮测试,我证实它的确是正方形 377 x 377矩阵

供参考:“1.jpg”是下面显示的图像

任何有关如何解决此错误的线索将不胜感激!

最佳答案

这里的每个“像素”都有一个值。 crop_img = [[255 for x in xrange(377)] for x in xrange(377) ]
但是稍后您将其中一些设置为3个值的列表。那就是问题所在。
我认为这应该解决它:crop_img = [[[255, 255, 255] for x in xrange(377)] for x in xrange(377) ]
尽管您不妨从形状为377,377的numpy数组开始,而不要先转换为3。

09-25 16:20