python opencv 利用鼠标点击事件移除标记区域

                                                         

上图中左边的图是需要进行移除正方形上的多边形，右图是移除后的结果。操作也比较简单，使用鼠标在多边形外围点击四下就可以移除该区域了，鼠标点击用左图红色圆圈表示。具体实现如下代码：

#-*- coding:utf-8 -*-
import cv2
import pandas as pd
import numpy as np
import os

def order_points(pts):
    ''' sort rectangle points by clockwise '''
    sort_x = pts[np.argsort(pts[:, 0]), :]
    Left = sort_x[:2, :]
    Right = sort_x[2:, :]
    Left = Left[np.argsort(Left[:, 1])[::-1], :]
    Right = Right[np.argsort(Right[:, 1]), :]
    return np.concatenate((Left, Right), axis=0)

def draw_rectangle(event,x,y,flags,param):
    global box
    if event == cv2.EVENT_LBUTTONDOWN:
        box.append([x, y])
    return box

def correct(img):
    cv2.namedWindow("image")
    cv2.setMouseCallback('image', draw_rectangle)
    cv2.imshow("image", img)
    cv2.waitKey(0)


if __name__ == '__main__':
    indir = r'..\CSDN\gallery'
    outdir = r'..\CSDN\gallery'
    imgname_in = 'inpainting_exp.jpg'
    imgname_out = 'repaired.jpg'
    img = cv2.imread(os.path.join(indir, imgname_in))
    box = []
    correct(img)
    print(box)
    box = order_points(np.intp(np.array(box)))
    after = cv2.drawContours(img, [box], -1, (0, 0, 0), -1)
    cv2.imwrite(os.path.join(outdir, imgname_out), after)