（二）清洗数据集标签、选择标签中指定对象并以此划分训练集和验证集

• traffic_data
  • images
    • 内含图像
  • yolo_label
    • 内含标签文件，除了后缀，文件名与图像名一一对应
  • new_label
    • 空文件夹
  • traffic_light_one_73
    • images
      • train
        -空文件夹
      • val
        -空文件夹
    • images
      • train
        -空文件夹
      • val
        -空文件夹
  • get_one.ipynb
    traffic_data为根目录，以下代码存放在get_one.ipynb。

1.判断图片数据集images与yolo格式标签文件夹yolo_label内，除后缀文件名是否相同。

1）构建函数

import os

def get_difference_from_two_folders(images_path,label_path):
    image_dirs = os.listdir(images_path)
    label_dirs = os.listdir(label_path)
    image_dirs.sort()
    label_dirs.sort()

    file_number = len(image_dirs)
    different_dirs =[]
    for i in range(file_number):
        if image_dirs[i].split('.')[0] != label_dirs[i].split('.')[0]:
            different_dirs.append(image_dirs[i])
    return different_dirs

2）执行代码

difference_dirs = get_difference_from_two_folders(