当前位置：首页 > news >正文

通过脚本实现自动将标签内容复制到下一个标签文件中

news 来源：原创 2025/5/21 20:33:36

只需要将下面内容运行前修改文件夹路径(控制修改范围的文件名
不需要的话就随便写一个不相同的文件名就行需要的话就是在这个文件名字之前的会被修改)

import os
import time

# 文件夹路径
image_directory = r"C:\Users\Lenovo\Desktop\新建文件夹\images" # 替换为你的图片文件夹路径
labels_directory = r"C:\Users\Lenovo\Desktop\新建文件夹\labels" # 替换为你的标签文件夹路径

# 控制修改范围的文件名
modify_up_to_file = "11 .txt" # 指定只修改文件名在 "5.txt" 之前的文件

def get_txt_files(directory):
return sorted(
[f for f in os.listdir(directory) if f.endswith('.txt') and f[0].isdigit()],
key=lambda x: int(x.split('.')[0])
)

def get_image_files(directory):
return sorted(
[f for f in os.listdir(directory) if f.lower().endswith(('.png', '.jpg', '.jpeg')) and f[0].isdigit()],
key=lambda x: int(x.split('.')[0])
)

def initialize_txt_files():
# 确保标签目录存在
if not os.path.exists(labels_directory):
os.makedirs(labels_directory)

# 获取所有图片文件
image_files = get_image_files(image_directory)

# 为每个图片创建对应的txt文件（如果不存在）
for img_file in image_files:
txt_name = os.path.splitext(img_file)[0] + '.txt'
txt_path = os.path.join(labels_directory, txt_name)
if not os.path.exists(txt_path):
with open(txt_path, 'w') as f:
pass # 创建空文件

# 初始化txt文件
initialize_txt_files()

# 保存文件的最后修改时间
last_modified_times = {}

# 初始化文件的最后修改时间
for filename in get_txt_files(labels_directory):
filepath = os.path.join(labels_directory, filename)
last_modified_times[filename] = os.path.getmtime(filepath)

print("开始监视文件变化...")

while True:
try:
# 获取当前文件列表
txt_files = get_txt_files(labels_directory)

# 找到指定文件的索引
modify_up_to_index = txt_files.index(modify_up_to_file) if modify_up_to_file in txt_files else len(txt_files)

for i in range(modify_up_to_index): # 只处理指定文件之前的文件
filename = txt_files[i]
filepath = os.path.join(labels_directory, filename)

# 检查文件的最后修改时间
current_modified_time = os.path.getmtime(filepath)
if current_modified_time != last_modified_times.get(filename, 0):
# 文件已被修改
print(f"{filename} 被修改")

# 复制内容到下一个文件
if i + 1 < modify_up_to_index: # 确保不复制到指定文件及其之后的文件
next_filename = txt_files[i + 1]
next_filepath = os.path.join(labels_directory, next_filename)
print(f"正在复制内容到 {next_filename}")

with open(filepath, 'r') as source_file:
content = source_file.read()

with open(next_filepath, 'w') as dest_file:
dest_file.write(content)

# 更新最后修改时间
last_modified_times[filename] = current_modified_time
break # 复制后跳出循环，防止连锁反应

except Exception as e:
print(f"发生错误: {str(e)}")
time.sleep(1)
continue

# 等待一段时间再检查
time.sleep(1) # 每秒检查一次

相关文章：