Automatic Grading of Knee Osteoarthritis from Plain Radiographs using Densely Connected Convolutional Networks

Dataset: https://drive.google.com/file/d/13G_wXTSWDQHJpF8s428eshiGHDS_Opfq/view?usp=sharing

First, download and unpack oai224.zip to dataset/ directory.

Training (see train.ipynb)

Customize parameters in block Changeable parameters and run.

Testing (see test.ipynb)

Customize parameters in block Choose architecture and model checkpoint and run.

Test ensembles (see test_ensemble.ipynb)

Customize parameters in block Prepare ensemble parts and run.