可以将一个文件夹中的图像和标签按指定关键字分离出

import os
import shutil

source_folder = "/home/zhaoyubing/yubing/这是oil_detect的数据集/VOCdevkit/VOC2007_0901_train"  # 原始文件夹路径
image_folder = "/home/zhaoyubing/yubing/这是oil_detect的数据集/VOCdevkit/VOC2007_0901_val"  # 新的图片文件夹路径
annotation_folder = "/home/zhaoyubing/yubing/这是oil_detect的数据集/VOCdevkit/VOC2007_0901_val"  # 新的标签文件夹路径
keyword = "mianji1"  # 关键字

image_destination_folder = os.path.join(image_folder, "JPEGImages")
annotation_destination_folder = os.path.join(annotation_folder, "Annotations")

# 创建新的图片文件夹和标签文件夹
os.makedirs(image_destination_folder, exist_ok=True)
os.makedirs(annotation_destination_folder, exist_ok=True)

# 构造 JPEGImages 文件夹路径和 Annotations 文件夹路径
jpeg_images_folder = os.path.join(source_folder, "JPEGImages")
annotations_folder = os.path.join(source_folder, "Annotations")

# 遍历 JPEGImages 文件夹中的图片文件
for file_name in os.listdir(jpeg_images_folder):
    if keyword in file_name and file_name.endswith(".jpg"):
        # 构造源文件路径和目标文件路径
        source_file_path = os.path.join(jpeg_images_folder, file_name)
        destination_image_file_path = os.path.join(image_destination_folder, file_name)

        # 移动图片文件到新的图片文件夹
        shutil.move(source_file_path, destination_image_file_path)
        print(f"移动图片文件：{file_name}")

# 遍历 Annotations 文件夹中的标签文件
for file_name in os.listdir(annotations_folder):
    if keyword in file_name and file_name.endswith(".xml"):
        # 构造源文件路径和目标文件路径
        source_file_path = os.path.join(annotations_folder, file_name)
        destination_annotation_file_path = os.path.join(annotation_destination_folder, file_name)

        # 移动标签文件到新的标签文件夹
        shutil.move(source_file_path, destination_annotation_file_path)
        print(f"移动标签文件：{file_name}")