改进YOLOv5s的输液瓶目标检测

作者:喻旸; 周箩鱼; 熊莉娅; 邹学玉*
来源:电子世界, 2022, (01): 182-197.
DOI:10.19353/j.cnki.dzsj.2022.01.085

摘要

<正>针对医疗环境趋于复杂,为提高输液瓶目标检测的精度,本文利用视频和图像素材,提出一种改进的YOLOv5s方法检测输液瓶目标。采用CIo U Loss和DIo U NMS等方法进行优化。在PASCAL VOC数据集和自建输液瓶目标数据集上检测m AP(mean Average Precision)和FPS(FramesPerSecond)。相较于原始YOLOv5s,改进后的YOLOv5sCIo U在m AP上提高5.53%,检测速度可达34.7FPS。

全文