22 lines
559 B
Python
22 lines
559 B
Python
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
|