|
@@ -17,24 +17,33 @@ def clean_label_files(labels_dir, output_labels_dir, log_file):
|
|
|
for label_file in label_files: # 遍历每一个txt文件名
|
|
|
file_path = os.path.join(labels_dir, label_file) # 拼接得到旧的txt文件地址
|
|
|
output_file_path = os.path.join(output_labels_dir, label_file) # 新的txt文件地址
|
|
|
- with open(file_path, 'r') as f: # 打开旧的txt文件
|
|
|
- lines = f.readlines() # 得到文件内容
|
|
|
- new_lines = []
|
|
|
- file_needs_logging = False
|
|
|
- for line in lines: # 遍历处理文件内容
|
|
|
- parts = line.strip().split()
|
|
|
- if parts:
|
|
|
- first_num = int(parts[0])
|
|
|
- other_nums = [float(num) for num in parts[1:]]
|
|
|
- if first_num not in [4, 20] and all(
|
|
|
- 0 <= num <= 1 for num in other_nums): # 只有当一行内容的第一个数字不是4或20,其他数字都在[0,1]之间时这一行内容才能被保留
|
|
|
- new_lines.append(line)
|
|
|
- elif not all(0 <= num <= 1 for num in other_nums): # 如果不满足上诉条件且有数字不在【0,1】的,要记录文件名
|
|
|
- file_needs_logging = True
|
|
|
- if file_needs_logging:
|
|
|
- processed_files.append(label_file)
|
|
|
- with open(output_file_path, 'w') as f: # 将文本内容写入新文件夹中
|
|
|
- f.writelines(new_lines)
|
|
|
+
|
|
|
+ _, file_extension = os.path.splitext(file_path)
|
|
|
+ if not file_extension.lower() == ".txt":
|
|
|
+ continue
|
|
|
+
|
|
|
+ try:
|
|
|
+ with open(file_path, 'r') as f: # 打开旧的txt文件
|
|
|
+ lines = f.readlines() # 得到文件内容
|
|
|
+ new_lines = []
|
|
|
+ file_needs_logging = False
|
|
|
+ for line in lines: # 遍历处理文件内容
|
|
|
+ parts = line.strip().split()
|
|
|
+ if parts:
|
|
|
+ first_num = int(parts[0])
|
|
|
+ other_nums = [float(num) for num in parts[1:]]
|
|
|
+ if first_num not in [4, 20] and all(
|
|
|
+ 0 <= num <= 1 for num in other_nums): # 只有当一行内容的第一个数字不是4或20,其他数字都在[0,1]之间时这一行内容才能被保留
|
|
|
+ new_lines.append(line)
|
|
|
+ elif not all(0 <= num <= 1 for num in other_nums): # 如果不满足上诉条件且有数字不在【0,1】的,要记录文件名
|
|
|
+ file_needs_logging = True
|
|
|
+ if file_needs_logging:
|
|
|
+ processed_files.append(label_file)
|
|
|
+ with open(output_file_path, 'w') as f: # 将文本内容写入新文件夹中
|
|
|
+ f.writelines(new_lines)
|
|
|
+
|
|
|
+ except Exception as e:
|
|
|
+ print(f"发生错误: {e}")
|
|
|
|
|
|
pbar.update(1) # 更新进度条
|
|
|
|