首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >如何找到对象(形状)的方向?- Python Opencv

如何找到对象(形状)的方向?- Python Opencv
EN

Stack Overflow用户
提问于 2019-10-31 03:31:29
回答 3查看 9.4K关注 0票数 3

我的图片总是这样的:

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

但要做到这一点,我需要找到对象的方向,知道对象的较薄部分必须在左侧。总而言之,图像是机翼,机翼的起点必须在左侧,机翼的末端必须在右侧。

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

EN

回答 3

Stack Overflow用户

发布于 2019-10-31 08:40:06

以下是Python/OpenCV中的一种方法。

  • 读取图像

  • 转换为灰度

  • Threshold

  • 获取外轮廓

  • 从外轮廓获取minAreaRect点和角度

  • 获取旋转矩形的顶点

  • 绘制旋转的矩形

  • 根据需要更正角度

  • 打印角度

  • 将图像与绘制在其上的旋转矩形一起保存

输入:

代码语言:javascript
运行
复制
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()

返回角度: 0.8814040422439575度

带旋转矩形的图像:

票数 4
EN

Stack Overflow用户

发布于 2019-10-31 05:15:41

你必须计算主轴(它是基于PCA的)。它将给你一个主要方向的好主意,然后你可以相应地旋转你的图像。

由于它是作为注释指向的,现在您必须测试薄的部分是否在图像的右侧,为此,您使用质心/重心:如果质心位于边界框的左侧,则机翼方向正确。

下面是完整的算法:

  • 计算主方向(主轴,用主成分分析计算)
  • 根据主轴和边界框旋转你的图像,质心在左边,那么你的图像定向良好,否则旋转大约180°。

这是结果。

票数 2
EN

Stack Overflow用户

发布于 2021-06-02 16:56:48

对于类似的任务,我多次使用这个python library进行面向对象的检测。这是一个可训练的神经网络,检测物体的位置和方向。知道方向角度后,可以使用opencv将对象旋转到所需的角度。您将需要标记一些图像来训练网络。

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/58632469

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档