基于小波变换的多聚焦图像融合算法——Python实现

# 毕设要做的东西，之前在CSDN上查资料，小波变换这部分全是MATLAB，而且没几个写的靠谱的。Python的pywavelet库资料太少了，我就把自己做的东西传上来吧。

# 什么你问我为什么不用MATLAB？嗯嗯，我讨厌这个软件，就这样。

# -*- coding: utf-8 -*-

import pywt
import numpy as np
import matplotlib.pyplot as plt
from PIL import Image
import cv2

'''
论文中的两种方案：
1，对低频分量的所有像素点计算其局部方差，每张图所有点的方差加起来除以两张图所有点加起来，得到两张图的权重
融合图像每个像素点的值为两张图对应像素点的值加权平均，这个权就是上面算出来的权值。
2，对每个高频分量的像素点用canny算子进行边缘提取，再对边缘图像的每一个像素点计算其局部方差，得到方差图像
融合图像每个像素点的值为方差图片中对应像素点取值较大的那张图片的像素点。
3，效果不是很好，在小波分块的边缘有明显的灰度跳变(就是有些论文里说的分块效应)，but why?
4，已找到原因，要求的不是某一个点对全图的方差，而是在某点附近开个小窗口求窗口的局部方差
5，现用局部方差的方法对多聚焦图像效果非常完美
'''

def imgOp