OpenCV 3 Computer Vision with Python Cookbook

How to do it...

You need to complete the following steps:

1. Import the modules:

import cv2import numpy as np

2. Open an input image and print its shape:

image_bgr = cv2.imread('../data/Lena.png', cv2.IMREAD_COLOR)print(image_bgr.shape)

3. Transform the image to a four-dimensional floating point tensor:

image_bgr_float = image_bgr.astype(np.float32)image_rgb = image_bgr_float[..., ::-1]tensor_chw = np.transpose(image_rgb, (2, 0, 1))tensor_nchw = tensor_chw[np.newaxis, ...]print(tensor_nchw.shape)