In this article, we are going to see how to find mean across the image channels in PyTorch. We have to compute the mean of an image across the channels Red, Green, and, Blue. we can find the mean across the image channel by using torch.mean() method.
torch.mean() method
torch.mean() method is used to find the mean of all elements in the input tensor but this method only accepts input as a tensor only so first we have to convert our image to a PyTorch tensor. After converting we use this PyTorch tensor as the input tensor. The below syntax is used to find mean across the image channels
Syntax: torch.mean(input, dim)
Parameter:
- input (Tensor): This is our input tensor.
- dim (int or tuple of python:ints): the dim is used for dimensions. we set dim = [1,2] to find mean across the image channels Red, Green, and Blue.
Return: This method returns the mean for all the elements present in the input tensor.
The below image is used for Example:
Â
Example 1: In the below example, we use PIL to read images from the computer and then we find mean across the image channels in PyTorch.
Python
# import required librariesimport torchfrom PIL import Imageimport torchvision.transforms as transformsÂ
# Read input imageimg = Image.open('img.png')Â
# create a transformtransform = transforms.ToTensor()Â
# convert the image to PyTorch TensorimgTensor = transform(img)Â
# Compute the mean of Image across the # channels RGBr, g, b = torch.mean(imgTensor, dim=[1, 2])Â
# Display Resultprint("Mean for Red channel: ", r)print("Mean for Green channel: ", g)print("Mean for Blue channel: ", b) |
Output:
Â
Example 2: In the below example, we use OpenCV to read images from the computer and then we find mean across the image channels in PyTorch.
Python
# import required librariesimport torchimport cv2import torchvision.transforms as transformsÂ
# Read input image using OpenCVimg = cv2.imread('img.png')img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)Â
# create a transformtransform = transforms.ToTensor()Â
# convert the image to PyTorch TensorimgTensor = transform(img)Â
# Compute the mean of Image across the # channels RGBr, g, b = torch.mean(imgTensor, dim=[1, 2])Â
# Display Resultprint("\n\nMean for Red channel: ", r)print("Mean for Green channel: ", g)print("Mean for Blue channel: ", b) |
Output:
Â
