在原始路径 中区分 JPG 和 PNG 图片，并将它们保存到该路径下的不同文件

import os
import shutil

# 原始图片文件夹路径
source_folder = r'D:\YUEYUE\SAR picture\SARimage_data\SARimage_data\train\palsar'

# 创建存放 JPG 和 PNG 图片的子文件夹路径
dest_folder_jpg = os.path.join(source_folder, 'JPGs')
dest_folder_png = os.path.join(source_folder, 'PNGs')

# 如果子文件夹不存在，则创建它们
if not os.path.exists(dest_folder_jpg):
    os.makedirs(dest_folder_jpg)
if not os.path.exists(dest_folder_png):
    os.makedirs(dest_folder_png)

# 遍历原始文件夹中的所有文件
for filename in os.listdir(source_folder):
    file_path = os.path.join(source_folder, filename)
    # 根据文件扩展名将文件移动到相应的子文件夹
    if filename.endswith(".jpg") or filename.endswith(".jpeg"):
        shutil.move(file_path, os.path.join(dest_folder_jpg, filename))
    elif filename.endswith(".png"):
        shutil.move(file_path, os.path.join(dest_folder_png, filename))

print("JPG and PNG images have been sorted into separate folders.")