生成退化的人脸图像代码

import cv2
import math
import random
import numpy as np
from dataset import gaussian_kernels as gaussian_kernels
import os

# degradation configurations
blur_kernel_size = 41
kernel_list = ['iso', 'aniso']
kernel_prob = [0.5, 0.5]
# Small degradation
blur_sigma = [0.1, 10]
downsample_range = [1, 12]
noise_range = [0, 15]
jpeg_range = [60, 100]
# Large degradation
blur_sigma_large = [1, 15]
downsample_range_large = [4, 30]
noise_range_large = [0, 20]
jpeg_range_large = [30, 80]
in_size = 512

img_path = '/gt/'
new_img_path = '/input/'
files = os.listdir(img_path)
for img_file in files:
    dst_img = os.path.join(os.path.abspath(new_img_path),img_file)
    img_gt = cv2.imread( os.path.join(img_path,img_file) )
    gt_size = img_gt.shape[0]
    size = (512, 512)
    # generate in image
    img_in = img_gt
    # gaussian blur
    kernel = gaussian_kernels.random_mixed_kernels(
        kernel_list,
        kernel_prob,
        blur_kernel_size,
        blur_sigma,
        blur_sigma,
        [-math.pi, math.pi],
        noise_range=None)
    img_in = cv2.filter2D(img_in, -1, kernel)

    # downsample
    scale = np.random.uniform(downsample_range[0], downsample_range[1])
    img_in = cv2.resize(img_in, (int(gt_size // scale), int(gt_size // scale)),interpolation=cv2.INTER_LINEAR)

    # noise
    if noise_range is not None:
        noise_sigma = np.random.uniform(noise_range[0] / 255., noise_range[1] / 255.)
        noise = np.float32(np.random.randn(*(img_in.shape))) * noise_sigma
        img_in = img_in + noise
        img_in = np.clip(img_in, 0, 1)

    # jpeg
    if jpeg_range is not None:
        jpeg_p = np.random.uniform(jpeg_range[0], jpeg_range[1])
        encode_param = [int(cv2.IMWRITE_JPEG_QUALITY), jpeg_p]
        _, encimg = cv2.imencode('.jpg', img_in * 255., encode_param)
        img_in = np.float32(cv2.imdecode(encimg, 1)) / 255.

    # resize to in_size
    img_in = cv2.resize(img_in, (in_size, in_size), interpolation=cv2.INTER_LINEAR)
    cv2.imwrite(dst_img, img_in)
    print(img_file)

print("over!")