Python 實現(xiàn)給圖片加文字或logo水印

更新時間：2021年11月22日 09:26:14 作者：劍客阿良_ALiang

本文主要為大家介紹了給圖片添加文字或者logo圖片水印的python工具，從而打造你的專屬圖片。代碼簡潔易懂，感興趣的小伙伴可以了解一下

前言

本文提供給圖片添加文字或者logo圖片水印的python工具，打造專屬圖片。

環(huán)境依賴

ffmpeg環(huán)境安裝，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 os
import uuid
from 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
 
 
# 圖片添加logo
def 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 result
 
 
def _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方法給圖片添加文字水印方法，主要參數(shù)為：圖片路徑、輸出目錄、

水印文字、字體名稱、字體大小（默認100）、文字左上角坐標(biāo)（默認0:0）、文字顏色（默認藍色）、是否需要背景（默認需要為1，不需要為0）、背景色（默認白色）。

2、image_add_logo方法給圖片添加logo，主要參數(shù)為：圖片路徑、輸出目錄、logo圖片地址、文字左上角坐標(biāo)（默認0:0）。

3、注意logo地址如果有類似C:/這種windows盤的路徑情況，需要對":"進行轉(zhuǎn)義。后面驗證的時候，可以看看我給的參數(shù)。

4、文件名為了避免重復(fù)，采用了uuid作為文件名。

5、圖片后綴格式校驗只有兩種，按需添加即可。

驗證一下

準(zhǔn)備的素材如下，分別為圖片與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地址路徑有什么不同。

執(zhí)行結(jié)果

PyDev console: starting.
Python 3.6.13 |Anaconda, Inc.| (default, Mar 16 2021, 11:37:27) [MSC v.1916 64 bit (AMD64)] on win32
runfile('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.jpg
ffmpeg version n4.3.1-20-g8a2acdc6da Copyright (c) 2000-2020 the FFmpeg developers
  built with gcc 9.3-win32 (GCC) 20200320
  configuration: --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=-lgomp
  libavutil      56. 51.100 / 56. 51.100
  libavcodec     58. 91.100 / 58. 91.100
  libavformat    58. 45.100 / 58. 45.100
  libavdevice    58. 10.100 / 58. 10.100
  libavfilter     7. 85.100 /  7. 85.100
  libswscale      5.  7.100 /  5.  7.100
  libswresample   3.  7.100 /  3.  7.100
  libpostproc    55.  7.100 / 55.  7.100
Input #0, image2, from 'C:/Users/huyi/Desktop/1.jpg':
  Duration: 00:00:00.04, start: 0.000000, bitrate: 181106 kb/s
    Stream #0:0: Video: mjpeg (Progressive), yuvj444p(pc, bt470bg/unknown/unknown), 3840x2160, 25 tbr, 25 tbn, 25 tbc
Stream mapping:
  Stream #0:0 -> #0:0 (mjpeg (native) -> mjpeg (native))
Press [q] to stop, [?] for help
Fontconfig 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.100
    Stream #0:0: Video: mjpeg, yuvj444p(pc), 3840x2160, q=2-31, 200 kb/s, 25 fps, 25 tbn, 25 tbc
    Metadata:
      encoder         : Lavc58.91.100 mjpeg
    Side data:
      cpb: bitrate max/min/avg: 0/0/200000 buffer size: 0 vbv_delay: N/A
frame=    1 fps=0.0 q=11.5 Lsize=N/A time=00:00:00.04 bitrate=N/A speed=0.158x    
video:634kB audio:0kB subtitle:0kB other streams:0kB global headers:0kB muxing overhead: unknown
C:/Users/huyi/Desktop/2226d7e0-5166-4ffd-ba95-9ca7d9d6f72d.jpg
ffmpeg -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.jpg
ffmpeg version n4.3.1-20-g8a2acdc6da Copyright (c) 2000-2020 the FFmpeg developers
  built with gcc 9.3-win32 (GCC) 20200320
  configuration: --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=-lgomp
  libavutil      56. 51.100 / 56. 51.100
  libavcodec     58. 91.100 / 58. 91.100
  libavformat    58. 45.100 / 58. 45.100
  libavdevice    58. 10.100 / 58. 10.100
  libavfilter     7. 85.100 /  7. 85.100
  libswscale      5.  7.100 /  5.  7.100
  libswresample   3.  7.100 /  3.  7.100
  libpostproc    55.  7.100 / 55.  7.100
Input #0, image2, from 'C:/Users/huyi/Desktop/1.jpg':
  Duration: 00:00:00.04, start: 0.000000, bitrate: 181106 kb/s
    Stream #0:0: Video: mjpeg (Progressive), yuvj444p(pc, bt470bg/unknown/unknown), 3840x2160, 25 tbr, 25 tbn, 25 tbc
Stream 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 correctly
Output #0, image2, to 'C:/Users/huyi/Desktop/c626411e-531f-4dab-9a78-8103d92bbf1d.jpg':
  Metadata:
    encoder         : Lavf58.45.100
    Stream #0:0: Video: mjpeg, yuvj420p(pc), 3840x2160, q=2-31, 200 kb/s, 25 fps, 25 tbn, 25 tbc
    Metadata:
      encoder         : Lavc58.91.100 mjpeg
    Side data:
      cpb: bitrate max/min/avg: 0/0/200000 buffer size: 0 vbv_delay: N/A
[Parsed_movie_0 @ 000001d6cc40bd80] EOF timestamp not reliable
frame=    1 fps=0.0 q=11.4 Lsize=N/A time=00:00:00.04 bitrate=N/A speed=0.176x    
video:455kB audio:0kB subtitle:0kB other streams:0kB global headers:0kB muxing overhead: unknown
C:/Users/huyi/Desktop/c626411e-531f-4dab-9a78-8103d92bbf1d.jpg

加文字水印效果