农业机械学报

作者：梁秀英，贾学镇，何磊，王翔宇，刘岩，杨万能

单位：华中农业大学,华中农业大学,华中农业大学,华中农业大学,华中农业大学,华中农业大学,

关键词：小鼠行为;多目标跟踪;YOLO v8n-seg;Strongsort

基金：国家自然科学基金项目（U21A20205）、湖北省洪山实验室重大项目（2022hszd024）和山东省重点研发计划项目（2022CXGC010609）

多目标小鼠跟踪是小鼠行为分析的基本任务，是研究社交行为的重要方法。针对传统小鼠跟踪方法存在只能跟踪单只小鼠以及对多目标小鼠跟踪需要对小鼠进行标记从而影响小鼠行为等问题，提出了一种基于实例分割网络YOLO v8n-seg和改进Strongsort相结合的多目标小鼠无标记跟踪方法。使用RGB摄像头采集多目标小鼠的日常行为视频，标注小鼠身体部位分割数据集，对数据集进行增强后训练YOLO v8n-seg实例分割网络，经过测试，模型精确率为97.7%，召回率为98.2%，mAP50为99.2%，单幅图像检测时间为3.5ms，实现了对小鼠身体部位准确且快速地分割，可以满足Strongsort多目标跟踪算法的检测要求。针对Strongsort算法在多目标小鼠跟踪中存在的跟踪错误问题，对Strongsort做了两点改进：对匹配流程进行改进，将未匹配上目标的轨迹和未匹配上轨迹的目标按欧氏距离进行再次匹配；对卡尔曼滤波进行改进，将卡尔曼滤波中表示小鼠位置和运动状态的小鼠身体轮廓外接矩形框替换为以小鼠身体轮廓质心为中心、对角线为小鼠体宽的正方形框。经测试，改进后Strongsort算法的ID跳变数为14，MOTA为97.698%，IDF1为85.435%，MOTP为75.858%，与原Strongsort相比，ID跳变数减少88%，MOTA提升3.266个百分点，IDF1提升27.778个百分点，与Deepsort、ByteTrack和Ocsort相比，在MOTA和IDF1上均有显著提升，且ID跳变数大幅降低，结果表明改进Strongsort算法可以提高多目标无标记小鼠跟踪的稳定性和准确性，为小鼠社交行为分析提供了一种新的技术途径。

来源：2024年第2期

《农业机械学报》期刊编辑部

查看农业机械学报杂志2024年第2期