24 lines
453 B
Python
24 lines
453 B
Python
import numpy as np
|
|
|
|
|
|
def preprocess_input(
|
|
x, mean=None, std=None, input_space="RGB", input_range=None, **kwargs
|
|
):
|
|
|
|
if input_space == "BGR":
|
|
x = x[..., ::-1].copy()
|
|
|
|
if input_range is not None:
|
|
if x.max() > 1 and input_range[1] == 1:
|
|
x = x / 255.0
|
|
|
|
if mean is not None:
|
|
mean = np.array(mean)
|
|
x = x - mean
|
|
|
|
if std is not None:
|
|
std = np.array(std)
|
|
x = x / std
|
|
|
|
return x
|