RetinaMnist数据集介绍以及下载使用

Posted 2022-06-03 Melody袁

来源：https://medmnist.com/
RetinaMnist数据集是MedMnist数据集中的一个，它的各项参数如下：

数据集统计	值
图像大小	宽度： 28， 高度： 28
训练样本数	1080
验证样品数量	120
测试样品数量	400

该数据集其实来源是DeepDrid数据集，只是像素不同，格式不同，该数据集为npz格式的数据集，官方提供了使用方式。而且数据集的划分是固定的，还有一些已有的数据供我们参考学习，避免了重复性的实验。
通过代码对数据集中的各个分类进行统计，图表如下所示：

示例图片（由于像素很低，很模糊）：

下面讲解一下官方的使用方式：

pip install medmnist

from tqdm import tqdm
import numpy as np
import torch
import torch.nn as nn
import torch.optim as optim
import torch.utils.data as data
import torchvision.transforms as transforms

import medmnist
from medmnist import INFO, Evaluator

print(f"MedMNIST vmedmnist.__version__ @ medmnist.HOMEPAGE")

打印成功，表示下载成功。

from PIL import Image
import numpy as np
import matplotlib.pyplot as plt

path1 = "retinamnist.npz"
data1 = np.load(path1)

img_data1 = data1['train_images']
img_label1 = data1['train_labels']

plt.xlabel("No DR")
plt.imshow(img_data1[0])
print(img_label1[0])

查看其中的图像：

具体的其他使用可以看官方代码：
https://github.com/MedMNIST/MedMNIST