python识别正方形,使用Python / OpenCV从图像中提取固定数量的正方形

I have several scanned images I would like to compute with Python/Opencv. Each of these images (see an example below) contains n rows of coloured squares. Each of these squares have the same size. The goal is to crop each of these squares and to extract the data from it.

I have found there a code which is able to extract squares from an image.

Here is my code where I have used it :

import numpy as np

import cv2

from matplotlib import pyplot as plt

def angle_cos(p0, p1, p2):

import numpy as np

d1, d2 = (p0-p1).astype('float'), (p2-p1).astype('float')

return abs( np.dot(d1, d2) / np.sqrt( np.dot(d1, d1)*np.dot(d2, d2) ) )

def find_squares(img):<