使用AWS Marketplace上的模型

SageMaker自身与AWS Marketplace集成,开发者可以到上面卖算法模型给其他用户。AWS Marketplace上简化了license的购买过程,而且提供了多种定价选择给用户。
本节我们将直接使用Marketplace上的一个模型做测试🐶

进入SageMaker,点击AWS Marketplace -> Model packages:

image-20220726220255537

这里我们选择YOLOv3的物体检测模型,点击View Product

image-20220726220313789

YOLOv3(you only look once)目前是业界比较流行的检测算法,在速度和精度上相对于其他检测算法都有很大的优势。针对于其他目标检测算法普遍存在的运算速度慢的缺点,yolo创造性的提出了one-stage,也就是将物体分类和物体定位在一个步骤中完成。

在新的marketplace页面中可以查看它的介绍及收费信息等。点击Try Product Demo进行测试:

image-20220726220351586

将以下两张图片下载到本地:

https://pingfan.s3.amazonaws.com/files/road.jpg

https://pingfan.s3.amazonaws.com/files/kitchen.jpg

然后分别上传进行测试:

image-20220726222039881

上传完成后,YOLOv3模型会输出检测到的物体:

image-20220726222300710

点击具体的物体,能查看置信度:

image-20220726222316001

image-20220726222339033