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

最新推荐文章于 2024-06-06 10:51:22 发布

Melody袁

最新推荐文章于 2024-06-06 10:51:22 发布

阅读量890

点赞数

文章标签：计算机视觉深度学习 python

本文链接：https://blog.csdn.net/qq_43941925/article/details/125097854

版权

来源：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 v{medmnist.__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])