将wider face数据集的标签修改为MTCNN(Tensorflow版)可用的标签格式

最近在网上看到了一个博主使用MTCNN来进行人脸识别，想要学习一下相关的知识，奈何在P-Net数据集的准备阶段就遭遇了挫折。

原博主博客在此处：MTCNN（Tensorflow）学习记录（生成PNet的人脸数据样本）

wider face的标签格式发生了改变，博主在评论里有提到这一点，比较遗憾的是他没有说明标签发生了怎样的改变。因此我在这里记录一下这个改变，以便将来不用大幅修改数据集分类的代码。

原版数据集标签是这个格式：

# 打开在wider face官网上下载的标签，原名叫wider_face_train_bbx_gt.txt
f = open('wider_face_train.txt', 'r')
# 生成新的标签
output = open('new_lable.txt', 'w')
line = f.readline()
while line != '':img = ''if line[-1] == 'g':img += lineimg += ' 'n = int(f.readline())if n != 0:while n > 0 :box = (f.readline().strip().split(' '))[0:4]box[2] = str(int(box[2]) + int(box[0]))box[3] = str(int(box[3]) + int(box[1]))for b in box:img += b img += ' 'n -= 1 img += '\n'output.write(img)line = f.readline().strip()
f.close
output.close
print('Done!')

新的数据集里面有四张图片是没有选框的，在代码里面跑同样会报错，但我的代码会在那种图片下面加一个换行，因此还是很好发现的，将其删除就可以正常使用新的数据集标签了。

本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！