OpenCV的RotatedRect在Python3中没有属性'size'。如何解决这个问题？

Question

我发现 OpenCV's RotatedRect class 的 .size.height 和 .size.width 运算符在 Python 中不起作用，而它们在 C++ 中起作用。让我用一个简化的代码片段来详细说明：

cap = cv2.VideoCapture('video1.mp4')
filter = RandomClass(20)

while(cap.isOpened()):
    ret, frame = cap.read()              # Read a frame
    res = filter.classMain(frame)        # Process the frame
    if (res == 0):
        print('Success')                 # If processing completed, print Success
cap.release()

其中class定义如下：

import cv2
import numpy as np

class RandomClass:
    def __inti__(self):
        self.set_skip_first(True)
    def get_skip_first(self):
        return self.skip_first
    def set_skip_first(self, value):
        self.skip_first = value
    def classMain(self, frame):
        if not get_skip_first():
            self.expand_minRect(100)        # expand the minRect by 100 pixels
            # use the expanded rectangle for some other processing here
        else:
            self.set_skip_first(False)
        # create a mask with cv2.inRange
        contour = cv2.findContours(mask, cv2.RETR_TREE, cv2.CHAIN_APPROX_NONE, offset=(0,0))[1]
        # iterate over each contour and find the index of the largest contour
        self.minRect = cv2.minAreaRect(np.array(self.contours[self.largest_contour_index]))
        # execute some other processing here
        return 0
    def expand_minRect(self, value):
        self.minRect.size.height = self.minRect.size.height + value
        self.minRect.size.width = self.minRect.size.width + value

我收到的错误如下。确切的行在上述代码的 C++ 版本中工作得很好。

File "filename", line 106, in expand_minRect

self.minRect.size.height = self.minRect.size.height + value

AttributeError: 'tuple' object has no attribute 'size'

我尝试了以下方法。我期望第二个打印值（变量 width2）比第一个打印值（变量 width1）大 value。

def expand_minRect(self, value):
    _,(width1, height1),_ = self.minRect
    print(width)
    self.minRect[1][0] = self.minRect[1][0] + value
    _,(width2,height2),_ = self.minRect
    print(w)

但是没用，因为self.minRect[1][0]的变量类型是元组，不能修改元组。

File "filename", line 111, in expand_minRect

self.minRect1[0] = self.minRect1[0] + value

TypeError: 'tuple' object does not support item assignment

我做了一些研究，找不到 RotatedRect 的 Python 文档，但我找到 Whosebug answer 说明

Python still lacks of RotatedRect class

所以所有的事情都放在一边，假设 Python3 中的 RotatedRect 支持不完整，我该如何解决这个问题并扩展我的 minRect 变量的宽度和高度？

Answer 1

根据 this tutorial，minAreaRect return 是一个带有 ((center_x,center_y),(width,height),angle) 的矩形。因此，如果您修改 expand_minRect 以使用正确的组件重新创建它，它应该可以工作。

def expand_minRect(self, value):
    self.minRect = (self.minRect[0],                                          # keep the center
                    (self.minRect[1][0] + value, self.minRect[1][1] + value), # update the size
                    self.minRect[2])                                          # keep the angle

注意：问题的出现是因为 OpenCV python 的面向对象实现比 OpenCV c++ 少。它不是 return 允许按名称访问每个属性的结构（如“.size”）。您需要了解每个元素的元组顺序和假设。

OpenCV的RotatedRect在Python3中没有属性'size'。如何解决这个问题？

OpenCV's RotatedRect has no attribute 'size' in Python3. How to work around this?

python

opencv

computer-vision

python-3.x

image-masking