我有一个由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语句以达到我的目的,很遗憾,我无法成功。我该怎么办?
答案 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
。