dataset.py

import pandas as pd
import numpy as np

fields = ['FACE_X','FACE_Y','FACE_WIDTH','FACE_HEIGHT']
df = pd.read_csv('a.txt', usecols=fields)

df['FACE_WIDTH']=df['FACE_X']+df['FACE_WIDTH']
df['FACE_HEIGHT']=df['FACE_Y']+df['FACE_HEIGHT']


# Some indexes may contain NAN
# All contain NA or inf and not possible to convert integer

# Shows the Null index
# for index, row in df['FACE_X'].iteritems():
#     if row != row:
#         print('index:', index, 'isnull')

# What needs to be done if we have NAN
# df['FACE_X'].fillna(0).astype(int)

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