最近需要一个证件照的换装小程序,使用了Python+openCV编写了一个,发现对于界面像素不高的有模糊边缘,处理后效果不好。
现在使用PIL来抠图,放到要放置的背景图上,完成背景底色变化,效果很好
效果图:
测试了好多照片,发现效果确实不错(图片为网上找的帅哥美女,侵权删)
界面实现部分代码没有放,此处简单说一个代码如何实现的。
1. 显示原图
点击按钮,在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)
2.抠图,生成无背景色的图片
# 去掉背景图,提取照片
rmbg.remove_background_from_img_file(filename)
3.放置到设计好的图片中
# 根据选择,更换不同的背景图
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()
4.预览效果
# 使用PIL来显示图
background.show()
5.保存制定的路径中
background.save(newfilepath)
QMessageBox.information(self, "保存图片", "图片已保存到"+newfilepath, QMessageBox.Yes, QMessageBox.Yes)
问题:
rmbg.remove_background_from_img_file(filename):缺点:这个方法只能免费使用50次,超过50次就会提示报错
所以想要有一个永久能使用的软件,需要付费或者使用backgroundremover
完整源码请关注公众号:一枚努力的程序猿