CovidPrognosis:基于MoCo训练图像表示的X射线胸片的COVID恶化预测 源码

Covid预后 该存储库包含用于复制以下论文的代码: 我们还提供了MoCo预培训过程中的模型,供有兴趣根据自己的数据进行微调的小组使用。 在使用此代码或预先训练的模型之前,请查阅。 安装 首先,按照安装PyTorch。 然后,导航到CovidPrognosis根目录并运行 pip install -e . 之后,您应该可以在cp_examples运行示例。 用法 对于预训练,您需要下载或数据集。 下载数据后,将路径添加到configs/data.yaml ,并且应将其用作默认路径。 cp_examples目录包含三个子目录,分别对应于本文中的培训阶段: moco_pretrai