OpenCV 3 Computer Vision with Python Cookbook

How to do it...

In order to complete this recipe, you need to perform the following steps:

1. Import the module:

import cv2

2. Load the text image:

img = cv2.imread('../data/scenetext01.jpg')

3. Load the pre-trained convolutional neural network and detect the text messages:

det = cv2.text.TextDetectorCNN_create(       "../data/textbox.prototxt", "../data/TextBoxes_icdar13.caffemodel")rects, probs = det.detect(img)

4. Draw the detected text bounding boxes with confidences higher than the threshold:

THR = 0.3for i, r in enumerate(rects):    if probs[i] > THR:        cv2.rectangle(img, (r[0], r[1]), (r[0]+r[2], r[1]+r[3]), (0, 255, 0), 2)

5. Visualize the results:

plt.figure(figsize=(10,8))plt.axis('off')plt.imshow(img[:,:,[2,1,0]])plt.tight_layout()plt.show()