arXiv:2502.00076v1 交叉类型: cross 摘要: 在近期的研究中，已经探讨了在无线胶囊内窥镜(WCE)中使用深度学习进行病理检测的问题。然而，深度学习模型可能会受到用于训练它们的数据集的颜色质量的影响，从而影响检测、分割和分类任务。在这项工作中，我们评估了颜色校正对病理检测的影响，使用了两种流行的物体检测模型：Retinanet和YOLOv5。我们首先使用两种不同的颜色校正函数生成了流行WCE数据集（即SEE-AI数据集）的两个颜色校正版本。然后，我们在原始数据集和颜色校正版本的数据集上评估了Retinanet和YOLOv5的性能。结果表明，颜色校正使得模型生成更大的边界框以及与真实标注更大的交集区域。此外，颜色校正还会导致某些病理情况下假阳性数量增加。然而，这些效果并未在如F1分数、IoU和AP50等性能指标上转化为一致的改进。代码可在https://github.com/agossouema2011/WCE2024获得。关键词: 无线胶囊内窥镜、颜色校正、Retinanet、YOLOv5、检测