关于python:深度学习开发过程中常用数据处理脚本持续更新中

56次阅读

共计 724 个字符,预计需要花费 2 分钟才能阅读完成。

将指定文件夹下的图片枚举并将门路存在列表中:

def list_images(folder_path):
    # 反对的图片扩展名
    extensions = ['.jpg', '.jpeg', '.png', '.gif']
    # 遍历文件夹中的所有文件
    image_list = []
    for root, dirs, files in os.walk(folder_path):
        for file in files:
            # 获取文件扩展名
            ext = os.path.splitext(file)[1]
            # 如果是图片文件,则将其门路增加到列表中
            if ext.lower() in extensions:
                image_list.append(os.path.join(root, file))
    return image_list

目录下的文件除扩展名与其余目录下的文件出扩展名外不雷同名称的移除

import os

dir_path = "/path/to/directory"  # 指定目录门路
target_dir_path = "/path/to/target/directory"  # 另一个目录门路

# 获取目标目录下的所有文件名(不包含扩展名)target_file_names = [os.path.splitext(f)[0] for f in os.listdir(target_dir_path)]

# 遍历指定目录下的所有文件
for file_name in os.listdir(dir_path):
    # 如果文件名(不包含扩展名)与目标目录下的某个文件名雷同,则删除该文件
    if os.path.splitext(file_name)[0] in target_file_names:
        os.remove(os.path.join(dir_path, file_name))

正文完
 0