Home

Awesome

Pytorch Medical Classification

<i>Read Chinese Introduction:<a href='https://github.com/MontaEllis/Pytorch-Medical-Classification/blob/main/README-zh.md'>Here!</a></i><br />

Recent Updates

Requirements

Notice

Prepare Your Dataset

Example

if your source dataset is :

categpry-0
├── source_1.png
├── source_2.png
├── source_3.png
└── ...
categpry-1
├── source_1.png
├── source_2.png
├── source_3.png
└── ...

then your should modify fold_arch as *.png, source_train_0_dir as categpry-0 and source_train_1_dir as categpry-1 in hparam.py

Training

set hparam.train_or_test to 'train'
python main.py
set hparam.train_or_test to 'train'
set hparam.ckpt to True
python main.py

Inference

set hparam.train_or_test to 'test'
python main.py

Done

Network

TODO

By The Way

This project is not perfect and there are still many problems. If you are using this project and would like to give the author some feedbacks, you can send Kangneng Zhou an email, his wechat number is: ellisgege666

Acknowledgements

This repository is an unoffical PyTorch implementation of Medical segmentation in 3D and 2D and highly based on pytorch-cifar100 and torchio.Thank you for the above repo. Thank you to Cheng Chen and Weili Jiang for all the help I received.