contour_utils.py 文件源码

def categorize_contour(
    contour,
    perimeter_heuristic=PERIMETER_HEURISTIC,
    linear_heuristic=LINEAR_HEURISTIC):
  """
  Categorize a contour as a box, a line, or speck

  """
  area = cv2.contourArea(contour)
  perimeter = cv2.arcLength(contour, True)

  if perimeter <= perimeter_heuristic:
    return CONTOUR_SPECK

  ratio = area / perimeter
  if ratio < linear_heuristic:
    return CONTOUR_LINE
  else:
    return CONTOUR_BOX