神剑山庄资源网 Design By www.hcban.com
这篇文章主要介绍了python OpenCV GrabCut使用实例解析,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
先上一个效果图:
使用Python3.7+OpenCV 3.x.
需要引入 numpy库。
在图上用鼠标左键和右键标记前景和后景即可.
如果需要重新标记图像,关闭程序重新运行.
以下是具体实现代码
# -*- coding:utf-8 -*- ''' Python: 3.5.7 opencv 3.x 在图上用鼠标左键和右键标记前景和后景即可. 如果需要重新标记图像,关闭程序重新运行. By Ynxf http://www.zhouws.com ''' import cv2 import numpy as np import time img_src = '../test_images/3.jpg' drawing = False mode = False class GrabCut: def __init__(self, t_img): self.img = t_img self.img_raw = img.copy() self.img_width = img.shape[0] self.img_height = img.shape[1] self.scale_size = 640 * self.img_width // self.img_height if self.img_width > 640: self.img = cv2.resize(self.img, (640, self.scale_size), interpolation=cv2.INTER_AREA) self.img_show = self.img.copy() self.img_gc = self.img.copy() self.img_gc = cv2.GaussianBlur(self.img_gc, (3, 3), 0) self.lb_up = False self.rb_up = False self.lb_down = False self.rb_down = False self.mask = np.full(self.img.shape[:2], 2, dtype=np.uint8) self.firt_choose = True # 鼠标的回调函数 def mouse_event2(event, x, y, flags, param): global drawing, last_point, start_point # 左键按下:开始画图 if event == cv2.EVENT_LBUTTONDOWN: drawing = True last_point = (x, y) start_point = last_point param.lb_down = True print('mouse lb down') elif event == cv2.EVENT_RBUTTONDOWN: drawing = True last_point = (x, y) start_point = last_point param.rb_down = True print('mouse rb down') # 鼠标移动,画图 elif event == cv2.EVENT_MOUSEMOVE: if drawing: if param.lb_down: cv2.line(param.img_show, last_point, (x,y), (0, 0, 255), 2, -1) cv2.rectangle(param.mask, last_point, (x, y), 1, -1, 4) else: cv2.line(param.img_show, last_point, (x, y), (255, 0, 0), 2, -1) cv2.rectangle(param.mask, last_point, (x, y), 0, -1, 4) last_point = (x, y) # 左键释放:结束画图 elif event == cv2.EVENT_LBUTTONUP: drawing = False param.lb_up = True param.lb_down = False cv2.line(param.img_show, last_point, (x,y), (0, 0, 255), 2, -1) if param.firt_choose: param.firt_choose = False cv2.rectangle(param.mask, last_point, (x,y), 1, -1, 4) print('mouse lb up') elif event == cv2.EVENT_RBUTTONUP: drawing = False param.rb_up = True param.rb_down = False cv2.line(param.img_show, last_point, (x,y), (255, 0, 0), 2, -1) if param.firt_choose: param.firt_choose = False param.mask = np.full(param.img.shape[:2], 3, dtype=np.uint8) cv2.rectangle(param.mask, last_point, (x,y), 0, -1, 4) print('mouse rb up') if __name__ == '__main__': img = cv2.imread(img_src) if img is None: print('error: 图像为空') g_img = GrabCut(img) cv2.namedWindow('image') # 定义鼠标的回调函数 cv2.setMouseCallback('image', mouse_event2, g_img) while (True): cv2.imshow('image', g_img.img_show) if g_img.lb_up or g_img.rb_up: g_img.lb_up = False g_img.rb_up = False start = time.process_time() bgdModel = np.zeros((1, 65), np.float64) fgdModel = np.zeros((1, 65), np.float64) rect = (1, 1, g_img.img.shape[1], g_img.img.shape[0]) print(g_img.mask) mask = g_img.mask g_img.img_gc = g_img.img.copy() cv2.grabCut(g_img.img_gc, mask, rect, bgdModel, fgdModel, 5, cv2.GC_INIT_WITH_MASK) elapsed = (time.process_time() - start) mask2 = np.where((mask == 2) | (mask == 0), 0, 1).astype('uint8') # 0和2做背景 g_img.img_gc = g_img.img_gc * mask2[:, :, np.newaxis] # 使用蒙板来获取前景区域 cv2.imshow('result', g_img.img_gc) print("Time used:", elapsed) # 按下ESC键退出 if cv2.waitKey(20) == 27: break
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。
神剑山庄资源网 Design By www.hcban.com
神剑山庄资源网
免责声明:本站文章均来自网站采集或用户投稿,网站不提供任何软件下载或自行开发的软件!
如有用户或公司发现本站内容信息存在侵权行为,请邮件告知! 858582#qq.com
神剑山庄资源网 Design By www.hcban.com
暂无python OpenCV GrabCut使用实例解析的评论...
更新日志
2024年09月30日
2024年09月30日
- 孙悦.1996-伙伴【正大国际】【WAV+CUE】
- 纪钧瀚《钢琴阅读时光 雨中书店聆听轻音乐》[FLAC/分轨][399.62MB]
- 证声音乐图书馆《走向自然 疗心爵士乐》[320K/MP3][87.4MB]
- 证声音乐图书馆《走向自然 疗心爵士乐》[FLAC/分轨][184.94MB]
- 陈慧娴.2018-Priscilla-Ism演唱会3CD(2024环球红馆40复刻系列)【环球】【WAV+CUE】
- 郑秀文.1999-我应该得到(国)【华纳】【WAV+CUE】
- 陈家慧.2011-钢琴酒吧2CD【龙吟唱片】【WAV+CUE】
- 证声音乐图书馆《雨季 蓝调吉他 Rainy Blues》[320K/MP3][45.01MB]
- 证声音乐图书馆《雨季 蓝调吉他 Rainy Blues》[FLAC/分轨][109.13MB]
- 赞多《序章》[320K/MP3][45.54MB]
- 许巍.2004-每一刻都是崭新的【步升大风】【WAV+CUE】
- 群星.2024-四方馆影视原声带【韶愔音乐】【FLAC分轨】
- 陈雷.1997-安锁咧【金圆唱片】【WAV+CUE】
- 关淑怡.2013-MY.FAVORITE.SK.3CD【环球】【WAV+CUE】
- Sweety.2006-花言乔语【丰华】【WAV+CUE】