MobileNet混淆矩阵

数据集获取（Figshare）

• 源站标题：Confusion matrix for MobileNet.
• DOI：10.1371/journal.pone.0327743.t009
• 发布日期：2025-08-01
• 文件数：1 个
• 文件总体积：5.63 KB
• 源站页面：https://plos.figshare.com/articles/dataset/Confusion_matrix_for_MobileNet_/29799357

命令行下载

curl -L -o "Table 9.xls" "https://ndownloader.figshare.com/files/56837754"

文件清单

• Table 9.xls（5.63 KB）

文件总体积 5.63 KB（源站 API 实测）。

数据加载示例（图像类）

from PIL import Image
import glob, os

files = (glob.glob(os.path.join(path, "**", "*.png"), recursive=True)
       + glob.glob(os.path.join(path, "**", "*.jpg"), recursive=True)
       + glob.glob(os.path.join(path, "**", "*.tif"), recursive=True))
print("图像文件数:", len(files))
img = Image.open(files[0]); print("尺寸/模式:", img.size, img.mode)

# torchvision Dataset 方式:
# from torchvision import datasets
# ds = datasets.ImageFolder(path)  # 要求 子目录=类别

目录组织与标注格式以源站说明和下载后实际文件为准。

数据说明

Introduction:
Retinal diseases, a significant global health concern, often lead to severe vision impairment and blindness, resulting in substantial functional and social limitations. This study explored a novel goal of developing and comparing the performance of multiple state-of-the-art convolutional neural network (CNN) models for the automated detection and classification of retinal diseases using optical coherence tomography (OCT) images.
Method:
The study utilized several models, including DenseNet121, ResNet50, Inception V3, MobileNet, and OCT images obtained from the WATBORG Eye Clinic, to detect and classify multiple retinal diseases such as glaucoma, macular edema, posterior vitreous detachment (PVD), and normal eye cases. The preprocessing techniques employed included data augmentation, resizing, and one-hot encoding. We also used the Gaussian Process-based Bayesian Optimization (GPBBO) approach to fine-tune the hyperparameters. Model performance was evaluated using the F1-Score, precision, recall, and area under the curve (AUC).
Result:
All the CNN models evaluated in this study demonstrated a strong capability to detect and classify various retinal diseases with high accuracy. MobileNet achieved the highest accuracy at 96% and AUC of 0.975, closely followed by DenseNet121, which had 95% accuracy and an AUC of 0.963. Inception V3 and ResNet50, while not as high in accuracy, showed potential in specific contexts, with 83% and 79% accuracy, respectively.
Conclusion:
These results underscore the potential of advanced CNN models for diagnosing retinal diseases. With the exception of ResNet50, the other CNN models displayed accuracy levels that are comparable to other state-of-the-art deep learning models. Notably, MobileNet and DenseNet121 showed considerable promise for use in clinical settings, enabling healthcare practitioners to make rapid and accurate diagnoses of retinal diseases. Future research should focus on expanding datasets, integrating multi-modal data, exploring hybrid models, and validating these models in clinical environments to further enhance their performance and real-world applicability.

许可

CC BY 4.0

引用

学术使用请引用 DOI 10.1371/journal.pone.0327743.t009（source: https://plos.figshare.com/articles/dataset/Confusion_matrix_for_MobileNet_/29799357）。