簡體   English   中英

在本地部署經過培訓的Sagemaker Mxnet對象檢測模型?

[英]Deploy trained Sagemaker Mxnet object detection model locally?

我在AWS Sagemaker上訓練了一個對象檢測模型。 我想在我的機器上本地使用這個模型。 我下載了這個模型,包括3個文件hyperparams.json,model-symbol.json和model-0000.params。 我已經看到很多教程在本地部署對象分類模型,但沒有得到任何對象檢測。

有不同的可能性:

  • 您可以在您的計算機上本地安裝SageMaker Python SDK,並使用SageMaker在本地訓練和部署您的模型因此您需要在MXNet Estimator中指明train_instance_type ='local'。 然后SageMaker將S3中的數據輸入您的本地機器並在那里訓練模型。 然后,mxnet_estimator.deploy將在本地部署模型。 你可以在這里找到詳細的描述
  • 您可以使用MXNet的Model Server。 您需要將模型文件轉換為模型存檔(.mar)您可以按以下方式部署它: mxnet-model-server --models mymodel=mymodel.mar

正如NRauschmayr所提到的,如果您還使用它來在SageMaker上進行訓練,那么使用SageMaker的Python SDK本地模式功能會很方便。 稍微糾正一下,您將預測器的instance_type設置為“local”,因為您對本地部署感興趣。 您可以在此處閱讀有關本地模式的更多信息: https//sagemaker.readthedocs.io/en/stable/overview.html#local-mode

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM