1. 项目结构
2. utils包中的convert.py
from PIL import Image
import os
def convert_tiff_to_jpg(folder_path):
file_list = os.listdir(folder_path)
for file_name in file_list:
if file_name.endswith('.tif'):
input_path = os.path.join(folder_path, file_name)
output_path = os.path.join(folder_path, file_name.rsplit('.', 1)[0] + '.jpg')
image = Image.open(input_path)
image.convert('RGB').save(output_path, format='JPEG')
image.close()
def convert_png_to_tiff(folder_path):
file_list = os.listdir(folder_path)
for file_name in file_list:
if file_name.endswith('.png'):
input_path = os.path.join(folder_path, file_name)
output_path = os.path.join(folder_path, file_name.rsplit('.', 1)[0] + '.tif')
image = Image.open(input_path)
image.save(output_path, format='TIFF')
image.close()
def convert_jpg_to_png(folder_path):
for filename in os.listdir(folder_path):
if filename.endswith(".jpg"):
source_path = os.path.join(folder_path, filename)
target_path = os.path.join(folder_path, os.path.splitext(filename)[0] + ".png")
image = Image.open(source_path)
image.save(target_path, "PNG")
image.close()
def convert_png_to_jpg(folder_path):
for filename in os.listdir(folder_path):
if filename.endswith(".png"):
source_path = os.path.join(folder_path, filename)
target_path = os.path.join(folder_path, os.path.splitext(filename)[0] + ".jpg")
image = Image.open(source_path)
image.save(target_path, "JPEG")
image.close()
3. main.py
from utils.convert import *
if __name__ == "__main__":
folder_path = '../Augement_schedular/data'
convert_png_to_jpg(folder_path)