我是在colab上面运行,示例笔记本, 推理时间咋这么长呀?
Colab只作为示例,我一般使用本地GPU进行测试,平均时间在6min左右。
作者写的test代码中推理过程,其实是把一整个病例作为推理数据,也就是一个病例数据不仅只有一张图像,而是"ind"张图像的叠加,比如case0008是148张叠加的(可以看case0008.npy.h5 的信息),所以推理一个病例是每一张叠加的ind数据都处理完之后才出结果的。