Skip to content

Latest commit

 

History

History
27 lines (20 loc) · 765 Bytes

REPRODUCIBILITY.md

File metadata and controls

27 lines (20 loc) · 765 Bytes

Instructions

Model was trained on a GTX 1080Ti home server.

cd src

## Separate from database
python3 train_val_csv.py

## Training
python3 train.py --classes 3 --part cardio --network resent50
python3 train.py --classes 4 --part main --network resent101
python3 train.py --classes 4 --part nothing --network resent101
python3 train.py --classes 7 --part lungs --network resent101
python3 train.py --classes 4 --part pleural --network resent101

## Result
python3 inference.py

## If you want to see the loss curves
tensorboard --logdir=../logs