Python 实现给图片加文字或logo水印

前言

本文提供给图片添加文字或者logo图片水印的python工具，打造专属图片。

环境依赖

ffmpeg环境安装，ffmpy安装：

pip install ffmpy -i https://pypi.douban.com/simple

代码

上代码。

#!/user/bin/env python# coding=utf-8\"\"\"@project : csdn@author  : 剑客阿良_ALiang@file   : image_add_watermark_tool.py@ide    : PyCharm@time   : 2021-11-20 14:18:13\"\"\"import osimport uuidfrom ffmpy import FFmpeg# 图片加文字水印def image_add_text(image_path: str,output_dir: str,text: str,font_name: str,font_size=100,position=(0, 0),font_color=\'blue\',box=1,box_color=\'white\'):ext = _check_format(image_path)result = os.path.join(output_dir, \'{}.{}\'.format(uuid.uuid4(), ext))ff = FFmpeg(inputs={image_path: None},outputs={result: \'-vf drawtext=\\\"fontsize={}:fontfile={}:text=\\\'{}\\\':x={}:y={}:fontcolor={}:box={}:boxcolor={}\\\"\'.format(font_size,font_name,text,position[0],position[1],font_color,box,box_color)})print(ff.cmd)ff.run()return result# 图片添加logodef image_add_logo(image_path: str,output_dir: str,logo_path: str,position=(0, 0)):ext = _check_format(image_path)result = os.path.join(output_dir, \'{}.{}\'.format(uuid.uuid4(), ext))filter_cmd = \'-vf \\\"movie=\\\'{}\\\' [wm];[in] [wm]overlay={}:{} [out]\\\"\'ff = FFmpeg(inputs={image_path: None}, outputs={result: filter_cmd.format(logo_path, position[0], position[1])})print(ff.cmd)ff.run()return resultdef _check_format(image_path: str):ext = os.path.basename(image_path).strip().split(\'.\')[-1]if ext not in [\'png\', \'jpg\']:raise Exception(\'format error\')return ext

代码说明

1、image_add_text方法给图片添加文字水印方法，主要参数为：图片路径、输出目录、

水印文字、字体名称、字体大小（默认100）、文字左上角坐标（默认0:0）、文字颜色（默认蓝色）、是否需要背景（默认需要为1，不需要为0）、背景色（默认白色）。

2、image_add_logo方法给图片添加logo，主要参数为：图片路径、输出目录、logo图片地址、文字左上角坐标（默认0:0）。

3、注意logo地址如果有类似C:/这种windows盘的路径情况，需要对\”:\”进行转义。后面验证的时候，可以看看我给的参数。

4、文件名为了避免重复，采用了uuid作为文件名。

5、图片后缀格式校验只有两种，按需添加即可。

验证一下

准备的素材如下，分别为图片与logo图

验证代码

if __name__ == \'__main__\':print(image_add_text(\'C:/Users/huyi/Desktop/1.jpg\',\'C:/Users/huyi/Desktop/\', \'剑客阿良_ALiang\', \'微软雅黑\', box=0))print(image_add_logo(\'C:/Users/huyi/Desktop/1.jpg\',\'C:/Users/huyi/Desktop/\', \'C\\\\:/Users/huyi/Desktop/logo.png\', (30, 10)))

注意我给出的logo地址路径有什么不同。

执行结果

PyDev console: starting.Python 3.6.13 |Anaconda, Inc.| (default, Mar 16 2021, 11:37:27) [MSC v.1916 64 bit (AMD64)] on win32runfile(\'D:/spyder/csdn/image_add_watermark_tool.py\', wdir=\'D:/spyder/csdn\')ffmpeg -i C:/Users/huyi/Desktop/1.jpg -vf drawtext=fontsize=100:fontfile=微软雅黑:text=\'剑客阿良_ALiang\':x=0:y=0:fontcolor=blue:box=0:boxcolor=white C:/Users/huyi/Desktop/2226d7e0-5166-4ffd-ba95-9ca7d9d6f72d.jpgffmpeg version n4.3.1-20-g8a2acdc6da Copyright (c) 2000-2020 the FFmpeg developersbuilt with gcc 9.3-win32 (GCC) 20200320configuration: --prefix=/ffbuild/prefix --pkg-config-flags=--static --pkg-config=pkg-config --cross-prefix=x86_64-w64-mingw32- --arch=x86_64 --target-os=mingw32 --enable-gpl --enable-version3 --disable-debug --enable-iconv --enable-zlib --enable-libxml2 --enable-libfreetype --enable-libfribidi --enable-gmp --enable-lzma --enable-fontconfig --enable-libvmaf --disable-vulkan --enable-libvorbis --enable-amf --enable-libaom --enable-avisynth --enable-libdav1d --enable-ffnvcodec --enable-cuda-llvm --disable-libglslang --enable-libass --enable-libbluray --enable-libmp3lame --enable-libopus --enable-libtheora --enable-libvpx --enable-libwebp --enable-libmfx --enable-libopencore-amrnb --enable-libopencore-amrwb --enable-libopenjpeg --enable-librav1e --enable-schannel --enable-sdl2 --enable-libsoxr --enable-libsrt --enable-libtwolame --enable-libvidstab --enable-libx264 --enable-libx265 --enable-libxvid --enable-libzimg --extra-cflags=-DLIBTWOLAME_STATIC --extra-cxxflags= --extra-ldflags=-pthread --extra-libs=-lgomplibavutil      56. 51.100 / 56. 51.100libavcodec     58. 91.100 / 58. 91.100libavformat    58. 45.100 / 58. 45.100libavdevice    58. 10.100 / 58. 10.100libavfilter     7. 85.100 /  7. 85.100libswscale      5.  7.100 /  5.  7.100libswresample   3.  7.100 /  3.  7.100libpostproc    55.  7.100 / 55.  7.100Input #0, image2, from \'C:/Users/huyi/Desktop/1.jpg\':Duration: 00:00:00.04, start: 0.000000, bitrate: 181106 kb/sStream #0:0: Video: mjpeg (Progressive), yuvj444p(pc, bt470bg/unknown/unknown), 3840x2160, 25 tbr, 25 tbn, 25 tbcStream mapping:Stream #0:0 -> #0:0 (mjpeg (native) -> mjpeg (native))Press [q] to stop, [?] for helpFontconfig error: Cannot load default config file[Parsed_drawtext_0 @ 0000014152ea5400] Using \"C:/Windows/fonts/msyh.ttc\"Output #0, image2, to \'C:/Users/huyi/Desktop/2226d7e0-5166-4ffd-ba95-9ca7d9d6f72d.jpg\':Metadata:encoder         : Lavf58.45.100Stream #0:0: Video: mjpeg, yuvj444p(pc), 3840x2160, q=2-31, 200 kb/s, 25 fps, 25 tbn, 25 tbcMetadata:encoder         : Lavc58.91.100 mjpegSide data:cpb: bitrate max/min/avg: 0/0/200000 buffer size: 0 vbv_delay: N/Aframe=    1 fps=0.0 q=11.5 Lsize=N/A time=00:00:00.04 bitrate=N/A speed=0.158xvideo:634kB audio:0kB subtitle:0kB other streams:0kB global headers:0kB muxing overhead: unknownC:/Users/huyi/Desktop/2226d7e0-5166-4ffd-ba95-9ca7d9d6f72d.jpgffmpeg -i C:/Users/huyi/Desktop/1.jpg -vf \"movie=\'C\\:/Users/huyi/Desktop/logo.png\' [wm];[in] [wm]overlay=30:10 [out]\" C:/Users/huyi/Desktop/c626411e-531f-4dab-9a78-8103d92bbf1d.jpgffmpeg version n4.3.1-20-g8a2acdc6da Copyright (c) 2000-2020 the FFmpeg developersbuilt with gcc 9.3-win32 (GCC) 20200320configuration: --prefix=/ffbuild/prefix --pkg-config-flags=--static --pkg-config=pkg-config --cross-prefix=x86_64-w64-mingw32- --arch=x86_64 --target-os=mingw32 --enable-gpl --enable-version3 --disable-debug --enable-iconv --enable-zlib --enable-libxml2 --enable-libfreetype --enable-libfribidi --enable-gmp --enable-lzma --enable-fontconfig --enable-libvmaf --disable-vulkan --enable-libvorbis --enable-amf --enable-libaom --enable-avisynth --enable-libdav1d --enable-ffnvcodec --enable-cuda-llvm --disable-libglslang --enable-libass --enable-libbluray --enable-libmp3lame --enable-libopus --enable-libtheora --enable-libvpx --enable-libwebp --enable-libmfx --enable-libopencore-amrnb --enable-libopencore-amrwb --enable-libopenjpeg --enable-librav1e --enable-schannel --enable-sdl2 --enable-libsoxr --enable-libsrt --enable-libtwolame --enable-libvidstab --enable-libx264 --enable-libx265 --enable-libxvid --enable-libzimg --extra-cflags=-DLIBTWOLAME_STATIC --extra-cxxflags= --extra-ldflags=-pthread --extra-libs=-lgomplibavutil      56. 51.100 / 56. 51.100libavcodec     58. 91.100 / 58. 91.100libavformat    58. 45.100 / 58. 45.100libavdevice    58. 10.100 / 58. 10.100libavfilter     7. 85.100 /  7. 85.100libswscale      5.  7.100 /  5.  7.100libswresample   3.  7.100 /  3.  7.100libpostproc    55.  7.100 / 55.  7.100Input #0, image2, from \'C:/Users/huyi/Desktop/1.jpg\':Duration: 00:00:00.04, start: 0.000000, bitrate: 181106 kb/sStream #0:0: Video: mjpeg (Progressive), yuvj444p(pc, bt470bg/unknown/unknown), 3840x2160, 25 tbr, 25 tbn, 25 tbcStream mapping:Stream #0:0 -> #0:0 (mjpeg (native) -> mjpeg (native))Press [q] to stop, [?] for help[swscaler @ 000001d6cc29ad80] deprecated pixel format used, make sure you did set range correctlyOutput #0, image2, to \'C:/Users/huyi/Desktop/c626411e-531f-4dab-9a78-8103d92bbf1d.jpg\':Metadata:encoder         : Lavf58.45.100Stream #0:0: Video: mjpeg, yuvj420p(pc), 3840x2160, q=2-31, 200 kb/s, 25 fps, 25 tbn, 25 tbcMetadata:encoder         : Lavc58.91.100 mjpegSide data:cpb: bitrate max/min/avg: 0/0/200000 buffer size: 0 vbv_delay: N/A[Parsed_movie_0 @ 000001d6cc40bd80] EOF timestamp not reliableframe=    1 fps=0.0 q=11.4 Lsize=N/A time=00:00:00.04 bitrate=N/A speed=0.176xvideo:455kB audio:0kB subtitle:0kB other streams:0kB global headers:0kB muxing overhead: unknownC:/Users/huyi/Desktop/c626411e-531f-4dab-9a78-8103d92bbf1d.jpg

加文字水印效果