证件照换装程序实现(Python+PIL),包括界面
最近需要一个证件照的换装小程序,使用了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