Find number of channels in your image

num_channels.py

"""
num_channels.py
Find number of channels in your image.

Author: liuhh02 https://machinelearningtutorials.weebly.com/
"""

from PIL import Image
import numpy as np

# name of your image file
filename = './directory/to/image/file/image.tif'

# open image using PIL
img = Image.open(filename)

# convert to numpy array
img = np.array(img)

# find number of channels
if img.ndim == 2:
    channels = 1
    print("image has 1 channel")
else:
    channels = image.shape[-1]
    print("image has", channels, "channels")

On line 25 where you have image.shape[-1], I believe it should be img.shape[-1] because shape is part of NumPy's attribute, not PIL image object.

@hchen98 thanks!

I needed to change the image to img = cv2.imread(filename) because the img in your code has no attribute ndim