Python和NumPy开发的实用小工具

一、图片转黑白:使用NumPy的灰度化功能将彩色图片转为黑白图片

import numpy as np 
from PIL import Image

img = Image.open('image.jpg')
img_arr = np.array(img)二
gray_img = np.zeros((img_arr.shape[0], img_arr.shape[1]))

for i in range(img_arr.shape[0]):
    for j in range(img_arr.shape[1]):
        gray_img[i, j] = int(np.average(img_arr[i, j]))

gray_img = Image.fromarray(gray_img.astype('uint8')).convert('RGB')
gray_img.save('gray_image.jpg')

二、添加水印:使用NumPy在图片上添加半透明的水印

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

img = Image.open('image.jpg')
img_arr = np.array(img)

watermark = Image.new('RGBA', img.size, (0, 0, 0, 0))
draw = ImageDraw.Draw(watermark)
font = ImageFont.truetype('font.ttf', 36)
draw.text((img.size[0] - 300, img.size[1] - 50), 'Watermark', font=font)

watermark_arr = np.array(watermark) 
img_arr = np.dstack((img_arr, watermark_arr))

watermarked_img = Image.fromarray(img_arr)
watermarked_img.save('watermarked.jpg')

三、 图片马赛克效果:使用NumPy给图片添加马赛克效果

import numpy as np 
from PIL import Image

size = 10  # 马赛克块大小

img = Image.open('image.jpg')
img_arr = np.array(img)

mosaic_img = np.empty((img_arr.shape[0], img_arr.shape[1], 3), dtype=np.uint8)

for i in range(0, img_arr.shape[0], size): 
    for j in range(0, img_arr.shape[1], size):
        rgb = [np.average(img_arr[i:i+size, j:j+size, c]) for c in range(3)]
        mosaic_img[i:i+size, j:j+size, :] = rgb  

mosaic_img = Image.fromarray(mosaic_img)
mosaic_img.save('mosaic.jpg')