从numpy数组中提取X坐标

时间:2015-12-03 04:46:07

标签: python linux numpy raspberry-pi2 simplecv

我通过实时两个相机流提取2个blob的x位置。我可以得到第一个x位置没问题,因为它是作为元组给出的(ex = ... object at(455,69))。问题是我需要第二个blob左下角的x位置,但它返回一个numpy数组,'blob.x'不起作用。如何获得numpy数组的x位置?非常感谢任何帮助/指导。

我收到以下错误:     ValueError:具有多个元素的数组的真值是不明确的。使用a.any()或a.all()

from SimpleCV import *
def getscoreforrgb(rgb):
    return rgbmap[rgb]


mog1 = MOGSegmentation(history = 200, nMixtures = 5, backgroundRatio = 0.9, noiseSigma = 16, learningRate = 0.9)
mog0 = MOGSegmentation(history = 200, nMixtures = 5, backgroundRatio = 0.9, noiseSigma = 16, learningRate = 0.9)
cam1 = SimpleCV.Camera(1, {'width': 640, 'height': 480 })
cam0 = SimpleCV.Camera(0, {'width': 640, 'height': 480 })
pixcol = Image('/home/pi/Darts/score/scoreboardpy.png')



while True:

    frame1 = cam1.getImage()
    frame0 = cam0.getImage().flipHorizontal()
    mog1.addImage(frame1)
    mog0.addImage(frame0)
    segmentedImage1 = mog1.getSegmentedImage()
    segmentedImage0 = mog0.getSegmentedImage()


#### second blob below, does not print x position     

    blobs0 = segmentedImage0.findBlobs()
    if blobs0 is not None:
            blobs0.sortArea()
            blobs0[-1].draw(Color.BLUE, width=4)
            first_blob = blobs0[-1]
            bottomLeftCorner = second_blob.bottomLeftCorners()
            print bottomLeftCorner
            if bottomLeftCorner:
                    print bottomLeftCorner.x,
                    y = int(bottomLeftCorner.x)
                    print y * 2, 'Y'
                    y2 = y * 2

#### First blob below, code prints x position    

    blobs1 = segmentedImage1.findBlobs()
    if blobs1 is not None:
            blobs1.sortArea()
            blobs1[-1].draw(Color.RED, width=4)
            second_blob = blobs1[-1]
            if second_blob:
                    print second_blob.x, 
                    x = int(second_blob.x)
                    print x * 2, 'X'
                    x2 = x * 2


            colrgb =  pixcol[x2, y2]
            print colrgb

1 个答案:

答案 0 :(得分:1)

如果y2,则不会定义

blobs0 is None,在这种情况下,您可能不想做任何事情。

我建议您将所有内容放在一个if块中:

blobs0 = segmentedImage0.findBlobs()
blobs1 = segmentedImage1.findBlobs()
if blobs0 is not None and blobs1 is not None:
    # all your code here

您似乎在第一个块中使用second_blob而不是first_blob。你应该理解你的代码正在做什么,而不是盲目地使用一些希望它可以工作的旧代码块。