读取widerface数据集进行标注框出人脸

import cv2
import os
f = open(r'D:\widerface\wider_face_split\wider_face_train_bbx_gt.txt','r')
lines = f.readlines()
imgnames = []
imgboxs = []
for i in range(len(lines)):
    if lines[i][-2] == 'g':
        line = lines[i].strip()
        imgnames.append(line)
for i in range(len(lines)):
    if lines[i][-2] == 'g':
        box = []
        imgboxs.append(box)
    else:
        box.append(lines[i].strip())
for i in range(len(imgboxs)):
    img = cv2.imread(os.path.join(r'D:\widerface\WIDER_train\images', imgnames[i]))
    for boxs in imgboxs[i]:
        box = boxs
        if len(box)>5:
            box = box.strip().split(' ')
            x1 = int(box[0])
            y1 = int(box[1])
            w = int(box[2])
            h = int(box[3])
            x2 = x1+w
            y2 = y1+h
            cv2.rectangle(img,(x1,y1),(x2,y2),(0,0,255),2)
    cv2.imshow('a',img)
    cv2.waitKey(0)