批量更改YOLO标签类别

批量转换YOLO目标检测标签类别

原创

已于 2022-04-14 20:18:12 修改 · 2.6k 阅读

26 ·

CC 4.0 BY-SA版权

文章标签：

#python #目标检测

于 2022-04-08 23:01:06 首次发布

博客内容讲述了如何处理一个情况，即原本的YOLO目标检测数据集中person类别的标签为1，但在新的大型数据集中person类别变为0。因此，需要批量修改labels文件中的类别编号，将所有person类别改为0，并创建新目录来存储更新后的标签。

原本的标签的classes.txt文件中person类别是1【即第二行才是person类】，而后来找到的数据集大且标注好了，好巧不巧person类别是0。故要将labels文件的类别都改成0。
要自己先创建好空的文件夹存放新标签。

import os
import random
import numpy as np
from numpy import *

txtfilepath = "/content/drive/MyDrive/person/labels/test" #原始txt文件所存文件夹，文件夹可以有一个或多个txt文件
savefilepath = "/content/drive/MyDrive/person/labels/TEST" #更改后txt文件存放的文件夹
total_txt = os.listdir(