基于旋转目标检测器实现毒麻药品检测
计算机视觉与模式识别
2021-10-20 v1 人工智能
摘要
近年来，深度学习视觉技术在医疗行业的进展与应用有目共睹。对特殊药品管理的智能设备需求巨大，这需要更精确的检测算法来识别药品的规格与位置。在这项工作中，基于 YOLO（You only look once）的目标检测器被定制用于毒麻药品检测任务。具体而言，提出了一种角度从 0° 到 90° 的更灵活旋转标注，以及一种基于掩码映射的非极大值抑制方法，以实现面向任意方向边界框的可行且高效的药品检测器。大量实验表明，旋转 YOLO 检测器更适用于识别密集排列的药品。所提出网络的最佳平均精度达到 0.811，而推理时间小于 300ms。
引用
@article{arxiv.2110.09777,
title = {Towards Toxic and Narcotic Medication Detection with Rotated Object Detector},
author = {Jiao Peng and Feifan Wang and Zhongqiang Fu and Yiying Hu and Zichen Chen and Xinghan Zhou and Lijun Wang},
journal= {arXiv preprint arXiv:2110.09777},
year = {2021}
}