python+opencv实现实时更新画框,避免画出实心方块,选择目标实现grabcut

在图切算法和目标跟踪交互的时候,往往需要用鼠标选定目标。用opencv+python实现画框选定目标和后续实现grabcut,代码如下。尤其注意画框时候的实时更新,否则画出来是实心的方块
import cv2
import numpy as np

ix,iy = -1,-1
bx,by = -1,-1
i=1
def draw(event,x,y,flags,param):
    global ix,iy,bx,by,i
    if event == cv2.EVENT_LBUTTONDOWN:
        print('按下鼠标')
        ix, iy = x,y
        print('坐标1:',str(ix),str(iy))
    elif event == cv2.EVENT_MOUSEMOVE and flags==cv2.EVENT_FLAG_LBUTTON:
        if i ==1:
            print('鼠标滑动')
            i = i+1
        if i >1:
            pass
        img = img1.copy()#实时更新这一步很重要,如果没有这一步画出来会是一个方块
        cv2.rectangle(img, (ix, iy), (x, y), (0, 0, 255), 3)
        cv2.imshow('iron', img)
    elif event == cv2.EVENT_LBUTTONUP:
        print('鼠标抬起')
        bx, by = x, y
        print('坐标2:',str(bx),str(by))


img = cv2.imread('D:/00000022.jpg',1)
img1 = img.copy()#为了上面的实时更新方框,如果不更新,最后是一个实心方块
cv2.namedWindow('iron',cv2.WINDOW_NORMAL)
cv2.imshow('iron',img1)#第一次显示
cv2.setMouseCallback('iron',draw)#回调鼠标
cv2.waitKey(0)
cv2.destroyAllWindows()

#下面是图切算法部分
print(img.shape[:2])#
mask = np.zeros(img.shape[:2], dtype=np.uint8)#src.shape[:2]=(500,700)
rect = (ix,iy,bx,by)#(50,0)roi边框左上角坐标。(580,500):...右下角坐标
bgdmodel = np.zeros((1,65),np.float64)#后台模型的临时数组。
fgdmodel = np.zeros((1,65),np.float64)#前台模型的临时数组。
cv2.grabCut(img,mask,rect,bgdmodel,fgdmodel,7,mode=cv2.GC_INIT_WITH_RECT)#7:算法迭代次数;算法收敛前,迭代次数越高效果越好;本例,迭代7次达到最优
mask2 = np.where((mask==1) | (mask==3), 255, 0).astype('uint8')
result = cv2.bitwise_and(img,img,mask=mask2)
cv2.imshow("result", result)
cv2.waitKey(0)
cv2.destroyAllWindows()
    # save_dizhi = '.\demo-sequences/vot15_bag3result/imgs/0000'+str(i)+'.jpg'
    # cv.imwrite(save_dizhi,result)
    # cv.waitKey(0)
    # cv.destroyAllWindows()

图2
图1选定目标

如果不实时更新,出来的是这种效果

  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 6
    评论
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值