from PIL import Image import numpy as np def preprocess_func(img_file_path, input_batch_size, input_channel, input_height, input_width): image = Image.open(img_file_path).convert("RGB") image = image.resize((input_width, input_height), Image.BILINEAR) np_data = np.array(image, dtype="float32") np_data = np_data.transpose(2, 0, 1) np_data = np_data.reshape( input_batch_size, input_channel, input_height, input_width, ) np_data = np_data / 256.0 np_data = np_data - 0.5 return np_data