AWS Sagemaker - 添加新类来检测

Question

我有一个成功训练的 sagemaker 模型，可以检测图像中的特定对象（罐头）。如果我想添加一个新的类（塑料瓶）来检测：

1. 我可以使用我原来的模型并添加一个新的类来检测吗？
2. 如果是，我的新训练集是否需要标记两个类别？例如。我想开始检测塑料瓶并且有一个训练数据集。我还需要在这些图片中给罐子贴上标签吗？

Answer 1

这是一个非常笼统的问题，如果不了解您正在使用的算法，就无法提供说明性指导。

如果您使用的是 SagMaker Object detection algorithm，这个算法确实支持incremental training.

请注意：

The input hyperparameters of both the new model and the pretrained model that you upload to the model channel must have the same settings for the base_network and num_classes input parameters.

SageMaker也是如此图像分类算法

The input hyperparameters of both the new model and the pretrained model that you upload to the model channel must have the same settings for the num_layers, image_shape and num_classes input parameters