19 lines
825 B
Python
19 lines
825 B
Python
from PIL import Image
|
|
import os
|
|
|
|
input_folder = r'D:\1BNS_projects\yolo_safehat_num_v8\datasets\2m' #要调整的图片所在的文件夹名称
|
|
output_folder = r'D:\1BNS_projects\yolo_safehat_num_v8\datasets\2m1' # 调整后的图片所放的文件夹名称
|
|
if not os.path.exists(output_folder):
|
|
os.makedirs(output_folder)
|
|
new_size = (640, 640) # 调整为256*256
|
|
|
|
for filename in os.listdir(input_folder):
|
|
# Check if the file is an image
|
|
if filename.endswith('.png') or filename.endswith('.jpg'):
|
|
# Open the image and resize it
|
|
img = Image.open(os.path.join(input_folder, filename))
|
|
img = img.resize(new_size)
|
|
|
|
# Save the resized image to the output folder
|
|
new_filename = os.path.splitext(filename)[0] + '.jpg'
|
|
img.save(os.path.join(output_folder, new_filename)) |