中文

RCS-YOLO:一种用于脑肿瘤检测的快速高精度目标检测器

计算机视觉与模式识别 2023-10-04 v2 信号处理 应用统计 机器学习

摘要

凭借在速度和精度之间的优异平衡,前沿的YOLO框架已成为目标检测最高效的算法之一。然而,使用YOLO网络在脑肿瘤检测中的性能鲜有研究。我们提出了一种基于通道混洗的重参数化卷积(RCS-YOLO)的新型YOLO架构。我们提出RCS和RCS的一次性聚合(RCS-OSA),将特征级联与计算效率相关联,以提取更丰富的信息并减少时间消耗。在脑肿瘤数据集Br35H上的实验结果表明,所提模型在速度和精度上均超越YOLOv6、YOLOv7和YOLOv8。值得注意的是,与YOLOv7相比,RCS-YOLO的精度提升1%,推理速度提升60%,达到每秒检测114.8张图像(FPS)。我们提出的RCS-YOLO在脑肿瘤检测任务上取得了最先进的性能。代码可在 https://github.com/mkang315/RCS-YOLO 获取。

关键词

引用

@article{arxiv.2307.16412,
  title  = {RCS-YOLO: A Fast and High-Accuracy Object Detector for Brain Tumor Detection},
  author = {Ming Kang and Chee-Ming Ting and Fung Fung Ting and Raphaël C. -W. Phan},
  journal= {arXiv preprint arXiv:2307.16412},
  year   = {2023}
}