labelimg删除用不到的标签（yolo格式）以及 下载使用

问题：当我们标注完成新的类别后后直接删除classes.txt中不需要的类别之后再次打开labelimg会闪退，如何删除不需要的标签并且能够正确运行呢？（yolo格式）

原因：当我们打开labelimg进行标注的时候，自动生成的classes.txt文件会包含一些预设存在的类别，例如，刚安装时，生成的文件内容就是labelImg-master/data/predefined_classes.txt文件 + 你所标注的标签，此时标注生成的txt文件每一行的第一个位置就是类别的序号，因为你删除了部分标签，导致重新打开labelimg的时候读取出错了故导致闪退。具体过程如下：

• 安装时自带的文件内容：
  

• 添加例如类别other后生成的classes.txt文件就为：
  

• 标注后生成的内容：
  

解决办法：可以看到此时的标注文件第一行的类别为15，因为我们标注时的classes.txt文件为十六个标签，而’other’标签的所在序列下标就是15，这就是报错内容所在。由此我们可以知道只需要在删除classess.txt中不需要的标签以外，再将所有txt文件的每一行表示标签的位置改成classes.txt序列的下表即可。例如我只需要other标签，那么我就将classes.txt中删除掉除other以外的标签，并且将所有生成的yolo格式的标注文件txt的每一行标签的位置改为0即可，故可以用以下的代码进行实现：

import os

def modify_labels(folder_path):
    # 获取labels文件夹中所有的txt文件
    txt_files = [file for file in os.listdir(folder_path) if file.endswith('.txt')]

    #逐个遍历
    for txt_file in txt_files:
        with open(os.path.join(folder_path, txt_file), 'r') as f:
            lines = f.readlines()

        # 将类别15替换为类别9
        new_lines = [modify_line(line) for line in lines]

        # 覆盖原来的txt文件内容
        with open(os.path.join(folder_path, txt_file), 'w') as f:
            f.writelines(new_lines)


def modify_line(line):
    #只需要改变标签所在位置
    class_id, *values = line.split()
    if class_id == '15':
        class_id = '0'
    new_line = class_id + " " + " ".join(values) + "\n"
    return new_line


if __name__ == "__main__":
    label_file = "label_file"#写你存放label的位置
    modify_labels(label_file)

labelimg的下载与使用

可以直接到github上下载

• 下载完成后用编译软件（pycharm）打开其中的labelimg.py即可
  
  注：若缺失库则需要在此环境下进行pip安装即可。
• 运行后的界面如下，选择好文件位置即可进行标注。
  
• 若要改变某一类别拉框的颜色，例如第一个类别，则到labelImg-master/libs/utils.py文件中的generate_color_by_text函数中进行修改：