从指定的源文件夹中随机选择170张png
格式图片,然后将这些图片转换为JPG格式,并保存到另一个指定的目标文件夹中。
import os
import shutil
import random
from PIL import Image
# 设置源文件夹和目标文件夹路径
source_folder = 'F:/acquisitions' # 替换为你的源文件夹路径
destination_folder = 'F:/results' # 替换为你的目标文件夹路径
# 确保目标文件夹存在
if not os.path.exists(destination_folder):
os.makedirs(destination_folder)
# 函数用于遍历文件夹并收集所有以rgb.png结尾的图片的路径
def collect_rgb_png_files(folder):
rgb_png_files = []
for root, dirs, files in os.walk(folder):
for file in files:
if file.lower().endswith('rgb.png'):
rgb_png_files.append(os.path.join(root, file))
return rgb_png_files
# 收集所有以rgb.png结尾的图片
rgb_png_files = collect_rgb_png_files(source_folder)
# 随机选择170张图片
selected_files = random.sample(rgb_png_files, min(170, len(rgb_png_files)))
# 函数用于生成唯一的文件名
def unique_filename(base_folder, ext):
counter = 1
while True:
new_file_name = f"image_{counter:04d}{ext}"
full_path = os.path.join(base_folder, new_file_name)
if not os.path.exists(full_path):
return new_file_name
counter += 1
# 将图片转换为JPG格式并重命名后保存到目标文件夹
for file_path in selected_files:
with Image.open(file_path) as img:
# 转换图片格式
img = img.convert('RGB')
# 生成新的文件名
new_file_name = unique_filename(destination_folder, '.jpg')
# 保存图片到目标文件夹
img.save(os.path.join(destination_folder, new_file_name))
print(f"已成功转换并保存 {len(selected_files)} 张图片到 '{destination_folder}'。")