imagecodecs_numpy_array_jpeg_encode_decode_roundtrip.py

1import imagecodecs
2import numpy
3
4# Create a sample numpy array (e.g., a 256x256 grayscale image)
5data = numpy.random.randint(0, 255, (256, 256), dtype='uint8')
6
7# Encode numpy array to JPEG compressed bytes
8encoded = imagecodecs.jpeg_encode(data)
9
10# Decode JPEG compressed bytes back to numpy array
11decoded = imagecodecs.jpeg_decode(encoded)
12
13# Verify the result
14assert numpy.array_equal(data, decoded)
15print(f"Original shape: {data.shape}, Decoded shape: {decoded.shape}")
16print(f"Encoded size: {len(encoded)} bytes")