飞道的博客

来,我教你用Python做个音乐海报

324人阅读  评论(0)

前言

前段时间在一个朋友那么得到的灵感,想到可以用音乐播放页面作为一张海报图片。其实接下来要讲的和海报还是有差距的,而具体实现也只是简单的图片粘贴,但是在效果上还是不错的。效果图如下,希望大家喜欢:

左边是原图,右边是需要添加到中间的图,也是图的主角。其实如果直接用ps实现上面的图是非常简单的,反倒是用代码实现有点曲折,不过实现过程还是非常有趣的,希望这篇博客可以可以让你学到知识。

用Pillow创建圆形图

在上面的图片中,中间是一个圆形图片,而Pillow本身是没有提供生成圆形图片的方法(也可能是我没找到),所以就需要自己实现。在实现之前,我们先安装Pillow模块:

pip install pillow

要创建圆形图,我们先根据原图的大小,创建一个RGBA模式的透明图:

# 该方法传入三个参数,第一个为模式,第二个为大小的元组,第三个为颜色
im = Image.new('RGBA', (300, 300), (255, 255, 255, 0))

上述代码是创建了一个完全透明的300*300的图片,我们在该图片上绘制一个最大的圆:

# 获取绘制者
drawer = ImageDraw.Draw(im)
# 绘制一个黄色的圆,ellipse方法传入三个参数,第一个为包含该圆的最小正方形的区域,第二个为颜色,第三个为边宽
drawer.ellipse((0, 0, 300, 300), fill=(255, 255, 0), width=0)

生成图片如下:

我们准备一张300*300的正方形图片,然后遍历图片的每个像素,如果像素值的A==0(即像素不透明)那我们就将图片该区域的像素值设置为透明。

代码如下:

# 打开要转换成圆形的图片,我们事先把图片裁剪好
pic = Image.open(img_path).convert('RGBA')
# 遍历图片的每个像素
for i in range(300):
	for j in range(300):
    	# 获取该像素点的像素
     	r, g, b, a = im.getpixel((i, j))
        # 当rgb值不是黄色时,即像素值为透明时
        if (r, g, b) != (255, 255, 0):
            # 将原图的像素值设置为透明
        	pic.putpixel((i, j), (255, 255, 255, 0))

我们的pic就是圆形图片了,完整代码如下:

# 背景图中圆的直径
radius = 533
# 图片的大小
circle_size = (radius, radius)
def generate_circle_image(img_path):
    # 创建一个透明的正方形
    im = Image.new('RGBA', circle_size, (255, 255, 255, 0))
    # 获取绘画者
    drawer = ImageDraw.Draw(im)
    # 在透明的正方形上画一个黄色的圆
    drawer.ellipse((0, 0, circle_size[0], circle_size[1]), fill=(255, 255, 0), width=0)
    # 打开要转换成圆形的图片,我们事先把图片裁剪好
    pic = Image.open(img_path).convert('RGBA')
    # 修改图片大小,让图片和圆大小一样
    re_pic = pic.resize(circle_size, Image.ANTIALIAS)
    # 遍历图片的每个像素
    for i in range(circle_size[0]):
        for j in range(circle_size[1]):
            r, g, b, a = im.getpixel((i, j))
            if (r, g, b) != (255, 255, 0):
                re_pic.putpixel((i, j), (255, 255, 255, 0))
    return re_pic

在上面的方法中,我们改进了一些代码,之前我们需要使用指定大小的图片作为素材,现在只需要是正方形图片就可以了。

生成海报

原本我以为直接将圆形图片粘贴到背景图片上就可以了,但是试过之后发现效果和我想的不太一样,反正就是没成功,效果如下:

是粘贴上去了没错,而且图片也是透明效果,但是从这个效果来看粘贴只是像素替换,而不是图片叠加。于是我又想到了遍历像素的办法。

我通过特殊手段获取了上面正方形离左边的像素,和离上边的像素(其实就是用ps看了一下)。我尝试过图形检测和像素判断的办法,想自动识别中间圆的位置,但是效果不佳(反正就是失败了),所以只能无耻的用ps查看像素。

回到正题,我们用遍历像素的办法不需要遍历整个图片,只需要从(left, top)像素开始,遍历到(left+radius, top+radius)区域即可,也就是遍历正方形区域(left、top和radius都是我通过ps获取的)。

我们先将背景图片拷贝一份,然后在副本上进行粘贴。然后遍历粘贴后的图片,如果像素值为透明,我们就将原图该位置的像素替换至副本处,实现原理和上面一样,代码如下:

def generate_music_post(circle_im, bg_im):
 	"""
 		传入圆形图片和背景图片生成音乐海报
 		circle_im:圆形图片
 		bg_im:背景图片
 		return:生成的图片
 	"""
	# 拷贝副本
    bg_copy = bg_im.copy()
    # 将圆形图片粘贴到副本上
    bg_copy.paste(circle_im, (left, top))
    # 遍历像素正方形区域
    for i in range(left, left+radius):
        for j in range(top, top+radius):
            # 获取像素值
            color = bg_copy.getpixel((i, j))
            # 如果像素透明。color的值为(r,g,b,a),color[3]为a的值,即透明值
            if color[3] == 0:
                # 将原图像素替换至副本透明处
                bg_copy.putpixel((i, j), bg_im.getpixel((i, j)))
	# 返回合成后的图片
    return bg_copy

这样就完成了。完整代码如下:

from PIL import Image, ImageDraw

left = 273			# 圆离左边的距离
top = 573			# 圆离上边的距离
radius = 533		# 圆的直径
circle_size = (radius, radius)		# 圆的外接正方形的大小

def generate_circle_image(img_path):
    # 创建一个透明的正方形
    im = Image.new('RGBA', circle_size, (255, 255, 255, 0))
    # 获取绘画者
    drawer = ImageDraw.Draw(im)
    # 在透明的正方形上画一个黄色的圆
    drawer.ellipse((0, 0, circle_size[0], circle_size[1]), fill=(255, 255, 0), width=0)
    # 打开要转换成圆形的图片,我们事先把图片裁剪好
    pic = Image.open(img_path).convert('RGBA')
    re_pic = pic.resize(circle_size, Image.ANTIALIAS)

    # 遍历图片的每个像素
    for i in range(circle_size[0]):
        for j in range(circle_size[1]):
            r, g, b, a = im.getpixel((i, j))
            if (r, g, b) != (255, 255, 0):
                re_pic.putpixel((i, j), (255, 255, 255, 0))

    return re_pic

def generate_music_post(circle_im, bg_im):
 	"""
 		传入圆形图片和背景图片生成音乐海报
 		circle_im:圆形图片
 		bg_im:背景图片
 		return:生成的图片
 	"""
	# 拷贝副本
    bg_copy = bg_im.copy()
    # 将圆形图片粘贴到副本上
    bg_copy.paste(circle_im, (left, top))
    # 遍历像素正方形区域
    for i in range(left, left+radius):
        for j in range(top, top+radius):
            # 获取像素值
            color = bg_copy.getpixel((i, j))
            # 如果像素透明。color的值为(r,g,b,a),color[3]为a的值,即透明值
            if color[3] == 0:
                # 将原图像素替换至副本透明处
                bg_copy.putpixel((i, j), bg_im.getpixel((i, j)))
	# 返回合成后的图片
    return bg_copy

# 生成圆形图片
pic = generate_circle_image('girl.jpeg')
# 以RGBA模式读取背景图片
bg_im = Image.open('music.jpg').convert('RGBA')
# 生成音乐海报
music_post = generate_music_post(pic, bg_im)
music_post.show()

另外,这个例子还可以更加智能。我们可以使用OpenCV识别主体图片的人脸,然后根据人脸区域计算一个比较适合的正方形区域,这样我们就不必传入正方形(不过还要考虑人脸识别的精确度等问题)。

代码已上传GitHub地址如下: https://github.com/IronSpiderMan/MusicPost

素材图片也在里面。


转载:https://blog.csdn.net/ZackSock/article/details/104476939
查看评论
* 以上用户言论只代表其个人观点,不代表本网站的观点或立场