70 lines
1.7 KiB
Python
70 lines
1.7 KiB
Python
|
import os
|
|||
|
|
import glob
|
|||
|
|
|
|||
|
|
# ======================
|
|||
|
|
# 配置部分（你要改这里）
|
|||
|
|
# ======================
|
|||
|
|
|
|||
|
|
# 标签文件所在文件夹
|
|||
|
|
LABEL_DIR = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/point2/2"
|
|||
|
|
|
|||
|
|
# 新的关键点顺序 (0-based)
|
|||
|
|
# 例如：原顺序 1,2,3,4 → 新顺序 1,2,4,3
|
|||
|
|
#new_order = [0, 1, 3, 2]
|
|||
|
|
new_order = [0, 2, 3, 1]
|
|||
|
|
|
|||
|
|
# ======================
|
|||
|
|
# 程序部分
|
|||
|
|
# ======================
|
|||
|
|
|
|||
|
|
def reorder_keypoints(line, new_order):
|
|||
|
|
parts = line.strip().split()
|
|||
|
|
|
|||
|
|
if len(parts) < 17:
|
|||
|
|
print(f"数据格式不对，字段数为 {len(parts)}，应 ≥ 17 ：", line)
|
|||
|
|
return line
|
|||
|
|
|
|||
|
|
# 前 5 个保持不变
|
|||
|
|
head = parts[:5]
|
|||
|
|
|
|||
|
|
# 后 12 个是 4 个关键点 (4 * 3)
|
|||
|
|
kps = parts[5:]
|
|||
|
|
|
|||
|
|
# 分组成 [(x1,y1,v1),(x2,y2,v2)...]
|
|||
|
|
keypoints = [kps[i:i+3] for i in range(0, 12, 3)]
|
|||
|
|
|
|||
|
|
# 按新顺序重排
|
|||
|
|
reordered = []
|
|||
|
|
for idx in new_order:
|
|||
|
|
reordered.extend(keypoints[idx])
|
|||
|
|
|
|||
|
|
return " ".join(head + reordered)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def process_all_labels(label_dir, new_order):
|
|||
|
|
txt_files = glob.glob(os.path.join(label_dir, "*.txt"))
|
|||
|
|
|
|||
|
|
print(f"找到 {len(txt_files)} 个标签文件\n")
|
|||
|
|
|
|||
|
|
for txt_path in txt_files:
|
|||
|
|
lines_out = []
|
|||
|
|
|
|||
|
|
with open(txt_path, "r") as f:
|
|||
|
|
lines = f.readlines()
|
|||
|
|
|
|||
|
|
for line in lines:
|
|||
|
|
if line.strip():
|
|||
|
|
line = reorder_keypoints(line, new_order)
|
|||
|
|
lines_out.append(line)
|
|||
|
|
|
|||
|
|
# 覆写文件
|
|||
|
|
with open(txt_path, "w") as f:
|
|||
|
|
f.write("\n".join(lines_out))
|
|||
|
|
|
|||
|
|
print(f"✓ 已处理：{txt_path}")
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
process_all_labels(LABEL_DIR, new_order)
|
|||
|
|
print("\n全部处理完成！")
|