实例介绍
【实例简介】
去除图像中的白色背景,得到透明背景的保留主体的png图像的python代码;
【实例截图】
原图
去除白色背景后得到的图片
【核心代码】
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 | def remove_white_bg(img_list, output_dir): for img_path in img_list: img = cv2.imread(img_path) if img is None : print (img_path) continue ## (1) Convert to gray, and threshold gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) th, threshed = cv2.threshold(gray, 235 , 255 , cv2.THRESH_BINARY_INV) ## (2) Morph-op to remove noise kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, ( 2 , 2 )) morphed = cv2.morphologyEx(threshed, cv2.MORPH_CLOSE, kernel) print (morphed.shape) w,h,c = img.shape img = np.transpose(img,[ 2 , 0 , 1 ]) z = np.ones([ 1 ,w,h]).astype(img.dtype) * 255 img = np.vstack((img,z)) img = np.transpose(img,[ 1 , 2 , 0 ]) cnts = cv2.findContours(morphed, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)[ - 2 ] stencil = np.zeros([w,h, 4 ]).astype(img.dtype) color = [ 255 , 255 , 255 , 255 ] cv2.fillPoly(stencil, cnts, color) result = cv2.bitwise_and(img, img, mask = morphed) result = cv2.bitwise_and(result, stencil) result = CropImage(result, result.shape[ 0 ],result.shape[ 1 ])[ 0 ] output_path = os.path.join(output_dir, os.path.split(img_path)[ - 1 ] '.png' ) cv2.imwrite(output_path, result) |
小贴士
感谢您为本站写下的评论,您的评论对其它用户来说具有重要的参考价值,所以请认真填写。
- 类似“顶”、“沙发”之类没有营养的文字,对勤劳贡献的楼主来说是令人沮丧的反馈信息。
- 相信您也不想看到一排文字/表情墙,所以请不要反馈意义不大的重复字符,也请尽量不要纯表情的回复。
- 提问之前请再仔细看一遍楼主的说明,或许是您遗漏了。
- 请勿到处挖坑绊人、招贴广告。既占空间让人厌烦,又没人会搭理,于人于己都无利。
关于好例子网
本站旨在为广大IT学习爱好者提供一个非营利性互相学习交流分享平台。本站所有资源都可以被免费获取学习研究。本站资源来自网友分享,对搜索内容的合法性不具有预见性、识别性、控制性,仅供学习研究,请务必在下载后24小时内给予删除,不得用于其他任何用途,否则后果自负。基于互联网的特殊性,平台无法对用户传输的作品、信息、内容的权属或合法性、安全性、合规性、真实性、科学性、完整权、有效性等进行实质审查;无论平台是否已进行审查,用户均应自行承担因其传输的作品、信息、内容而可能或已经产生的侵权或权属纠纷等法律责任。本站所有资源不代表本站的观点或立场,基于网友分享,根据中国法律《信息网络传播权保护条例》第二十二与二十三条之规定,若资源存在侵权或相关问题请联系本站客服人员,点此联系我们。关于更多版权及免责申明参见 版权及免责申明
网友评论
我要评论