Whole Sclice Images Features Classification

Environment

• python 3.11
• pip 23.3.1
• torch 2.5.1
• pytorch-lightning 1.7.7

pip install -r requirements.txt

Train

python main.py --stage train --config path/to/config --batch_size 32

Resume checkpoint

python main.py --stage train --config path/to/config --batch_size 32 --resume path/to/model

Test

python main.py --stage test --config path/to/config --resume path/to/model