Gets VGGFace2 dataset and changes it into yolo v3 format

yolov3format.py

import pandas as pd
import numpy as np
from lxml import etree
#import xmlAnnotation.etree.cElementTree as ET
import xml.etree.ElementTree as ET

fields = ['NAME_ID', 'XMIN', 'YMIN', 'W', 'H', 'XMAX', 'YMAX']
df = pd.read_csv('loose_bb_test.csv', usecols=fields)


# Change the name of the file.
# This will replace the / with -
def nameChange(x):
    x = x.replace("/", "-")
    return x


df['NAME_ID'] = df['NAME_ID'].apply(nameChange)


np.savetxt(r'b.txt', df.values, fmt='%s', delimiter=',')


import fileinput

with fileinput.FileInput("b.txt", inplace=True, backup='.bak') as file:
    for line in file:
        print(line.replace(",", " ",1), end='')