yolov5数据集生成（celeba、wideface、xml解析）

import re import linecache import os from PIL import Image # FILEDIR = "D:/code_data/widerface/wider_face/WIDER_train/" # IMGDIR = "D:/code_data/widerface/wider_face/WIDER_train/images/" # file = open(FILEDIR + 'wider_face_train_bbx_gt.txt', 'r') FILEDIR = "D:/code_data/widerface/wider_face/WIDER_val/" IMGDIR = "D:/code_data/widerface/wider_face/WIDER_val/images/" TXT_DIR = 'wider_face_val_bbx_gt.txt' file = open(FILEDIR + TXT_DIR, 'r') def count_lines(file): lines_quantity = 0 while True: buffer = file.read(1024 * 8192) if not buffer: break lines_quantity += buffer.count('\n') file.close() return lines_quantity lines = count_lines(file) for i in range(lines): line = linecache.getline(FILEDIR + TXT_DIR, i) if re.search('jpg', line): position = line.index('/') file_name = line[position + 1: -5] folder_name = line[:position] print(folder_name+"/"+file_name) img_dir = IMGDIR+folder_name+"/"+file_name+".jpg" img = Image.open(img_dir) img_w,img_h = img.size i += 1 face_count = int(linecache.getline(FILEDIR + TXT_DIR, i)) for j in range(face_count): box_line = linecache.getline(FILEDIR + TXT_DIR, i + j + 1) # x1, y1, w, h, x1,y1 为人脸框左上角的坐标 po_x1 = box_line.index(' ') x1 = box_line[:po_x1] po_y1 = box_line.index(' ', po_x1 + 1) y1 = box_line[po_x1:po_y1] po_w = box_line.index(' ', po_y1 + 1) w = box_line[po_y1:po_w] po_h = box_line.index(' ', po_w + 1) h = box_line[po_w:po_h] # **************************** dw = 1. / (int(img_w)) dh = 1. / (int(img_h)) x = (float(x1)+float(w)/2) - 1 y = (float(y1) + float(h)/2) - 1 x = x * dw w = float(w) * dw y = y * dh h = float(h) * dh # **************************** # coordinates = x + y + w + h # print(coordinates) if not (os.path.exists(FILEDIR + "labels\\" + folder_name)): os.makedirs(FILEDIR + "labels\\" + folder_name) with open(FILEDIR + "labels\\" + folder_name + "\\" + file_name + ".txt", 'a') as f: f.write(f"0 {x} {y} {w} {h}\n") i += i + j + 1