tuoheng_alg_fangke_discard_20240715/util/ImageUtils.py

import time
from io import BytesIO

import cv2
import requests
from PIL import Image, ImageDraw, ImageFont
import numpy as np

'''
    文字水印
'''


class TextWaterMark():
    def __init__(self):
        self.color_dict = {
            # R G B
            # 网址查看：https://tool.oschina.net/commons?type=3
            # 网址查看：http://tools.jb51.net/static/colorpicker/
            'white': (255, 255, 255, 255),
            'black': (0, 0, 0, 255),
            'gray': (205, 201, 201, 255),
            'red': (255, 0, 0, 255),
            'yellow': (255, 215, 0, 255),
            'blue': (0, 0, 170, 255),
            'purple': (205, 105, 201, 255),
            'green': (0, 205, 0, 255)
        }
        self.position_list = [1, 2, 3, 4]

    """
        普通照片水印
        params:
            image:图片
            text:水印文字
            position:水印位置
                    1：左上
                    2：右上
                    3：右下
                    4：左下
            fontface: 字体        
            fontsize:字体大小
            fontcolor:字体颜色
                    [white, black, gray, red, yellow, blue, purple, green]
    """

    def common_water(self, image, text, position=1, fontface='../font/simsun.ttc', fontsize=20, fontcolor='black'):
        flag = False
        if isinstance(image, np.ndarray):  # 判断是否OpenCV图片类型
            flag = True
            image = Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGRA2RGBA))
        if position not in self.position_list:
            position = 1
        w, h = image.size[:2]
        keys = self.color_dict.keys()
        if fontcolor not in keys:
            fontcolor = 'black'
        color = self.color_dict[fontcolor]
        fnt = ImageFont.truetype(fontface, fontsize)
        im = image.convert('RGBA')
        mask = Image.new('RGBA', im.size, (0, 0, 0, 0))
        d = ImageDraw.Draw(mask)
        size_w, size_h = d.textsize(text, font=fnt)
        if position == 1:
            weizhi = (w * 0.1, h * 0.1)
        elif position == 2:
            weizhi = (w * 0.9 - size_w, h * 0.1)
        elif position == 3:
            weizhi = (w * 0.9 - size_w, h * 0.9 - size_h)
        else:
            weizhi = (w * 0.1, h * 0.9 - size_h)
        # position 为左上角位置
        d.text(weizhi, text, font=fnt, fill=color)
        out = Image.alpha_composite(im, mask)
        if flag:
            out = cv2.cvtColor(np.asarray(out), cv2.COLOR_BGRA2RGBA)
        return out

    """
        半透明水印，布满整张图，并且自动旋转45°
        params:
            image:图片
            text:文字
            fontsize:文字大小
    """

    def fill_water(self, image, text, fontsize):
        flag = False
        if isinstance(image, np.ndarray):  # 判断是否OpenCV图片类型
            flag = True
            image = Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGRA2RGBA))
        font = ImageFont.truetype('../font/simsun.ttc', fontsize)
        # 添加背景
        new_img = Image.new('RGBA', (image.size[0] * 3, image.size[1] * 3), (255, 255, 255, 255))
        new_img.paste(image, image.size)

        # 添加水印
        font_len = len(text)
        rgba_image = new_img.convert('RGBA')
        text_overlay = Image.new('RGBA', rgba_image.size, (0, 0, 0, 0))
        image_draw = ImageDraw.Draw(text_overlay)

        for i in range(0, rgba_image.size[0], font_len * 40 + 100):
            for j in range(0, rgba_image.size[1], 200):
                # print(f'i:{i}, j:{j}, text:{text}, font:{font}')
                image_draw.text((i, j), text, font=font, fill=(0, 0, 0, 50))
        text_overlay = text_overlay.rotate(-45)
        image_with_text = Image.alpha_composite(rgba_image, text_overlay)

        image_with_text = image_with_text.crop((image.size[0], image.size[1], image.size[0] * 2, image.size[1] * 2))
        if flag:
            image_with_text = cv2.cvtColor(np.asarray(image_with_text), cv2.COLOR_BGRA2RGBA)
        return image_with_text


class PictureWaterMark:

    def __init__(self):
        self.logo = cv2.imread("./image/logo.png", -1)

    def common_water(self, image, logo):
        width, height = image.shape[1], image.shape[0]
        mark_width, mark_height = logo.shape[1], logo.shape[0]
        rate = int(width * 0.2) / mark_width
        logo_new = cv2.resize(logo, None, fx=rate, fy=rate, interpolation=cv2.INTER_NEAREST)
        position = (int(width * 0.95 - logo_new.shape[1]), int(height * 0.95 - logo_new.shape[0]))
        b = Image.new('RGBA', (width, height), (0, 0, 0, 0))  # 创建新图像：透明'
        a = Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
        watermark = Image.fromarray(cv2.cvtColor(logo_new, cv2.COLOR_BGRA2RGBA))
        # 图片旋转
        # watermark = watermark.rotate(45)
        b.paste(a, (0, 0))
        b.paste(watermark, position, mask=watermark)
        return cv2.cvtColor(np.asarray(b), cv2.COLOR_BGR2RGB)

    def common_water_1(self, image, logo, alpha=1):
        h, w = image.shape[0], image.shape[1]
        if w >= h:
            rate = int(w * 0.1) / logo.shape[1]
        else:
            rate = int(h * 0.1) / logo.shape[0]
        mask = cv2.resize(logo, None, fx=rate, fy=rate, interpolation=cv2.INTER_NEAREST)
        mask_h, mask_w = mask.shape[0], mask.shape[1]
        mask_channels = cv2.split(mask)
        dst_channels = cv2.split(image)
        # b, g, r, a = cv2.split(mask)
        # 计算mask在图片的坐标
        ul_points = (int(h * 0.95) - mask_h, int(w - h * 0.05 - mask_w))
        dr_points = (int(h * 0.95), int(w - h * 0.05))
        for i in range(3):
            dst_channels[i][ul_points[0]: dr_points[0], ul_points[1]: dr_points[1]] = dst_channels[i][
                                                                                      ul_points[0]: dr_points[0],
                                                                                      ul_points[1]: dr_points[1]] * (
                                                                                              255.0 - mask_channels[3] * alpha) / 255
            dst_channels[i][ul_points[0]: dr_points[0], ul_points[1]: dr_points[1]] += np.array(
                mask_channels[i] * (mask_channels[3] * alpha / 255), dtype=np.uint8)
        dst_img = cv2.merge(dst_channels)
        return dst_img

#差值感知算法
def dHash(image):
    #缩放9*8
    image=cv2.resize(image,(9,8),interpolation=cv2.INTER_CUBIC)
    #转换灰度图
    image=cv2.cvtColor(image,cv2.COLOR_BGR2GRAY)
    #     print(image.shape)
    hash=[]
    #每行前一个像素大于后一个像素为1，相反为0，生成哈希
    for i in range(8):
        for j in range(8):
            if image[i,j]>image[i,j+1]:
                hash.append(1)
            else:
                hash.append(0)
    return hash

#计算汉明距离
def Hamming_distance(hash1,hash2):
    num = 0
    for index in range(len(hash1)):
        if hash1[index] != hash2[index]:
            num += 1
    return num

def url2Array(url):
    response = requests.get(url)
    image = Image.open(BytesIO(response.content))
    image1= np.array(image)
    img_bgr = cv2.cvtColor(image1, cv2.COLOR_RGB2BGR)
    return img_bgr


if __name__ == '__main__':
    # img = cv2.imread("../test/a.jpg", -1)
    # fontcolor = 'yellow'
    #
    # water = TextWaterMark()
    # text = "hello world"
    #
    # # fill_img = water.common_water(img, text, position=4, fontface='../font/庞门正道标题体2.0增强版.ttf', fontsize=20, fontcolor='black')
    # fill_img = water.fill_water(img, text, 20)
    # # 一定要保存为png格式
    # cv2.imshow('result', fill_img)
    # cv2.waitKey(111111110)
    # print('finish')
    pic = PictureWaterMark()
    image = cv2.imread("a.jpg")
    logo = cv2.imread("../image/logo.png", -1)
    # print(image, logo)
    start = time.time()
    frame = pic.common_water(image, logo)
    print(time.time() - start)
    start1 = time.time()
    frame1 = pic.common_water_1(image, logo)
    # cv2.imwrite("watermarked.jpg", frame1)
    print(time.time() - start1)
    # cap = cv2.VideoCapture("../data/111111.mp4")
    # logo = cv2.imread("../image/logo.png", -1)
    # while True:
    #         is_opened, frame = cap.read()
    #         frame = pic.common_water(frame, logo)
    #         cv2.imshow('frame', frame)
    #         cv2.waitKey(1)  # 等待输入任何按键
    # # 关闭
    # cap.release()