【Python】爬取微信好友头像完成马赛克拼图～

代码已重新整理上传到了我的KLab，为了更好的阅读体验可点击🔗【千图成像】爬取微信好友头像完成马赛克拼图🧩查看～

马赛克拼图

何谓马赛克拼图（千图成像），简单来说就是将若干小图片平凑成为一张大图，如下图路飞一样，如果放大看你会发现里面都是一些海贼王里面的图片。
LUFFY

Our Tragets

爬取所有微信好友的头像🕺🏻🕺🏻🕺🏻
将所有微信好友头像拼凑成一张图片🏙🏙🏙
然后就可以去朋友圈愉快的装逼了🤪🤪🤪

Requirements

其实整个项目很小，项目总共代码量不过100行左右。

爬取微信头像依赖第三方库itchat；
马赛克拼图依赖依赖numpy和PIL库。

Content

爬取微信好友头像

我这边是用的所有微信好友头像作为的数据源，你们如果有其他的数据源也可以的，可以直接跳过这步。
爬取微信好友头像我使用的是itchat，里面已经有封装好了的API，直接调用就可以，将所有的好友头像保存到一个文件夹供后期使用。

代码部分

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Author  : AwesomeTang
# @File    : Wechat_Icon.py
# @Version : Python 3.7
# @Time    : 2019-06-29 23:35import os
import itchatitchat.login()friends = itchat.get_friends(update=True)base_folder = 'wechat'
if os.path.isdir(base_folder):pass
else:os.mkdir(base_folder)for item in friends:img = itchat.get_head_img(item['UserName'])# 使用用户昵称作为文件名path = os.path.join(base_folder, '{}.jpg'.format(item['NickName'].replace('/', '')))with open(path, 'wb') as f:f.write(img)print('{} 写入完成...'.format(item['NickName']))

实现效果

马赛克拼图

思路梳理

选好你需要拼凑的图片，将其切割为若干小块，切割的越细生成的图片效果会更好。
gay里gay气的示意图：
分别去我们之前保存的图片中找与与之最相似的，最后将其拼接完成。

说起来好像很简单，但实际操作起来只能…

最相似的图片

其实困难的地方很明显，那就是我们如何从一堆图片中找出最相似的那张。
我们可以分为两个步骤：

颜色相似

这个应该不难理解，我在代码中实现了灰度图像和RGB通道图像的筛选方法：

灰度图像：
直接计算所有像素灰度值的平均值，取最接近n个图像供后期再次筛选；
RGB通道：
分别计算R,G,B的平均值，对于一个图像我们得到的是一个类似与[20, 30,40]的数组，然后我们计算欧式距离，取最接近n个图像供后期再次筛选。

结构相似

为什么还需要结构相似，举个例子：

如果单纯按照上述方法去判定相似，那上图中的图A和图B肯定是最相似的，所以我们不能单纯的因为两张图片中包含的颜色差不多就去判断为最相似，还需要去判断颜色的“位置”也要相似。
这部分实现方法参考了阮一峰的博客，具体逻辑如下：

只需要提取图片结构，颜色意义不大，为计算简便，我们直接将所有图片转为灰度通道；
将每张头像resize为（8，8），然后计算所有像素值的平均值。
我们总共有64（即 $8 * 8$ ）个像素点，分别去与平均值比较大小，高于平均值的记为1，小于平均值的记为0，这样我们每张图片都会得到一个长度为64类似[0,1,1,0,1,0…0,1,1]的‘编码’。
对于切割的小图片我们也进行上述操作，注意要保证是同一顺序（譬如从左上角到右下角），然后分别去与每个头像的‘编码’进行比较，这边在阮一峰的博客中是采用的计算汉明距离，我这边使用的就直接通过np.equal()计算相同的点了，取相同位数最多的那张头像即为最相似的图片。

我在代码中是先筛选颜色最接近的50张图片，然后再在这50中图片去寻找结构最相似的图片，最后实现效果如下：

RGB通道
因为图片源只有500来张，效果只能说凑合，勉强能看出来是啥。
灰度
)
放大之后是这个效果：

然后便能去朋友圈愉快的装逼了😎😎😎

Talk is cheap, show me the code.

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Author  : AwesomeTang
# @Version : Python 3.7
# @Time    : 2019-06-23 13:52from PIL import Image
import os
import numpy as np
from tqdm import tqdmclass Config:corp_size = 40filter_size = 20num = 100class PicMerge:def __init__(self, pic_path, mode='RGB', pic_folder='wechat'):if mode.upper() not in ('RGB', 'L'):raise ValueError('Only accept "RGB" or "L" MODE, but we received "{}".'.format(self.mode))else:self.mode = mode.upper()print('Coding for every picture in folder "{}".'.format(pic_folder))self.mapping_table, self.pictures = self.mapping_table(pic_folder)self.picture = self.resize_pic(pic_path).convert(self.mode)@staticmethoddef resize_pic(pic_path):picture = Image.open(pic_path)width, height = picture.sizeto_width = Config.corp_size * Config.numto_height = ((to_width / width) * height // Config.corp_size) * Config.corp_sizepicture = picture.resize((int(to_width), int(to_height)), Image.ANTIALIAS)return picturedef merge(self):width, height = self.picture.sizew_times, h_times = int(width / Config.corp_size), int(height / Config.corp_size)picture = np.array(self.picture)print('Corp & Merge...')for i in tqdm(range(w_times), desc='CORP'):for j in range(h_times):if self.mode == 'L':section = picture[j * Config.corp_size:(j + 1) * Config.corp_size,i * Config.corp_size:(i + 1) * Config.corp_size]section_mean = section.mean()candidate = sorted([(key_, abs(np.array(value_).mean() - section_mean))for key_, value_ in self.pictures.items()],key=lambda item: item[1])[:Config.filter_size]most_similar = self.structure_similarity(section, candidate)picture[j * Config.corp_size:(j + 1) * Config.corp_size,i * Config.corp_size:(i + 1) * Config.corp_size] = most_similarelif self.mode == 'RGB':section = picture[j * Config.corp_size:(j + 1) * Config.corp_size,i * Config.corp_size:(i + 1) * Config.corp_size, :]candidate = self.color_similarity(section)most_similar = self.structure_similarity(section, candidate)picture[j * Config.corp_size:(j + 1) * Config.corp_size,i * Config.corp_size:(i + 1) * Config.corp_size, :] = most_similarpicture = Image.fromarray(picture)picture.show()picture.save('result.jpg')print('Work Done...')def structure_similarity(self, section, candidate):section = Image.fromarray(section).convert('L')one_hot = self.pic_code(np.array(section.resize((8, 8), Image.ANTIALIAS)))candidate = [(key_, np.equal(one_hot, self.mapping_table[key_]).mean()) for key_, _ in candidate]most_similar = max(candidate, key=lambda item: item[1])return self.pictures[most_similar[0]]def color_similarity(self, pic_slice, top_n=Config.filter_size):slice_mean = self.rgb_mean(pic_slice)diff_list = [(key_, np.linalg.norm(slice_mean - self.rgb_mean(value_)))for key_, value_ in self.pictures.items()]filter_ = sorted(diff_list, key=lambda item: item[1])[:top_n]return filter_@staticmethoddef rgb_mean(rgb_pic):"""if picture is RGB channel, calculate average [R, G, B]."""r_mean = np.mean(rgb_pic[:, :, 0])g_mean = np.mean(rgb_pic[:, :, 1])b_mean = np.mean(rgb_pic[:, :, 2])val = np.array([r_mean, g_mean, b_mean])return valdef mapping_table(self, pic_folder):"""What this function do?1. transverse every image in PIC_FOLDER;2. resize every image in (8, 8) and covert into GREY;3. CODE for every image, CODE like [1, 0, 1, 1, 0....1]4. build a dict to gather all image and its CODE.:param pic_folder: path of pictures folder.:return: a dict"""suffix = ['jpg', 'jpeg', 'JPG', 'JPEG', 'gif', 'GIF', 'png', 'PNG']if not os.path.isdir(pic_folder):raise OSError('Folder [{}] is not exist, please check.'.format(pic_folder))pic_list = os.listdir(pic_folder)results = {}pic_dic = {}for idx, pic in tqdm(enumerate(pic_list), desc='CODE'):if pic.split('.')[-1] in suffix:path = os.path.join(pic_folder, pic)try:img = Image.open(path).resize((Config.corp_size, Config.corp_size), Image.ANTIALIAS)results[idx] = self.pic_code(np.array(img.convert('L').resize((8, 8), Image.ANTIALIAS)))if self.mode == 'RGB':pic_dic[idx] = np.array(img.convert(self.mode))else:pic_dic[idx] = np.array(img.convert(self.mode))except OSError:passreturn results, pic_dic@staticmethoddef pic_code(image: np.ndarray):"""To make a one-hot code for IMAGE.AVG is mean of the array(IMAGE).Traverse every pixel of IMAGE, if the pixel value is more then AVG, make it 1, else 0.:param image: an array of picture:return: A sparse list with length [picture's width * picture's height]."""width, height = image.shapeavg = image.mean()one_hot = np.array([1 if image[i, j] > avg else 0 for i in range(width) for j in range(height)])return one_hotif __name__ == "__main__":P = PicMerge(pic_path='海贼王.jpeg', mode='RGB')P.merge()