如何在if语句中收集元组值?

时间:2018-07-18 21:04:33

标签: python opencv if-statement image-processing opencv-drawcontour

我有一个由if循环中的for语句组成的代码,如下所示,

for i in range(len(contours)):

    x, y, w, h = cv2.boundingRect(contours[i])
    mask[y:y+w, x:x+w] = 0
    cv2.drawContours(mask, contours, i, (255, 255, 255), -1)
    r = float(cv2.countNonZero(mask[y:y+h, x:x+w])) / (w * h)

    if r > 0.5 and w > 8 and h > 8:
        cv2.rectangle(rgb, (x, y), (x+w, y+h), (255, 255, 255), -1)
        cv2.circle(rgb, (x+w/2,y+h/2), 3, (180, 25, 20), -1)
        start_target_states = x+w/2 , y+h/2
        print "start_target_states: %s" % (start_target_states,)

运行此代码时,结果如下;

start_target_states: (704, 463)
start_target_states: (83, 15)

但是,对于第一个结果,start_target_states变量必须命名为start_state,然后,对于第二个结果,必须将其命名为target_state。例如;

target_state: (704, 463)
start_state: (83, 15)

此外,我想将这两个元组分配给变量名。这样我以后才能使用它们。

TargetState = target_state
StartState = start_state

我试图修改if语句以达到我的目的,很遗憾,我无法成功。我该怎么办?

1 个答案:

答案 0 :(得分:0)

如果以后需要访问它们,可以将它们添加到列表中。

states = []
for i in range(len(contours)):

    x, y, w, h = cv2.boundingRect(contours[i])
    mask[y:y+w, x:x+w] = 0
    cv2.drawContours(mask, contours, i, (255, 255, 255), -1)
    r = float(cv2.countNonZero(mask[y:y+h, x:x+w])) / (w * h)

    if r > 0.5 and w > 8 and h > 8:
        cv2.rectangle(rgb, (x, y), (x+w, y+h), (255, 255, 255), -1)
        cv2.circle(rgb, (x+w/2,y+h/2), 3, (180, 25, 20), -1)
        start_target_states = x+w/2 , y+h/2
        states.append(start_target_states)
        print "start_target_states: %s" % (start_target_states,)

由于您已将它们包含在列表中,因此您仍然可以访问它们,因为您不会每次都覆盖它们。

target_state = states[0]
start_state = states[1]

或更笼统地说,如果要捕获第一个和最后一个:

target_state = states[0]
start_state = states[-1]

此外,这不是您要问的,但对于这种循环,最好使用enumerate

for i, contour in enumerate(contours):

,然后不用counters[i],而只能使用contour