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
