update example and build
This commit is contained in:
@@ -28,7 +28,7 @@ from multiprocessing import Queue, Process
|
||||
import time
|
||||
|
||||
|
||||
def sensor_worker(sensor_id, vid_src, config_name, result_queue, stop_event):
|
||||
def sensor_worker(sensor_id, vid_src, config_name, cal_path, result_queue, stop_event):
|
||||
"""
|
||||
传感器工作进程函数
|
||||
|
||||
@@ -46,7 +46,8 @@ def sensor_worker(sensor_id, vid_src, config_name, result_queue, stop_event):
|
||||
isstitch=True,
|
||||
cuda=True,
|
||||
verbose=False, # 减少子进程输出
|
||||
config_name=config_name
|
||||
config_name=config_name,
|
||||
cal_path=cal_path
|
||||
)
|
||||
|
||||
print(f"传感器 {sensor_id} 已启动")
|
||||
@@ -116,12 +117,16 @@ def main():
|
||||
num_sensors = 2
|
||||
sensor_configs = [
|
||||
{
|
||||
'sensor_id': 0,
|
||||
'vid_src': './s1.mp4',
|
||||
'config_name': 'prototype'
|
||||
'config_name': 'prototype',
|
||||
'cal_path': './config/cal.npy'
|
||||
},
|
||||
{
|
||||
'sensor_id': 1,
|
||||
'vid_src': './s1.mp4', # 可以改为不同的视频源或摄像头索引
|
||||
'config_name': 'prototype'
|
||||
'config_name': 'prototype',
|
||||
'cal_path': './config/cal2.npy'
|
||||
},
|
||||
]
|
||||
|
||||
@@ -138,7 +143,7 @@ def main():
|
||||
for i, config in enumerate(sensor_configs):
|
||||
p = Process(
|
||||
target=sensor_worker,
|
||||
args=(i, config['vid_src'], config['config_name'], result_queues[i], stop_events[i])
|
||||
args=(i, config['vid_src'], config['config_name'], config['cal_path'], result_queues[i], stop_events[i])
|
||||
)
|
||||
p.start()
|
||||
processes.append(p)
|
||||
|
||||
Reference in New Issue
Block a user