#注：此代码稍作修改也可以用于WFLW人脸数据集的标注文件解析，
#参见其README.md文件了解其每一行的信息，从中解析出相应字
#段即可。

import os

import cv2

def draw(image_list,src_img_dir = None, tar_img_dir = None):

    if not os.path.exists(tar_img_dir):

        os.mkdir(tar_img_dir)

    for item in image_list:

        sub_path = item["path"]

        path_seg = sub_path.split("/")

        path = os.path.join(src_img_dir,sub_path)

        boxes = item["boxes"]

        img = cv2.imread(path)

        for box in boxes:

            ord = box.split(" ")

            x, y, w, h = int(ord[0]),int(ord[1]),int(ord[2]),int(ord[3])

            cv2.rectangle(img,(x,y),(x+w,y+h),(0,255,0), 1)

        tar_dir = os.path.join(tar_img_dir,path_seg[0])

        if not os.path.exists(tar_dir):

            os.mkdir(tar_dir)

        tar_path = os.path.join(tar_dir,path_seg[1])

        cv2.imwrite(tar_path,img)

def parse(label_file_path, src_img_dir, tar_img_dir):

    fr = open(label_file_path,'r')

    image_list = []

    line = fr.readline().rstrip()

    while line:

        mdict = {}

        path = line

        mdict["path"] = path

        num = fr.readline().rstrip()

        boxes_list = []

        for n in range(int(num)):

            box = fr.readline().rstrip()

            boxes_list.append(box)

        mdict["boxes"]=boxes_list

        image_list.append(mdict)

        line = fr.readline().rstrip()

    draw(image_list,src_img_dir,tar_img_dir)

if __name__=="__main__":

    file_path = "/projects/DSOD/wider_face/datasets/wider_face_split/wider_face_train_bbx_gt.txt"

    source_img_dir = "/projects/DSOD/wider_face/datasets/train/images"

    target_img_dir = "/projects/DSOD/wider_face/datasets/drew"

    parse(file_path,source_img_dir,target_img_dir)

 0--Parade/0_Parade_marchingband_1_849.jpg

 1

 449 330 122 149 0 0 0 0 0 0

 0--Parade/0_Parade_Parade_0_904.jpg

 1

 361 98 263 339 0 0 0 0 0 0

 0--Parade/0_Parade_marchingband_1_799.jpg

 21

 78 221 7 8 2 0 0 0 0 0

 78 238 14 17 2 0 0 0 0 0

 113 212 11 15 2 0 0 0 0 0

 134 260 15 15 2 0 0 0 0 0

 163 250 14 17 2 0 0 0 0 0

 201 218 10 12 2 0 0 0 0 0

 182 266 15 17 2 0 0 0 0 0

 245 279 18 15 2 0 0 0 0 0

 304 265 16 17 2 0 0 0 2 1

 328 295 16 20 2 0 0 0 0 0

 389 281 17 19 2 0 0 0 2 0

 406 293 21 21 2 0 1 0 0 0

 436 290 22 17 2 0 0 0 0 0

 522 328 21 18 2 0 1 0 0 0

 643 320 23 22 2 0 0 0 0 0

 653 224 17 25 2 0 0 0 0 0

 793 337 23 30 2 0 0 0 0 0

 535 311 16 17 2 0 0 0 1 0

 29 220 11 15 2 0 0 0 0 0

 3 232 11 15 2 0 0 0 2 0

 20 215 12 16 2 0 0 0 2 0

 0--Parade/0_Parade_marchingband_1_117.jpg

 9

 69 359 50 36 1 0 0 0 0 1

 227 382 56 43 1 0 1 0 0 1

 296 305 44 26 1 0 0 0 0 1

 353 280 40 36 2 0 0 0 2 1

 885 377 63 41 1 0 0 0 0 1

 819 391 34 43 2 0 0 0 1 0

 727 342 37 31 2 0 0 0 0 1

 598 246 33 29 2 0 0 0 0 1

 740 308 45 33 1 0 0 0 2 1

  .....................