首先设置多路监测,yolo官方支持多路视频流监测,方法如下:
1、直接在yolo根目录下新建一个stream.txt
2、在txt内,每行一个视频流或摄像头http/https格式网络地址(0可以是调用电脑摄像头)
3、在detect.py中将设备选择一行代码,更改如下
parser.add_argument('--source', type=str, default='streams.txt', help='file/dir/URL/glob/screen/0(webcam)')
判断检测到的对象是哪个设备的画面,方法如下:
在detect.py中,这行代码上面,添加一行
s += f"{n} {names[int(c)]}{'s' * (n > 1)}, " # add to string
添加如下代码:
device_id = s.split(':')[-2].split(' ')[-1]
device_id就是stream.txt中对应的设备id