赞
踩
最近需要一个证件照的换装小程序,使用了Python+openCV编写了一个,发现对于界面像素不高的有模糊边缘,处理后效果不好。
现在使用PIL来抠图,放到要放置的背景图上,完成背景底色变化,效果很好
测试了好多照片,发现效果确实不错(图片为网上找的帅哥美女,侵权删)
界面实现部分代码没有放,此处简单说一个代码如何实现的。
点击按钮,在labtext中显示出原图
- # 返回的是文件的路径及文件类型
- filename, filetype = QtWidgets.QFileDialog.getOpenFileName(self, "选取文件", "./",
- "All Files (*);; JPEG Files (*.jpg);;PNG Files(*.png);; BMP Files(*.bmp)")
- # 读取照片
- image = cv2.imread(filename)
- pixmap = QPixmap(filename)
- # 显示图片
- self.lab_show_choose.setPixmap(pixmap)
- # 这一行是显示格式的设置,我们添加这一行就可以,避免整个图片显示不全
- self.lab_show_choose.setScaledContents(True)
- # 去掉背景图,提取照片
- rmbg.remove_background_from_img_file(filename)
- # 根据选择,更换不同的背景图
- if self.radioButton_red.isChecked():
- # 背景图,大小同前景图
- background = Image.new('RGBA', foreground.size, BACKGROUND_COLOR['RED'])
-
- if self.radioButton_blue.isChecked():
- background = Image.new('RGBA', foreground.size, BACKGROUND_COLOR['BLUE'])
-
- if self.radioButton_white.isChecked():
- background = Image.new('RGBA', foreground.size, BACKGROUND_COLOR['WHITE'])
-
- # 更换背景后的图片
- background.paste(foreground, mask=foreground)
- # 使用PIL来显示图
- background.show()
- # 使用PIL来显示图
- background.show()
- background.save(newfilepath)
- QMessageBox.information(self, "保存图片", "图片已保存到"+newfilepath, QMessageBox.Yes, QMessageBox.Yes)
rmbg.remove_background_from_img_file(filename):缺点:这个方法只能免费使用50次,超过50次就会提示报错
所以想要有一个永久能使用的软件,需要付费或者使用backgroundremover
完整源码请关注公众号:一枚努力的程序猿
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。