aoi_ocr/sid_process.py
@@ -84,7 +84,11 @@ # cv2.rectangle(image,(x,y),( x + w, y + h ),(0,255,0),2) cv2.imwrite("/tmp/sid_no_{}.png".format(i), roi) sid_no = sid_no + str(classifier.predict(roi.reshape(1, -1) / 255.0)[0]) sid_currno= str(classifier.predict(roi.reshape(1, -1) / 255.0)[0]) sid_no = sid_no + sid_currno fname="/tmp/SID/"+str(sid_currno)+"/sid"+str(np.random.randint(0,1000000))+".png" print("Writing all the sid images ", fname) cv2.imwrite(fname, roi) return sid_no
