彩色图像高频与低频成分的分解

彩色图像高频与低频成分的分解

import torchgeometry
import cv2
import torch
import numpy as np
def calc_Freq(torch_img, kernel=3):
    if kernel == 3:
        sigma = 3
    elif kernel == 5:
        sigma = 1.5
    else:
        sigma = 1
    lowFreq = torchgeometry.image.gaussian_blur(
        torch_img, (kernel, kernel), (sigma, sigma)
    )
    highFreq = torch_img - lowFreq
    return lowFreq, highFreq