import os import cv2 # 定义原始和更改大小后的文件夹路径 original_folder_path = 'E:\shujuji\Intel Image Classification\seg_train\seg_train/buildings' resized_folder_path = 'E:\shujuji\Intel Image Classification\seg_train\change/buildings' # 如果更改大小后的文件夹不存在,则创建它 if not os.path.exists(resized_folder_path): os.makedirs(resized_folder_path) # 循环遍历原始文件夹中的所有图像文件 for filename in os.listdir(original_folder_path): if filename.endswith('.jpg') or filename.endswith('.png') or filename.endswith('.jpeg'): # 读取原始图像 img = cv2.imread(os.path.join(original_folder_path, filename)) # 将图像大小更改为(224,224,3) img_resized = cv2.resize(img, (224, 224)) # 保存更改大小后的图像到新文件夹中,使用原始文件名 cv2.imwrite(os.path.join(resized_folder_path, filename), img_resized),对这段代码进行更改,使之满足要求
时间: 2024-03-22 10:40:15 浏览: 84
以下是修改后的代码,满足获取文件夹下所有文件夹的所有图片,更改图片的尺寸后生成文件夹,按照原始文件夹进行排布,文件名使用原始文件名的要求:
```python
import os
import cv2
# 定义函数,用于调整图片尺寸
def resize_image(image_path, output_path, new_size):
img = cv2.imread(image_path)
img_resized = cv2.resize(img, new_size)
cv2.imwrite(output_path, img_resized)
# 定义函数,用于获取指定文件夹下所有图片
def get_all_images(folder_path):
images = []
for root, dirs, files in os.walk(folder_path):
for file in files:
if file.endswith('.jpg') or file.endswith('.png') or file.endswith('.jpeg'):
images.append(os.path.join(root, file))
return images
# 定义函数,用于生成新文件夹并存储调整后的图片
def generate_resized_images(source_folder, target_folder, new_size):
# 获取所有图片
images = get_all_images(source_folder)
# 按照原始文件夹进行排布
for image_path in images:
relative_path = os.path.relpath(image_path, source_folder)
output_folder = os.path.join(target_folder, os.path.dirname(relative_path))
os.makedirs(output_folder, exist_ok=True)
output_path = os.path.join(output_folder, os.path.basename(image_path))
# 调整图片尺寸并存储
resize_image(image_path, output_path, new_size)
# 定义原始和更改大小后的文件夹路径
original_folder_path = r'E:\shujuji\Intel Image Classification\seg_train\seg_train\buildings'
resized_folder_path = r'E:\shujuji\Intel Image Classification\seg_train\change\buildings'
# 如果更改大小后的文件夹不存在,则创建它
if not os.path.exists(resized_folder_path):
os.makedirs(resized_folder_path)
# 调用函数,生成新文件夹并存储调整后的图片
new_size = (224, 224) # 调整后的图片尺寸为 224x224
generate_resized_images(original_folder_path, resized_folder_path, new_size)
```
在使用时,需要将代码中的 `original_folder_path` 替换为要处理的原始文件夹的路径,`resized_folder_path` 替换为生成的新文件夹的路径,`new_size` 替换为要调整的图片尺寸。代码会自动获取原始文件夹下所有图片,并按照原始文件夹进行排布,生成新文件夹并存储调整后的图片。新文件夹的文件名使用原始文件名。
阅读全文