# Transforming the image to improve detection def w2d(img, mode='haar', level=1): imArray = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY) coeffs = pywt.wavedec2(imArray, mode, level=level) coeffs[0] *= 0 imArray_H = pywt.waverec2(coeffs, mode) imArray_H = np.clip(imArray_H, 0, 255) imArray_H = np.uint8(imArray_H) return imArray_H