GitHub

SimpleITK==1.2.4

matplotlib==3.3.4

nibabel==3.2.1

scikit-image==0.17.2

torch==1.5.1

tqdm==4.55.1

运行时需修改mytrain.py和predict.py中的数据集路径

best_model.pth为预训练模型，可直接加载使用

Name		Name	Last commit message	Last commit date
Latest commit History 33 Commits
MedMnist		MedMnist
test_result		test_result
val_result		val_result
KiUnet.py		KiUnet.py
README.md		README.md
UNet.py		UNet.py
best_model.pth		best_model.pth
dataset.py		dataset.py
evaluation.py		evaluation.py
lossfunction.py		lossfunction.py
metrics.py		metrics.py
modifiedUnet.py		modifiedUnet.py
mytrain.py		mytrain.py
nii_dataset.py		nii_dataset.py
predict.py		predict.py
transforms.py		transforms.py

Provide feedback