使用YOLOv8和ByteTracker进行实时人员跟踪和计数(转载)
1.原文参见:
https://mp.weixin.qq.com/s/pmhlttCbfwvKoozlh29FOA
2.重要代码:
(1)self.model = YOLO(model_path)
model_path:yolov8m.pt,是自动下载,原文件设置错误。
(2)results = self.model.track(
source, show=show, stream=True, tracker=self.tracker_config, conf=self.conf,
device=self.device, iou=self.iou, stream_buffer=True, classes=[0], imgsz=self.img_size
)
3.源码:https://github.com/ishworrsubedii/person_tracker
(1)params.yaml 为总运行的参数。- device: "cpu" 在本人机器上的设置。
- video_source: palace.mp4。palace.mp4是源码中所带的一个文件。为0时是摄像头。
-
tracker_config: "resources/config/bytetrack.yaml"训练参数的设置。
-
model_path: "resources/models/yolov8m.pt"模型的设置。(
(2) log_params.yaml 日志文件的设置
- 在uils中 logger.py的def setup_logger中设置。
- 在ipeline.py中的def run_pipeline中调用。
logger = setup_logger('PersonTrackerLogger', 'tracker.log')
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· 没有Manus邀请码?试试免邀请码的MGX或者开源的OpenManus吧
· 【自荐】一款简洁、开源的在线白板工具 Drawnix
· 园子的第一款AI主题卫衣上架——"HELLO! HOW CAN I ASSIST YOU TODAY
· Docker 太简单,K8s 太复杂?w7panel 让容器管理更轻松!