import cv2
import numpy as np
import matplotlib.pyplot as plt
import jovian
img = cv2.imread('coins.jpg')
img = cv2.cvtColor(img,cv2.COLOR_BGR2RGB)
plt.imshow(img)
plt.axis("off")
plt.show()
Notebook Image
data = img.reshape((-1,3))
data = np.float32(data)
print(data.shape)
(786432, 3)
classes = 4

criteria = (cv2.TERM_CRITERIA_EPS + 
           cv2.TERM_CRITERIA_MAX_ITER,
           50,0.1)