本文属于机器翻译版本。若本译文内容与英语原文存在差异，则一律以英文原文为准。

# 图像分类- TensorFlow 算法的输入和输出接口
<a name="IC-TF-inputoutput"></a>

 TensorFlow Hub Models 中列出的每个预训练模型都可以微调到具有任意数量图像类的任何数据集。请注意如何格式化训练数据，以便输入到图像分类- TensorFlow 模型中。
+ **训练数据输入格式：**您的训练数据应该是一个目录，其中具有与类的数量相同的子目录数。每个子目录都应包含属于该类的图像，且格式为 .jpg、.jpeg 或 .png。

以下是输入目录结构的示例。此示例数据集有两个类：`roses` 和 `dandelion`。每个类文件夹中的图像文件可以使用任意名称。输入目录应托管在 Amazon S3 存储桶中，路径类似于如下所示：`s3://bucket_name/input_directory/`。请注意，结尾的 `/` 是必需的。

```
input_directory
    |--roses
        |--abc.jpg
        |--def.jpg
    |--dandelion
        |--ghi.jpg
        |--jkl.jpg
```

经过训练的模型输出标签映射文件，这些文件将类文件夹名称映射到输出类概率列表中的索引。此映射按字母顺序排列。例如，在前面的示例中，dandelion 类的索引为 0，roses 类的索引为 1。

训练完成后，您将得到经过微调的模型，您可以使用增量训练对其进行进一步训练，也可以部署该模型进行推理。图像分类 TensorFlow 算法会自动将预处理和后处理签名添加到经过微调的模型中，以便它可以将图像作为输入并返回类概率。将类索引映射到类标签的文件与模型一起保存。

## 增量训练
<a name="IC-TF-incremental-training"></a>

您可以使用之前使用 SageMaker AI 训练过的模型中的工件为新模型的训练做种子。当您想训练具有相同或类似数据的新模型时，这种增量训练可节省训练时间。

**注意**  
您只能为图像分类（带有另一种 SageMaker 图像分类的 TensorFlow 模型）播种，该 TensorFlow 模型是在 SageMaker AI 中训练的。

只要类别集合保持不变，就可以使用任何数据集进行增量训练。增量训练步骤与微调步骤类似，但不是使用预先训练的模型开始，而是从现有的微调模型开始。有关使用 SageMaker AI 图像分类 TensorFlow 算法进行增量训练的示例，请参阅 “[图像分类简介](https://github.com/aws/amazon-sagemaker-examples/blob/main/introduction_to_amazon_algorithms/image_classification_tensorflow/Amazon_TensorFlow_Image_Classification.ipynb)” 示例笔记本。 SageMaker TensorFlow 

## 使用图像分类进行推理- TensorFlow 算法
<a name="IC-TF-inference"></a>

您可以托管 TensorFlow 图像分类训练产生的微调模型以进行推理。任何用于推理的输入图像都必须采用 `.jpg`、.`jpeg` 或 `.png` 格式，并且内容类型为 `application/x-image`。图像分类- TensorFlow 算法会自动调整输入图像的大小。

运行推理会生成概率值、所有类的类标签以及与概率最高的类索引对应的预测标签，这些标签以 JSON 格式编码。图像分类- TensorFlow 模型根据请求处理单个图像，并且仅输出一行。以下是 JSON 格式响应的示例：

```
accept: application/json;verbose

 {"probabilities": [prob_0, prob_1, prob_2, ...],
  "labels":        [label_0, label_1, label_2, ...],
  "predicted_label": predicted_label}
```

如果 `accept` 设置为 `application/json`，则模型仅输出概率。有关使用图像分类 TensorFlow 算法进行训练和推理的更多信息，请参阅 “[图像分类简介](https://github.com/aws/amazon-sagemaker-examples/blob/main/introduction_to_amazon_algorithms/image_classification_tensorflow/Amazon_TensorFlow_Image_Classification.ipynb)” 示例笔记本。 SageMaker TensorFlow