激光与光电子学进展, 2020, 57 (18): 181022, 网络出版: 2020-09-02
基于FL-ResNet50的皮肤镜图像分类方法 下载: 1449次
Dermoscopic Image Classification Method Based on FL-ResNet50
图像处理 卷积神经网络 皮肤镜图像 图像分类 样本不平衡 数据增强 损失函数 image processing convolutional neural network dermoscopy image image classification sample imbalance data augmentation loss function
摘要
提出一种采用深度卷积神经网络模型对七类病变皮肤镜图像进行分类的方法。使用数据增强方法扩增训练集,提出一种基于ResNet50模型和多分类Focal Loss函数的多分类模型(FL-ResNet50模型),实现皮肤镜图像的多分类。实验结果表明,FL-ResNet50模型的微平均F1值为0.88,优于传统的ResNet50模型。所提方法实现了对七类皮肤镜图像的分类,将图像预处理、特征提取及预测模型学习形成一个完整连续的系统模型,提高了分类性能和效率,具有重要的应用价值。
Abstract
In this paper, a classification method for seven types of dermoscopic images based on deep convolution neural network model is proposed. The training set is amplified using the data augmentation method. For the multiclassification of dermoscopic images, the multiclassification model (FL-ResNet50 model) based on ResNet50 model and multiclassification Focal Loss function is proposed. The experimental results show that the micromean F1 value of FL-ResNet50 model is 0.88, which is better than the results obtained using the traditional ResNet50 model. The proposed method realizes seven types of dermoscopic image classification and forms a complete and continuous system model by image preprocessing, feature extraction, and prediction model learning. The FL-ResNet50 model improves upon the classification performance and efficiency of the previous models and has important application value.
罗清, 周维, 马梓钧, 许海霞. 基于FL-ResNet50的皮肤镜图像分类方法[J]. 激光与光电子学进展, 2020, 57(18): 181022. Qing Luo, Wei Zhou, Zijun Ma, Haixia Xu. Dermoscopic Image Classification Method Based on FL-ResNet50[J]. Laser & Optoelectronics Progress, 2020, 57(18): 181022.