提示
基于改进yolov5的黑色素瘤图像自动诊断
首发时间：2024-07-08
摘要：为解决现有黑色素瘤智能诊断模型中存在的对毛发遮挡目标识别精度不足、样本不均以及轻量化程度不够的问题，提出一种改进的yolov5模型。首先，基于改进的c3结构和自注意力机制设计cs_neck结构，从而有效区分黑色素瘤和毛发的相关特征；其次，提出一种二次筛选难样本挖掘方法，利用焦点损失函数降低简单样本权重，引入损失秩排序(loss rank mining，lrm)思想降低简单样本数量；最后，设计轻量级骨干网络，提出使用改进的repvgg结构替换普通卷积提取特征，提高推理速度，并引入宽度乘子降低参数量和权重，实现模型轻量化。基于isic2019数据集的实验结果表明，所提算法的权重和参数量仅为7.9 mb和4.0 m，精度达到92.9%。所提算法有效提升了精度且实现了轻量化，可以满足高效诊断黑色素瘤的要求。
关键词： 黑色素瘤检测 yolov5 注意力机制 难样本挖掘 轻量化
for information in english, please click here
automatic diagnosis of melanoma image based on improved yolov5
abstract：in order to solve the problems of insufficient recognition accuracy, uneven samples and insufficient lightweight of hair occlusion targets in existing melanoma intelligent diagnosis models, an improved yolov5 model was proposed. firstly, the cs_neck structure was designed based on the improved c3 structure and self-attention mechanism, so as to effectively distinguish the related features of melanoma and hair. secondly, a method of mining difficult samples with secondary screening is proposed, which uses focus loss function to reduce the weight of simple samples, and introduces the idea of loss rank mining (lrm) to reduce the number of simple samples. finally, the lightweight backbone network is designed, and the improved repvgg structure is proposed to replace the common convolutional extraction features, improve the inference speed, and the width multiplier is introduced to reduce the number of parameters and weights, so as to realize the lightweight model. the experimental results based on isic2019 data set show that the weights and parameters of the proposed algorithm are only 7.9 mb and 4.0 m, and the accuracy reaches 92.9%. the proposed algorithm can effectively improve the accuracy and achieve lightweight, and can meet the requirements of efficient diagnosis of melanoma.
keywords： melanoma detection yolov5 attention mechanism mining hard samples lightweight
论文图表：
引用
导出参考文献
no.****
同行评议
勘误表
基于改进yolov5的黑色素瘤图像自动诊断
评论
全部评论