使用OpenCV拼接图片，表白神器，我靠它成功找对来女朋友

本文链接：https://blog.csdn.net/weixin_42263486/article/details/108611618

使用OpenCV拼接图片，表白神器，我靠它成功找对来女朋友

程序员没浪漫？

呜呜呜，总是说程序员没有浪漫，谁说的，看我直接做个图片墙不直接拿下？我把她的生活照偷偷的收集起来，然后用Python写了一个图片拼器，直接把女朋友看哭了

所用的包

cv2处理图片的包
matpltlib图片展示包
numpy矩阵处理包

看看效果

在这里插入图片描述

有了它女朋友看哭了

代码实现

import cv2
import matplotlib.pyplot as plt
import numpy as np

# 读入你自己的原图
im = cv2.imread('111.jpg', cv2.IMREAD_COLOR)
im = cv2.cvtColor(im, cv2.COLOR_BGR2RGB)
im_h,im_w,im_c = im.shape

import collections
import random

# 从cifar中读入数据，将每张图片的平均像素值作为这些照片的key，那么我们通过key就可以直接通过像素快速找到自己想要替换的照片
import pickle
pixel_dict = collections.defaultdict(list)
with open('./cifar-10-batches-py/data_batch_1', 'rb') as f:
    imgs = pickle.load(f, encoding='bytes')[b'data']
for img in imgs:
    pixel = int(np.average(img))
    pixel_dict[pixel].append(img.reshape((3, 32, 32)).transpose((1,2,0)))

# 使用28*28的图片进行拼接
joint_img = np.empty((32*im_h, im_w*32, 3)).astype('float32')
pre_pixel = None
print(joint_img.shape)
for h_index, pixel_h in enumerate(np.mean(im, axis=2)):
    for w_index, pixel_w in enumerate(pixel_h):
        choice_pixel = pixel_dict[int(pixel_w)]
        if not choice_pixel:
            choice_pixel = pre_pixel
        else:
            pre_pixel = choice_pixel
        im_pixel = random.choice(choice_pixel)
        joint_img[h_index*32:(h_index+1)*32, w_index*32:(w_index+1)*32] = im_pixel

# 保存
plt.imsave('a.jpg', joint_img/255., cmap='gray')