INNER CODE UNIT · Python
preprocess_pytorch_style
DataTalksClub/machine-learning-zoomcamp · cohorts/2026/10-kubernetes/workshop/service/app.py:10
def preprocess_pytorch_style(X):
X = X / 255.0
mean = np.array([0.485, 0.456, 0.406]).reshape(1, 3, 1, 1)
std = np.array([0.229, 0.224, 0.225]).reshape(1, 3, 1, 1)
X = X.transpose(0, 3, 1, 2)
X = (X - mean) / std
return X.astype(np.float32)
preprocessor = create_preprocessor(
preprocess_pytorch_style,
target_size=(224, 224)
)
session = ort.InferenceSession(
"clothing-model.onnx", providers=["CPUExecutionProvider"]