如何找到物体(形状)的方向?- Python Opencv

Joã*_*oão 4 python opencv image-processing rotation

我的图像总是这样的: 原始图像

原始图像

但我需要将它们旋转为这样: 旋转图像 旋转图像

但要做到这一点,我需要找到物体的方向,知道物体较薄的部分必须位于左侧。总之,图像是翅膀,翅膀的起点必须在左侧,翅膀的终点必须在右侧。

我希望有人能给我一个建议,我已经尝试了很多不同的策略,但到目前为止还没有好的结果。

fmw*_*w42 7

这是 Python/OpenCV 中的一种方法。

  • 阅读图像

  • 转换为灰度

  • 临界点

  • 获取外轮廓

  • 获取 minAreaRect 点和与外轮廓的角度

  • 获取旋转矩形的顶点

  • 绘制旋转后的矩形

  • 根据需要修正角度

  • 打印角度

  • 保存绘制有旋转矩形的图像


输入:

在此输入图像描述

import cv2
import numpy as np

# load image as HSV and select saturation
img = cv2.imread("wing2.png")
hh, ww, cc = img.shape

# convert to gray
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

# threshold the grayscale image
ret, thresh = cv2.threshold(gray,0,255,0)

# find outer contour
cntrs = cv2.findContours(thresh, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
cntrs = cntrs[0] if len(cntrs) == 2 else cntrs[1]

# get rotated rectangle from outer contour
rotrect = cv2.minAreaRect(cntrs[0])
box = cv2.boxPoints(rotrect)
box = np.int0(box)

# draw rotated rectangle on copy of img as result
result = img.copy()
cv2.drawContours(result,[box],0,(0,0,255),2)

# get angle from rotated rectangle
angle = rotrect[-1]

# from https://www.pyimagesearch.com/2017/02/20/text-skew-correction-opencv-python/
# the `cv2.minAreaRect` function returns values in the
# range [-90, 0); as the rectangle rotates clockwise the
# returned angle trends to 0 -- in this special case we
# need to add 90 degrees to the angle
if angle < -45:
    angle = -(90 + angle)
 
# otherwise, just take the inverse of the angle to make
# it positive
else:
    angle = -angle

print(angle,"deg")

# write result to disk
cv2.imwrite("wing2_rotrect.png", result)

cv2.imshow("THRESH", thresh)
cv2.imshow("RESULT", result)
cv2.waitKey(0)
cv2.destroyAllWindows()
Run Code Online (Sandbox Code Playgroud)
返回角度:0.8814040422439575 度

带有旋转矩形的图像:

在此输入图像描述