基于yolov8的人员轨迹跟踪算法

详侃历史

于 2024-07-17 04:27:08 发布

阅读量35

点赞数

文章标签： YOLO

基于yolov8的人员轨迹跟踪算法_深度学习

欢迎关注我👆，收藏下次不迷路┗|｀O′|┛ 嗷~~

一.效果

二.部分代码

三.获取项目

一.效果

基于yolov8的人员轨迹跟踪算法_ide_02

二.部分代码

### Detect, track, annotate, save
    with sv.VideoSink(target_path=target_video_path, video_info=video_info) as sink:
        for frame in frames_generator:
            result = model(
                source=frame,
                classes=[0],  # only person class
                conf=confidence_threshold,
                iou=iou_threshold,
                # show_conf = True,
                # save_txt = True,
                # save_conf = True,
                # save = True,
                device=None,  # use None = CPU, 0 = single GPU, or [0,1] = dual GPU
            )[0]

            detections = sv.Detections.from_ultralytics(result)  # get detections

            detections = byte_tracker.update_with_detections(
                detections
            )  # update tracker

            ### draw heatmap
            annotated_frame = heat_map_annotator.annotate(
                scene=frame.copy(), detections=detections
            )

            ### draw other attributes from `detections` object
            labels = [
                f"#{tracker_id}"
                for class_id, tracker_id in zip(
                    detections.class_id, detections.tracker_id
                )
            ]

            label_annotator.annotate(
                scene=annotated_frame, detections=detections, labels=labels
            )