Question

所以我和我的朋友们正在联合组装这个自制的激光测距仪，有人发布在这个网站上：http://shaneormonde.wordpress.com/2014/01/25/webcam-laser-rangefinder/。在我们点击python代码部分之前一切正常，因为我们不熟悉任何编程语言。正如网站上的项目描述的那样，它应该在网络摄像头的视野中检测到一个红点。不幸的是，我们已经购买了绿色激光。如何更改代码以便它检测到绿点？这是代码：

## program written by Shane Ormonde 7th sept 2013
## updated on 25th January 2014
## calculates the distance of a red dot in the field of view of the webcam.


import cv2
from numpy import *
import math

#variables
loop = 1

dot_dist = 0

cv2.namedWindow("preview")
vc = cv2.VideoCapture(1)



if vc.isOpened(): # try to get the first frame
    rval, frame = vc.read()

else:
    rval = False
    #print "failed to open webcam"


if rval == 1 :

    while loop == 1:
            cv2.imshow("preview", frame)
            rval, frame = vc.read()
            key = cv2.waitKey(20)
            if key == 27: # exit on ESC
                loop = 0
            num = (frame[...,...,2] > 236)
            xy_val =  num.nonzero()

            y_val = median(xy_val[0])
            x_val = median(xy_val[1])

            dist = ((x_val - 320)**2 + (y_val - 240)**2 )**0.5 # distance of dot from center pixel
            dist = abs(x_val - 320) # distance of dot from center x_axis only

            print " dist from center pixel is " + str(dist) 

            # work out distance using D = h/tan(theta)

            theta =0.0011450*dist + 0.0154         
            tan_theta = math.tan(theta)



            if tan_theta > 0: # bit of error checking
                obj_dist =  int(5.33 / tan_theta)


            print "\033[12;0H" + "the dot is " + str(obj_dist) + "cm  away"
elif rval == 0:
        print " webcam error "

Answer 1

rval, frame = vc.read()以BGR格式读取图像。你直接链接的教程页面就是这样说的。

因此所有像素数据都存储在一个名为frame的numpy数组中。默认情况下，opencv会抓取一个640 x 480的图像（x轴为640，y为480）。 numpy数组有三个维度，（x）（y）（BGR），因此前两个维度在每个维度中只有一个数字指定像素的x或y坐标。第三维包含三个数字，即该像素的RGB内容。虽然由于某种原因，这些值被排列为BGR。

因此，改变行

num = (frame[...,...,2] > 236)

到

num = (frame[...,...,1] > 236)

应该为你做的伎俩。

为了简要解释这条线的作用，它在一个颜色平面上检查高强度值（超过236 - 其中min = 0和max = 255）并记下所有这些像素。我们的想法是，如果你发出红色激光，红色平面会在点区周围产生高强度值。

如何更改此激光测距仪python代码以检测绿点而不是红点？

1 个答案: