简介

EasyMedAI 是一个基于 PyTorch 和 MMEngine 实现的，用于医疗图像大模型下游任务训练的快速开始库，包含集成主流医疗数据。

医疗数据集

• MedMNIST 18x Standardized Datasets for 2D and 3D Biomedical Image Classification.
• MSD Generalisable 3D Semantic Segmentation

开始

1.安装

从源代码安装

git clone https://github.com/yinghemedical/EasyMedAI.git
cd EasyMedAI/EasyMedAI
pip install .

从pip安装

pip install EasyMedAI

安装Aim

pip install aim==3.20.1

2.准备数据集

3.开始训练