我们可以在 AWS 的 SageMaker 中训练 TensorFlow 自定义对象检测模型吗？

Question

您能帮我解决以下几点吗：

1. 能否在 AWS 的 SageMaker 中训练 TensorFlow 自定义对象检测模型？

2. 我遇到了 SageMaker 的图像分类算法？我们可以在训练模型后使用它来检测视频中的特定对象吗？

3. 对 SageMaker 的定价计划感到困惑。他们说“为您提供每月 250 小时 t2.medium 笔记本使用时间的免费套餐”；这是否意味着我们可以免费使用 t2.medium notebook 250 小时？

最终目标是训练一个用于自定义对象检测的模型，就像我们过去在纸空间或 floydhub 中以非常低的价格训练一样。

提前致谢。

Answer 1

1- 当然。您可以将任何 TensorFlow 代码带到 SageMaker。 https://docs.aws.amazon.com/sagemaker/latest/dg/tf-examples.html

2- 这是一个分类模型（仅标签），而不是检测模型（标签 + 边界框）。话虽如此，是的，您绝对可以使用它来预测从视频中提取的帧。

3- 是的，在您创建 AWS 账户后的前 12 个月内。

希望这会有所帮助。

Answer 2

任何 TensorFlow 模型都可以使用/移植到 SageMaker。您可以在此处https://github.com/awslabs/amazon-sagemaker-examples/tree/master/sagemaker-python-sdk#amazon-sagemaker-examples 找到移植到 SageMaker 的 TensorFlow 模型示例。