arcface_learning_ranzcr 源码

概括 这是用于arcface学习的管道,旨在用于我们的管道中的检索预测。 怎么跑 将ChestX数据集( )下载到此目录并解压缩,然后将其命名为chestx 。 如果您有kaggle API,请执行以下操作: ! kaggle datasets download -d nih-chest-xrays/data 打开ranzcr_chestx_arcface_learning.ipynb并点击运行所有 完成2后,打开generate_features_chest_x.ipynb并点击运行所有 生成的文件将是chest_x_features.npy ,其中包含ChestX数据集中每个图像的向量表示形式以及一个名为dense121_feature_extractor.pth的模型。 这两个文件将在推理过程中使用: : scriptVersionId 预训练的模型和功能下载链接 che