是否可以在ultralytics yolov3、Pytorch上同时运行2个视频

Question

我想知道是否可以同时运行 2 个视频作为 ultralytics yolo v3 的输入..

我的目标是融合两个视频的bbox...（数据融合）

但我很好奇这最初是否可能。

我想问一下有人以前尝试过类似的事情..

尝试操纵 detector.py

Answer 1

是的，可以使用 Ultralytics YOLOv3 和 PyTorch 同时处理两个视频。

Python 示例代码：

...
import multiprocessing
...
if __name__ == "__main__":
    video1 = 'path_to_video1.mp4'
    video2 = 'path_to_video2.mp4'
    
    p1 = multiprocessing.Process(target=process_video, args=(video1,))
    p2 = multiprocessing.Process(target=process_video, args=(video2,))
    
    p1.start()
    p2.start()
    
    p1.join()
    p2.join()
...

Detect.py 中的 LoadStreams 类：

if webcam:
    video_paths = ['path_to_video1.mp4', 'path_to_video2.mp4']
    dataset = LoadStreams(video_paths, img_size=imgsz)
    bs = len(dataset)
elif screenshot:
...