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

时间:2018-04-26 10:54:34

标签: amazon-web-services tensorflow object-detection-api amazon-sagemaker

你能否帮我解决以下几点:

  1. 我们可以在AWS的SageMaker中训练tensorflow自定义对象检测模型吗?

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

  3. 对SageMaker的定价计划感到困惑。他们说"每月免费提供250小时的t2.medium笔记本电脑使用时间&#34 ;;这是否意味着我们可以免费使用t2.medium笔记本250小时?

  4. 最终的AIM是训练自定义物体检测的模型,就像我们以非常低的价格在paperpace或floydhub进行训练一样。

    提前致谢。

2 个答案:

答案 0 :(得分:2)

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

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

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

希望这会有所帮助。

答案 1 :(得分:1)

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