예제 #1
0
def main(argv):
    try:
        opts, args = getopt.getopt(argv, 'rgycp')
    except getopt.GetoptError:
        print "Arguments: -r -g -y -c -p"
    color = Color.Red
    for opt, arg in opts:
        if opt == '-r':
            color = Color.Red
        if opt == '-g':
            color = Color.Green
        if opt == '-y':
            color = Color.Yellow
        if opt == '-c':
            color = Color.Cyan
        if opt == '-p':
            color = Color.Purple
    try:
        vis = Vision(True)
        while cv2.waitKey(10) == -1:
            vis.grabFrame()
            print vis._detectObjectCentroid(color)
    except:
        pass