pt tensor to cv2 image

tensor_2_cv2.py

import cv2
import numpy as np
import torch

# Create a random torch tensor
tensor = torch.randn(3, 256, 256)

# Convert the tensor to a numpy array
numpy_image = tensor.numpy()

# Convert the numpy array to a cv2 image
cv2_image = np.transpose(numpy_image, (1, 2, 0))
cv2_image = cv2.cvtColor(cv2_image, cv2.COLOR_BGR2RGB)

# Display the image using cv2
cv2.imshow("Image", cv2_image)
cv2.waitKey(0)
cv2.destroyAllWindows()