黑科技编程开发，利用Python使图片完美去除水印

PIL：Python Imaging Library 是 python 图像处理标准库非常强大，但只能支持 python 2.所以有志愿者 PIL 在此基础上创建了支持 python 3的 pillow，并添加了一些新的特性。

pip install pillow 

pymupdf 可以用 python 访问扩展名称*.pdf、.xps、.oxps、.epub、.cbz或*.fb2.文件。它还支持许多流行的图像格式，包括多页TIFF图像。

pip install PyMuPDF 

导入需要使用的模块

from PIL import Image from itertools import product import fitz import os 

pdf 去除水印的原理与图片去除水印的原理相似。小编从去除上图的水印开始。

通过微信截图查看水印像素点 RGB。

可见水印 RGB 是 (210, 210, 210)，这里用 RGB 的和超过 620 判断是水印点，此时将像素颜色替换为白色。最后保存图片。

rgb = img.getpixel(pos)[:3] if(sum(rgb) >= 620):     img.putpixel(pos, (255, 255, 255))  img.save('d:/qsy.png') 

示例结果：

PDF 去水印的原理和图片去水印的原理大致相同，用 PyMuPD 打开 pdf 文件后，将 pdf 的每一页都转换为图片 pixmap，pixmap 有它自己的 RGB，只需要将 pdf 水印中的 RGB 改为(255, 255, 255) 最后保存为图片。

def remove_pdf():
    page_num = 0
    pdf_file = input("请输入 pdf 地址：")
    pdf = fitz.open(pdf_file);
    for page in pdf:
        pixmap = page.get_pixmap()
        for pos in product(range(pixmap.width), range(pixmap.height)):
            rgb = pixmap.pixel(pos[0], pos[1])
            if(sum(rgb) >= 620):
                pixmap.set_pixel(pos[0], pos[1], (255, 255, 255))
        pixmap.pil_save(f"d:/pdf_images/{page_num}.png")
        print(f"第{page_num}水印去除完成")
        page_num = page_num + 1

示例结果：

图片转 pdf 需要注意的是图片的排序，数字文件名必须先转换为 int 类型后排序。用 PyMuPDF 模块打开图片后将图片用 convertToPDF() 函数转成单页的 pdf。插入到新的 pdf 文件中。

def pic2pdf():
    pic_dir = input("请输入图片文件夹路径：")
    
    pdf = fitz.open()
    img_files = sorted(os.listdir(pic_dir),key=lambda x:int(str(x).split('.')[0]))
    for img in img_files:
        print(img)
        imgdoc = fitz.open(pic_dir + '/' + img)  
        pdfbytes = imgdoc.convertToPDF()   
        imgpdf = fitz.open("pdf", pdfbytes)
        pdf.insertPDF(imgpdf)       
    pdf.save("d:/demo.pdf")         
    pdf.close()

pdf 和图片上恼人的水印终于可以在强大的 python 面前消失了。小伙伴们学会了吗？

学好 Python 不论是就业还是做副业赚钱都不错，但要学会 Python 还是要有一个学习规划。最后大家分享一份全套的 Python 学习资料，给那些想学习 Python 的小伙伴们一点帮助！

Python所有方向路线就是把Python常用的技术点做整理，形成各个领域的知识点汇总，它的用处就在于，你可以按照上面的知识点去找对应的学习资源，保证自己学得较为全面。

工欲善其事必先利其器。学习Python常用的开发软件都在这里了，给大家节省了很多时间。

我们在看视频学习的时候，不能光动眼动脑不动手，比较科学的学习方法是在理解之后运用它们，这时候练手项目就很适合了。

光学理论是没用的，要学会跟着一起敲，要动手实操，才能将自己的所学运用到实际当中去，这时候可以搞点实战案例来学习。

我们学习Python必然是为了找到高薪的工作，下面这些面试题是来自阿里、腾讯、字节等一线互联网大厂最新的面试资料，并且有阿里大佬给出了权威的解答，刷完这一套面试资料相信大家都能找到满意的工作。

Python资料、技术、课程、解答、咨询也可以直接点击下面名片，添加官方客服斯琪↓