def draw_rectangle(event,x,y,flags,param):
if event == cv2.EVENT_LBUTTONDOWN:
drawing = True
param.drawing=True
param.setxy(x,y)
param.printxy()
elif event == cv2.EVENT_MOUSEMOVE:
if param.drawing == True:
a=layering(param.layer_data,param.layer_index)
img_temp_rect=a.copy()
img_temp_rect=cv2.rectangle(img_temp_rect,(param.ix,param.iy),(x,y),(0,255,0),2)
cv2.imshow('image',img_temp_rect)
elif event == cv2.EVENT_LBUTTONUP:
param.drawing = False
shape_dimension1 = np.asarray((param.ix,param.iy))
shape_dimension2=np.asarray((x,y))
param.layer_data.resize(param.layer_data.shape[0]+1,4)
k = param.layer_data.shape
param.layer_data[k[0]-1, 0] = 1
param.layer_data[k[0]-1, 1] = shape_dimension1
param.layer_data[k[0]-1, 2] = shape_dimension2
param.layer_data[k[0]-1,3]=param.layer_index+1
param.layer_index=param.layer_index+1
print param.layer_data
a=layering(param.layer_data,param.layer_index)
评论列表
文章目录