preprocessing.py

import pandas as pd
import numpy as np
import cv2

path = 'data/vali_modified.csv'
df = pd.read_csv(path)

image_path_array = df['image_path'].as_matrix()
label_array = df['category'].as_matrix()
x1 = df['x1'].as_matrix().astype(np.float32)
y1 = df['y1'].as_matrix().astype(np.float32)
x2 = df['x2'].as_matrix().astype(np.float32)
y2 = df['y2'].as_matrix().astype(np.float32)


for i in range(len(image_path_array)):
    path = image_path_array[i]
    img = cv2.imread(path)
    if img is None:
        continue
    h = img.shape[0]
    w = img.shape[1]

    x1[i] = x1[i] * 1.0 /w
    x2[i] = x2[i] * 1.0/ w
    y1[i] = y1[i] * 1.0/ h
    y2[i] = y2[i] * 1.0/ h

df['x1_modified'] = pd.DataFrame(x1)
df['y1_modified'] = pd.DataFrame(y1)
df['x2_modified'] = pd.DataFrame(x2)
df['y2_modified'] = pd.DataFrame(y2)

df.to_csv('data/vali_modified2.csv', index=False)
print(df.head())