widerface2retina 标签格式转换

在使用WIDER FACE训练retianface人脸检测器的时候,需要将wider face训练集标签转换为retinaface的标签格式,下面给出的是训练集标签转换方式,验证集转换给是类似。

以下代码也可以用于自己标注的数据转换为retinaface训练数据格式。

'''
Function: transfer the wider face train label to retinaface label, it can be used at any data
Method  : transfer in reference to the format of "retinaface_gt_v1.1"
Author  : xiakj
Date    : 2019/9/9
'''

import os
import sys
import numpy as np

def transTrainLabel(input_path, out_path):
    out_fp = open(out_path, "w")
    with open(input_path, "r") as in_fp:
        while True:
            # read file name
            line = in_fp.readline()
            if not line:
                break
            # write file name to out_fp
            out_fp.write("# " + line)
            # read bbox number
            num = int(in_fp.readline())
            if num == 0:
                num = 1
            for i in range(num):
                line = in_fp.readline()
                values = [x for x in line.strip().split()]
                out_fp.write(values[0] + ' ' + values[1] + ' ' + values[2] + ' ' + values[3])
                out_fp.write(' -1.0'*16)
                out_fp.write('\n')
    out_fp.close()

def main():
    train_path = "C:/Users/Administrator/Desktop/wider_face_train_bbx_gt.txt"
    out_path = "C:/Users/Administrator/Desktop/out.txt"
    transTrainLabel(train_path, out_path)

if __name__ == "__main__":
    main()



 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值