基于多模态机器学习的膝关节骨关节炎进展的平原X线片和临床数据预测。
纸张代码和预先训练的模型。
Arxiv预印本: ://arxiv.org/abs/1904.06236
(c)奥卢大学Aleksei Tiulpin,2018-2019年。
关于
该存储库包含完整的代码,可重现本文中的培训过程。 要从头开始训练模型,您需要从MOST和OAI数据集中获取DICOM图像。 您还需要获取相应的元数据(可从网站下载。中提供了有关获取数据的更多说明。可在找到注释这些图像所需的元数据。
安装,培训和评估
依存关系
为了运行该项目,我们使用了Ubuntu 16.04 , Docker和nvidia-docker 。 这些是您唯一需要的软件依赖项。 请安装这些就可以了。 如果不想使用Docker ,则可以按照给定的Dockerfile遵循安装过程。
我们使用3xGTX1080Ti NVIDIA
1