kornia-rs

import kornia_rs as K import numpy as np import os # For demonstration, create a dummy image file if it doesn't exist # In a real scenario, replace 'dummy.jpeg' with your image path. if not os.path.exists('dummy.jpeg'): try: from PIL import Image img = Image.new('RGB', (256, 256), color = 'red') img.save('dummy.jpeg') print("Created 'dummy.jpeg' for quickstart.") except ImportError: print("Pillow not installed. Skipping dummy image creation.") print("Please create a 'dummy.jpeg' file or adjust the quickstart path.") exit() # Or handle gracefully without exiting # 1. Read an image into a kornia_rs.Tensor # This uses the image-rs backend by default (or turbojpeg if enabled/available). image_tensor = K.read_image_jpeg('dummy.jpeg') print(f"Original image tensor shape: {image_tensor.shape}") # 2. Convert the kornia_rs.Tensor to a NumPy array (zero-copy via DLPack) np_image = np.from_dlpack(image_tensor) print(f"NumPy array shape: {np_image.shape}, dtype: {np_image.dtype}") # 3. Perform a basic image operation, e.g., resize # The resize function expects a kornia_rs.Tensor and returns one. resized_tensor = K.resize(image_tensor, (128, 128), interpolation="bilinear") print(f"Resized image tensor shape: {resized_tensor.shape}") # Convert resized tensor back to NumPy for further processing or visualization np_resized_image = np.from_dlpack(resized_tensor) print(f"Resized NumPy array shape: {np_resized_image.shape}, dtype: {np_resized_image.dtype}") # Clean up dummy image if created if os.path.exists('dummy.jpeg') and 'Created' in locals(): os.remove('dummy.jpeg') print("Removed 'dummy.jpeg'.")