"""
@Description ：将Tif转为RGB
@Author ：hhx
@Date ：2022/5/9 14:49 
"""

import numpy as np
from PIL import Image, ImageEnhance
import os
from utils import *

def color_enhancement(im):
    # 色彩增强
    img = Image.fromarray(np.uint8(im))
    contrast = ImageEnhance.Contrast(img)
    img = contrast.enhance(1.5)
    contrast = ImageEnhance.Sharpness(img)
    img = contrast.enhance(2)
    contrast = ImageEnhance.Brightness(img)
    img = contrast.enhance(1.2)
    return img


if __name__ == '__main__':
    path = r'G:\哨兵2号数据\江边'
    c=0
    for file in os.listdir(path):
        if file.endswith('tif'):
            data = readGeoTIFF(os.path.join(path, file))
            img = np.stack((data[3], data[2], data[1])).transpose(1, 2, 0)
            print(img.shape)
            img1 = Image.fromarray(np.uint8(nor(img)))
            # img1 = color_enhancement(img1)
            img1.save(os.path.join('JB','{}_{}.jpg'.format(rename(c), file.split('.')[0])))
            c+=1