yolov7对视频文件进行人体关键点检测

Question

用yolov7对视频进行人体关键点检测，在run方法中传入了一个视频文件。如果想同时检测多个视频文件应该怎么改？

def run(poseweights="yolov7-w6-pose.pt", source="", device='cpu', view_img=False,
        save_conf=False, line_thickness=3, hide_labels=False, hide_conf=True):
    frame_count = 0  # count no of frames
    total_fps = 0  # count total fps
    time_list = []  # list to store time
    fps_list = []  # list to store fps

    device = select_device(opt.device)  # select device
    half = device.type != 'cpu'

    model = attempt_load(poseweights, map_location=device)  # Load model
    _ = model.eval()
    names = model.module.names if hasattr(model, 'module') else model.names  # get class


    if source.isnumeric():
        cap = cv2.VideoCapture(int(source))  # pass video to videocapture object
    else:
        cap = cv2.VideoCapture(source)  # pass video to videocapture object

    if (cap.isOpened() == False):  # check if videocapture not opened
        print('Error while trying to read video. Please check path again')
        raise SystemExit()

    else:
        frame_width = int(cap.get(3))  # get video frame width
        frame_height = int(cap.get(4))  # get video frame height

        vid_write_image = letterbox(cap.read()[1], (frame_width), stride=64, auto=True)[0]  # init videowriter
        resize_height, resize_width = vid_write_image.shape[:2]
        out_video_name = f"{source.split('/')[-1].split('.')[0]}"
        out = cv2.VideoWriter(f"{source}_keypoint.mp4",
                              cv2.VideoWriter_fourcc(*'mp4v'), 30,
                              (resize_width, resize_height))

Answer 1

• 这个问题的回答你可以参考下: https://ask.csdn.net/questions/7418672
• 你也可以参考下这篇文章：YoloV7实战：手把手教你使用Yolov7进行物体检测（附数据集）
• 除此之外, 这篇博客: YOLOV7详细解读（三）技术要点归纳中的 一、YOLOV7是什么？ 部分也许能够解决你的问题, 你可以仔细阅读以下内容或跳转源博客中阅读:
  

• YOLO算法作为one-stage目标检测算法最典型的代表，其基于深度神经网络进行对象的识别和定位，运行速度很快，可以用于实时系统。
  YOLOV7是目前YOLO系列最先进的算法，在准确率和速度上超越了以往的YOLO系列。
  了解YOLOV7是对目标检测算法研究的一个必须步骤。

• 您还可以看一下 白勇老师的YOLOv5实战口罩佩戴检测课程中的 准备数据集小节, 巩固相关知识点