您好,这段代码旨在存储用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。