#制作数据集时,对高分辨率的图片进行压缩#
import cv2
import os
def compress_folder(folder):
for filename in os.listdir(images_folder):
if filename.endswith('.jpg') or filename.endswith('.jpeg') or filename.endswith('.png'):
img_path = os.path.join(images_folder, filename)
img = cv2.imread(img_path)
new_path = os.path.join(images_folder, "compressed_" + filename)
cv2.imwrite(new_path, img, [int(cv2.IMWRITE_JPEG_QUALITY), 50])
print(f"Compressed {img_path} to {new_path}")
compress_folder(r"文件夹路径")
遍历读取文件夹中的图片
for filename in os.listdir(images_folder):
#循环读取文件夹中的文件名称
if filename.endswith('.jpg') or filename.endswith('.jpeg') or filename.endswith('.png'):
#识别文件夹中的图片文件
img_path = os.path.join(images_folder, filename)
#使用os函数对文件夹路径以及图片文件的名称进行拼接
img = cv2.imread(img_path)
#生成的图片路径进行读