Extracting the detected boxes

@YoungminBaek Now that Craft have detected the boxes and saved them into a .txt file, can you add a script to extract/crop the detected boxes into images.

[akarazniewicz]

This is trivial, You can use something like this:

# convert bounding shape to bounding box
def bounding_box(points):
            points = points.astype(np.int16)
            x_coordinates, y_coordinates = zip(*points)
            return [(min(x_coordinates), min(y_coordinates)), (max(x_coordinates), max(y_coordinates))]

# detection
bboxes, score_text = test_net(net, image, args.text_threshold, args.link_threshold, args.low_text, args.cuda)

for i, bbs in enumerate(bboxes):
            crop = bounding_box(bbs)
            cropped = image[crop[0][1]:crop[1][1],crop[0][0]:crop[1][0]]
            cv2.imwrite(result_folder + '/res_' + filename + '_cropped_' + str(i) + '.png', cropped)

[erikestany]

Hi,

I also need to export the detected boxes from the Colab notebook, but I don’t understand how to do it.

With the code suggested by @akarazniewicz, I get this error:

---------------------------------------------------------------------------
NameError                                 Traceback (most recent call last)
[<ipython-input-7-b12c4da62a69>](https://localhost:8080/#) in <module>()
      7 
      8 # detection
----> 9 bboxes, score_text = test_net(net, image, args.text_threshold, args.link_threshold, args.low_text, args.cuda)
     10 
     11 for i, bbs in enumerate(bboxes):

NameError: name 'test_net' is not defined

What am I doing wrong?

Thank you so much!