Course project of Machine Learning (PKU 2023 Autumn)
- Install all the requirements with
pip install -r requirements.txt
- Install SAM
- Download BTCV dataset.
- Each task has a separate folder named
task*
(* is 1/2/3), modify path to BTCV dataset in config fileconfig.yaml
. - Run
main.py
in foldercode/task*
- We provide visulization tool as
code/task3/visulize3D.py
Here we provide pretrained classifier and fintuned SAM model with best mDice score(prompt type is 'bbox')