OpenCV Python Error in simple digit recognition

Question

import cv2
import numpy as np
#######   training part    ###############
samples = np.loadtxt('generalsamples.data',np.float32)
responses = np.loadtxt('generalresponses.data',np.float32)
responses = responses.reshape((responses.size,1))
model = cv2.KNearest()
model.train(samples,responses)
############################# testing part  #########################
im = cv2.imread('/home/manoj/Pictures/Untitled-1.jpg')
out = np.zeros(im.shape,np.uint8)
gray = cv2.cvtColor(im,cv2.COLOR_BGR2GRAY)
thresh = cv2.adaptiveThreshold(gray,255,1,1,11,2)
contours,hierarchy = cv2.findContours(thresh,cv2.RETR_LIST,cv2.CHAIN_APPROX_SIMPLE)
for cnt in contours:
    if cv2.contourArea(cnt)>50:
        [x,y,w,h] = cv2.boundingRect(cnt)
        if  h>28:
            cv2.rectangle(im,(x,y),(x+w,y+h),(0,255,0),2)
            roi = thresh[y:y+h,x:x+w]
            roismall = cv2.resize(roi,(10,10))
            roismall = roismall.reshape((1,100))
            roismall = np.float32(roismall)
            retval, results, neigh_resp, dists = model.find_nearest(roismall, k = 1)
            string = str(int((results[0][0])))
            cv2.putText(out,string,(x,y+h),0,1,(0,255,0))
cv2.imshow('im',im)
cv2.imshow('out',out)
cv2.waitKey(0)

I used this python code with Opencv for character recognition but I get this error while running the code.

AttributeError: 'module' object has no attribute 'KNearest'

gallium · Accepted Answer · 2016-02-08 07:35:13Z

2

Since the new OpenCV's version you need to replace

model = cv2.KNearest()

by:

model = cv2.ml.KNearest_create()

Moreover, the line:

model.train(samples,responses)

is also going to raise an error, this should fix it:

model.train(samples,cv2.ml.ROW_SAMPLE,responses) # Might be adapted

Hope this will help.

answered Feb 8, 2016 at 7:35

gallium

3016 silver badges12 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

OpenCV Python Error in simple digit recognition

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related