import os
import shutil
def copy_matching_files(
source_dir, source_extension, target_dir, target_extension, destination_dir
):
# 确保保存文件夹存在
if not os.path.exists(destination_dir):
os.makedirs(destination_dir)
# 从A文件夹获取所有特定扩展名的文件的基本名称
source_base_names = {
os.path.splitext(f)[0]
for f in os.listdir(source_dir)
if f.endswith(source_extension)
}
# 从B文件夹获取所有特定扩展名的文件的完整路径
target_files = {
os.path.splitext(f)[0]: os.path.join(target_dir, f)
for f in os.listdir(target_dir)
if f.endswith(target_extension)
}
# 遍历A文件夹中的特定扩展名文件基本名称
for base_name in source_base_names:
# 检查B文件夹中是否存在具有相同基本名称的文件
if base_name in target_files:
# 获取文件的完整路径
target_file_path = target_files[base_name]
# 构建文件在目标文件夹中的新路径
destination_file_path = os.path.join(
destination_dir, os.path.basename(target_file_path)
)
# 复制文件
shutil.copy2(target_file_path, destination_file_path)
print(f"Copied: {os.path.basename(target_file_path)} to {destination_dir}")
else:
print(f"No corresponding file in B folder for base name: {base_name}")
print("Copy process completed.")
# 使用示例
source_dir = "images/test" # 源文件夹路径
source_extension = ".jpg" # 源文件的扩展名
target_dir = "VOCdevkit/Annotations" # 目标文件夹路径
target_extension = ".xml" # 目标文件的扩展名
destination_dir = "xml/test" # 保存复制文件的文件夹路径
copy_matching_files(
source_dir, source_extension, target_dir, target_extension, destination_dir
)
46行,是参考文件夹的路径,
47行,看是啥文件,是.jpg还是其他
48行,就是要找的文件夹路径,
49行,看找的文件夹文件后缀是啥,这里是.xml
50行,保存指定的路径