tensorflow - 是否有用于图像分类的 TPU 兼容的预训练检查点？

Question

我发现 TPU 兼容的预训练检查点可用于对象检测，但不适用于图像分类。我对 Mobilenet_v2 特别感兴趣。如果这样的检查点可用而 Mobilenet_v2 的检查点不可用，我会切换到另一个网络。

Answer 1

是的，这些 GCS 文件中有一个与 TPU 兼容的 Resnet 检查点：

gs://cloud-tpu-artifacts/resnet/resnet-nhwc-2018-02-07/checkpoint
gs://cloud-tpu-artifacts/resnet/resnet-nhwc-2018-02-07/model.ckpt-112603.data-00000-of-00001
gs://cloud-tpu-artifacts/resnet/resnet-nhwc-2018-02-07/model.ckpt-112603.index
gs://cloud-tpu-artifacts/resnet/resnet-nhwc-2018-02-07/model.ckpt-112603.meta

如果它符合您的要求，您可以使用此检查点。如果没有，您可以按照本教程自行训练。