我们使用python进行AI识别测试。具体方法是打开本地电脑的摄像头进行实时识别,或者直接传入一张图片进行行人检测。传给识别的时候看到source='0',但是这个参数是打开本地电脑的摄像头流,然后进行行人检测。但是我们这里需要修改,使用rtsp流来进行AI行人识别,那么我们需要分析代码,找到可以修改的地方,或者摸一个参数修改RTSP流。我找到了视频流的传输位置。下面分析里面的代码,改成rtsp流,把rtsp流写入里面做实时分析,实现行人检测的效果。分析的时候发现source这个参数只有LoadStreams用到,直接传入。输入source参数查看,发现里面有个默认值,就是读取文件。如果不是文件,则设置source=[source],然后遍历source值。遍历中同样使用opencv打开本地电脑的摄像头流,然后开启一个线程进行行人实时识别。代码使用了opencv中的cv2.VideoCapture函数。从网上对这个函数的使用,我们知道这个函数可以直接传入rtsp流地址,这样问题就容易解决了。cv2.VideoCapture函数可以传入rtsp地址,所以尝试传入rtsp地址,发现传入rtsp地址没有问题。只需修改source参数,最终实现检测:
