module numpy has no attribute unit8

Question

Respected Sir/Madam

I am trying to run the below Image segmentation code in my system, but it is showing me an error.

import cv2
import numpy as np
import matplotlib.pyplot as plt
image=cv2.imread("nn.jpg")
image=cv2.cvtColor(image,cv2.COLOR_BGR2RGB)
pixel_values=image.reshape((-1,3))
pixel_values=np.float32(pixel_values)
print(pixel_values.shape)
criteria=(cv2.TERM_CRITERIA_EPS+cv2.TERM_CRITERIA_MAX_ITER,100,0.2)
k=3
_,labels,(centers)=cv2.kmeans(pixel_values,k,None,criteria,10,cv2.KMEANS_RANDOM_CENTERS)
centers=np.unit8(centers)
labels=labels.flatten()
segmented_image=centers[labels.flatten()]
segmented_image=segmented_image.reshape(image.shape)
plt.imshow(segmented_image)
plt.show()
module 'numpy' has no attribute 'unit8'

Answer 1

Hi, @Ganesh,

You have used unit8 in your code. But NumPy has no attribute named unit8. You have to use uint8 instead of unit8 in your code. So make changes in your code and try once again. It will work.

Answer 2

Hi@Ganesh,

I think you are calling the wrong function from your Numpy module. The Numpy module doesn't have unit8(). Just replace unit8() with uint8(). It will work.